2018 ACM-ICPC 亞洲區域賽青島站 C-Flippy Sequence(計數)
阿新 • • 發佈:2018-12-03
思路來源
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; }