ZOJ - Problem Set - 3985 String of CCPC

#include<bits/stdc++.h>
#include<cmath>
#define mem(a,b) memset(a,b,sizeof a);
using namespace std;
typedef long long ll;
const int maxn=2e5+10;
char s[maxn];
//int vis[maxn]; // 太大了，可以放在main函数里面
int main()
{
    int T; scanf("%d",&T);
    int n;
    while(T-- && ~scanf("%d%s",&n,s))
    {
//        int vis[n+10]; // AC
//        mem(vis,0);
        int rs=0;
        for(int i=0;i<n-3;i++)
        {
            if(s[i]=='C' && s[i+1]=='C' && s[i+2]=='P' && s[i+3]=='C')
            {
                rs++;
//                vis[i+1]=vis[i+2]=1;
                s[i+1]=s[i+2]='#'; // 替代每次都需要mem(vis,0) TLE
            }
        }
        for(int i=0;i<n-2;i++)
        {
            if(s[i]=='C' && s[i+1]=='C' && s[i+2]=='C') // && !vis[i+2]
            {
                rs++;
                break;
            }
            if(s[i]=='C' && s[i+1]=='P' && s[i+2]=='C') // && !vis[i] && !vis[i+1]
            {
                rs++;
                break;
            }
            if(s[i]=='C' && s[i+1]=='C' && s[i+2]=='P') // && !vis[i+1] && !vis[i+2]
            {
                rs++;
                break;
            }
        }
        printf("%d\n",rs);
    }
    return 0;
}