Krdytkyu

I am solving ROCK problem on SPOJ. I dont know why I am getting WA here. Tried many testcases.

#include <bits/stdc++.h>

using namespace std;



#define fastIO ios_base::sync_with_stdio(false);cin.tie(NULL)

#define pb push_back

#define mp make_pair

#define ll long long int

#define ull unsigned long long int

#define mod 1000000007

#define UB upper_bound

#define LB lower_bound



int main()

{

    fastIO;

    #ifndef ONLINE_JUDGE

        freopen("input.txt", "r", stdin);

        freopen("output.txt", "w", stdout);

    #endif



    int t;

    cin>>t;

    while(t--)

    {

        int n,i,j;

        string s;

        map <int,int> m;

        cin>>n>>s;

        int a[n+1];

        a[0]=0;

        for(i=0;i<n;i++)

        {

            int x=(s[i]=='1') ? 1 : -1; 

            a[i+1]=a[i]+x;

        }



        for(i=1;i<=n;i++)

        {

            m[a[i]]=i;

        }

        int ans=0,p=0;

        for(i=1;i<=n;i++)

        {

            int r=0;

            int j=i;

            while(j<=n)

            {

                int p=a[j-1];

                int x=m[p+1];

                if(x<j)

                {

                    j++;

                    continue;

                }

                r+=(x-j+1);

                j=x+1;

            }

            ans=max(ans,r);

        }

        cout<<ans<<"n";

    }

    return 0;

}

Please suggest where my code is wrong. Thanks