【ZOJ4060】 Flippy Sequence（规律）

题意：给定两个长度为n的01序列A和B，要求分别从两个序列中取两段将段中数字取反，使得A和B完全相同，求方案数

n<=1e6，sum(n)<=1e7

思路：现场8Y……

将A和B异或之后问题转化成了在一个01串中取两端，将段中数字取反，使得最终为全0序列，求方案数

找规律之后可得与连续的1的段数有关，设段数为s

s=0时ans=(n+1)*n/2

s=1时ans=2*(n-1)

s=2时ans=6

s>2时ans=0

#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#include<vector>
using namespace std;
typedef long long ll;
typedef unsigned int uint;
typedef unsigned long long ull;
typedef pair<int,int> PII;
typedef vector<int> VI;
#define fi first
#define se second 
#define MP make_pair
#define N      1100000
#define M      51
#define MOD 1000000007
#define eps 1e-8 
#define pi     acos(-1)
#define oo     1010000000

char ch[N];
int a[N],b[N],c[N];

int main()
{ 
    //freopen("zoj4060.in","r",stdin);
    //freopen("zoj4060.out","w",stdout);
     int cas;
     scanf("%d",&cas);
     for(int v=1;v<=cas;v++) 
     {
         int n;
         scanf("%d",&n);
         scanf("%s",ch+1);
         for(int i=1;i<=n;i++) a[i]=ch[i]-'0';
         scanf("%s",ch+1);
         for(int i=1;i<=n;i++) b[i]=ch[i]-'0';
         for(int i=1;i<=n;i++) c[i]=a[i]^b[i];
         int s=0;
         c[0]=c[n+1]=0;
         for(int i=0;i<=n;i++)
          if(c[i]==0&&c[i+1]==1) s++;
         ll ans=0;
         if(s==0) ans=(ll)(n+1)*n/2;
         if(s==1) ans=(n-1)*2;
        if(s==2) ans=6;
        if(s==3) ans=0;
        printf("%lld
",ans); 
    }
    return 0;
}