1. 程式人生 > >2018 ACM-ICPC 亞洲區域賽青島站 C-Flippy Sequence(計數)

2018 ACM-ICPC 亞洲區域賽青島站 C-Flippy Sequence(計數)

 

思路來源

https://blog.csdn.net/Game_Acm/article/details/83745455

題意

給兩個01串A串和B串,在A串中可以進行區間翻轉操作,可以使得一個區間內所有的0變成1,所有的1變成0,求恰好翻轉兩次使得A串變為B串的方案數。

兩個方案數不同當且僅當至少有一個區間端點不同。

題解

先將兩個串模擬異或一下得到一個新串。

統計新串的1的段數,

 

若大於3,則無法通過兩次操作還原,為0

 

若為2,不妨記[l1,r1]在新串中是1,[l2,r2]在新串中是1,l1<r1<l2<r2

則答案一定是[l1,r1,l2,r2],[l1,l2-1,r1+1,r2],[l1,r2,r1+1,l2-1]三種,考慮順序共6種

即恰好翻轉兩個區間,和將中間的空白區間翻轉兩次情況下翻轉整個區間。

 

若為1,不妨記[l3,r3]在新串中是1,

首先,考慮只還原1的情形,即加擋板將[l3,r3]分為兩段,r3-l3種,

考慮將0的部分翻轉兩次,

①固定左端點為l3的情形,右端點在r3右取,ans+=n-r3

②固定右端點為r3的情形,左端點在l3左取,ans+=l3-1

三段相加為n-1,考慮順序,2*(n-1)

 

若為0,考慮將某段區間翻轉2次,

答案為區間數,n*(n+1)/2

程式碼

#include <iostream>
#include <algorithm> 
#include <cstring>
#include <cstdio>
#include <cmath>
#include <set>
#include <map>
#include <vector>
#include <bitset>
#include <stack>
#include <queue>
#include <functional>
const int INF=0x3f3f3f3f;
const int maxn=1e6+10; 
const int mod=1e9+7;
const int MOD=998244353;
const double eps=1e-7;
typedef long long ll;
#define vi vector<int> 
#define si set<int>
#define pii pair<int,int> 
#define pi acos(-1.0)
#define pb push_back
#define mp make_pair
#define lowbit(x) (x&(-x))
#define sci(x) scanf("%d",&(x))
#define scll(x) scanf("%lld",&(x))
#define sclf(x) scanf("%lf",&(x))
#define pri(x) printf("%d",(x))
#define rep(i,j,k) for(int i=j;i<=k;++i)
#define per(i,j,k) for(int i=j;i>=k;--i)
#define mem(a,b) memset(a,b,sizeof(a)) 
using namespace std;
int t,n,cnt,now,q;
char a[maxn],b[maxn],ans[maxn];
int main()
{  
   sci(t);
   while(t--)
   {
   	cnt=0,now=0,q=0;
   	sci(n);
   	scanf("%s",a);
   	scanf("%s",b);
   	rep(i,0,n-1)
   	{
   		if(a[i]=='0'&&b[i]=='0')ans[i]='0';
   		else if(a[i]=='1'&&b[i]=='1')ans[i]='0';
   		else ans[i]='1';
   		if(ans[i]=='1')now++;
   		else if(now)
		{
		   if(cnt==0)q=now;
		   cnt++,now=0;
	    }
   	}
   	if(now)
	{
	   if(cnt==0)q=now;
	   cnt++,now=0;
        } 
   	if(cnt>2)puts("0");
	else if(cnt==2)puts("6");
	else if(cnt==0)printf("%lld\n",n*(n+1)/2);
	else printf("%d\n",2*(n-1));//n-q+(q-1)
	}
   return 0;
}