giúp mình làm bài này bằng c++ với ạ

Bạn thử code sau nhé:

#include <bits/stdc++.h>
using namespace std;
int n,a[100005],f[100005],ans;
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=1;i<=n;i++)
    {
        if(a[i]%2==0) f[i]=f[i-1]+1;
        else f[i]=f[i-1]-1;
    }
    map<int,int> mp;
    mp[0]=1;
    for(int i=1;i<=n;i++)
    {
        ans+=mp[f[i]];
        mp[f[i]]++;
    }
    cout<<ans<<endl;
    return 0;
}