M - Marvelous Necklace（前缀和）

题目



题意:给出一个字符串（首位相接），由A,B组成，若是能分割成A,B数量一样的两段，输出"YES"，输出截取的位置两端a,b;否则输出"NO";
思路：题目时间是500ms.字符串为1e5,若是普通暴力会超时，进行优化思路。
字符串中A的数量为AA，B的数量为BB；
1.AA和BB为任一为奇数就输出“No；
2.否则先记录从i=0到i<len/2，计数A，B的个数（已经为题目要求的长度len/2）：若是AAA/2&&BBB/2，输出“Yes”，1,len/2+1两个端点。否则继续判断。
3.从i=len/2到i=len-1，动态更新len/2长度的A与B的数量，若是满足
AAA/2&&BBB/2，记录当前位置now,标记为1；更新之后若flag=1,输出“Yes”,输出两个端点;否则输出“No”。

AC代码

#include <bits/stdc++.h>

using namespace std;
string s;
int main()
{
    cin>>s;
    int len=s.size();
    int AA=0,BB=0;
    for(int i=0; i<len; i++)
    {
        if(s[i]=='A')
        AA++;
        else
        BB++;
    }
    if(AA%2!=0||BB%2!=0)
    cout<<"NO"<<endl;
    else
    {
        int A=0,B=0;
        for(int i=0; i<len/2; i++)
        {
            if(s[i]=='A')
            A++;
            else
            B++;
        }
        if(A==AA/2&&B==BB/2)
        {
            cout<<"YES"<<endl;
            cout<<1<<" "<<len/2+1<<endl;
        }
        else
        {
            int flag=0,now=0;
            for(int i=len/2; i<len; i++)
            {
                if(s[i]=='A')
                A++;
                else
                B++;
                if(s[i-len/2]=='A')
                A--;
                else
                B--;
                if(A==AA/2&&B==BB/2)
                {
                    now=i;
                    flag=1;
                }
            }
            if(flag==1)
            {
                cout<<"YES"<<endl;
                cout<<now+2-len/2<<" "<<now+2<<endl;
            }
            else
                cout<<"NO"<<endl;

        }
    }
    return 0;
}