C.Tokitsukaze and Balance String (hard)

合集 - 牛客训练(31)

1.校门外的树（一维差分）2024-12-222.铺地毯（二维差分/枚举区间）2024-12-223.晾衣服（二分）2024-12-234.丢手绢（尺取法/快慢指针）2024-12-255.二分（离散化/哈希）2024-12-286.C-猪猪养成计划1（set）01-187.E-双生双宿之错01-228.D-字符串里串01-249.H-一起画很大的圆01-2510.C-字符串外串01-2511.E—一起走很长的路01-2712.M—数值膨胀之美01-2813.J—硝基甲苯之袭01-3014.H—井然有序之窗02-0115.C—智乃的NotePad（esay version）02-0116.I. 小鸡的排列构造的checker02-1617.J. 铁刀磨成针02-1618.华华开始学信息学02-1619.B.好伙计猜拳02-1720.E.小L的井字棋02-1721.C.小L的位运算02-1722.D.Tokitsukaze and Concatenate‌ Palindrome02-18

23.C.Tokitsukaze and Balance String (hard)02-18

24.E.智乃的小球02-1925.G.智乃与模数02-2026.D.小L的字符串翻转02-2027.L.小L的构造02-2028.C.兢兢业业之移02-2229.F.双生双宿之探02-2230.E.全都要03-0331.D.大预言家03-03

收起

题目链接：https://ac.nowcoder.com/acm/contest/95336/C

题意：

给定一个字符串s，规定一个字符串平衡 当且仅当01子串数量=10子串数量
字符串中有p个？，？需要将其变为0或1
规定一个字符串的val为 将某个0/1反置后得到的字符串是平衡的 位置数量
求这个字符串的val和

思路：

记‘？’数量为cnt
发现首尾两个字符如果相同，那么该字符串一定平衡，反之如果不同，一定不平衡
所以当首尾两个字符都不为‘？’时，1.首尾相同，只能反置首尾之间的数字，且反置后一定平衡，val=pow（2，cnt）（n-2）2.首尾不同 ，只能先反置首或尾val=pow（2，cnt）2
易得当首尾有1个‘？’或2个‘？’时，val=pow（2，cnt-1）*n
特判n==1
乘法取余
发现当cnt过大时，pow（2，cnt）溢出ll，需要用同余定理一边乘2一遍取模

#include<bits/stdc++.h>
#define rep(i,a,n) for(int i=a;i<=n;i++)
#define pb push_back
#define endl "\n"
#define int long long
#pragma GCC optimize(3)
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int inf=0x3f3f3f3f;
const ll llmax=LLONG_MAX;
const int maxn=1e5+5;
int n;
const ll mod=1e9+7;
signed main()
{
	ios::sync_with_stdio(false),cin.tie(0);
	int T;cin>>T;
	while(T--){
		cin>>n;
		string s;cin>>s;
		ll cnt=0;
		ll ans=0;
		if(n==1){
			if(s=="?")ans=2;
			else ans=1;
			cout<<ans<<endl;continue;
		}
		rep(i,0,n-1) if(s[i]=='?')cnt++;
		
		int p=1,q=1;
		rep(i,1,cnt){
			p*=2;
			p%=mod;
		}
		rep(i,1,cnt-1){
			q*=2;
			q%=mod;
		}
		char a=s[0],b=s[n-1];
		if(a!='?'&&b!='?'){
			if(a==b){
				ans=p*(n-2);
			}else{
				ans=p*2;
			}
		}
		else{
			ans=q*n;
		}
		cout<<ans%mod<<endl;	
	}	
	return 0;
}