2019ICPC区域赛南昌 C - And and Pair

2019ICPC南昌 C - And and Pair

一：题目：

一个巨大的数 n 可表示为由0、1组成的字符串s，寻找可能的数对 ( i , j ) ,满足条件：

1：0 <= j <= i <= n

2: i & n = i

3: i & j = 0

求每个n最多有多少数对(i,j)

其中字符串S长度限定为 $0<=length<=10^5$

样例1：S = 111 ， num = 14

样例2：S = 1010，num = 15

二：分析

法一：找规律 + 二项式定理 + 快速幂预存

举个例子：S = 10010110

i & n = i : 字符串位上是0时，i 必须为0，当字符串位上是1时，i取值0，1均可

i & j = 0 : i，j位上不能同时存在1和1

通过枚举找规律可得 对于任意 i ：j可行的个数为 $2^i$个

当且仅当S的最高位为1时, i 的可能取值 为 100_0_ _0

这里面进行分类讨论：(i，j可能组合的个数)

​ 其他位均为 0 ：$C_3^0\ast 2^7$ (0的个数为7)

​ 存在一个1：$C_3^1\ast 2^6$

​ 存在两个1：$C_3^2\ast 2^5$

​ 存在三个1：$C_3^3\ast 2^4$

​ sum = $C_3^0\ast 2^7+C_3^1\ast 2^6+C_3^2\ast 2^5+C_3^3\ast 2^4$

继续化简：

​ $sum=2^4\ast (C_3^0\ast 2^3+C_3^1\ast 2^2+C_3^2\ast 2^1+C_3^3\ast 2^0)$

​ 由二项式定理得：

$(x+1{)}^n=C_n^0{X}^0+C_n^1{X}^1+C_n^2{X}^2+...+C_n^n{X}^n$

故 $sum=2^k\ast 3^t$

其中 k = 除最高位以外 0 的个数

​ t = 除最高位以外 1 的个数

最后遍历字符串，针对为1的位置操作

注：快速幂nlogn预测 $2^n,3^n$

​ 最后加 1 ，i = 0000时永远可行

AC代码

#include <bits/stdc++.h>
#define ll long long
#define maxn 100001
using namespace std;
const ll mod = 1000000007;

ll pow_2[maxn+10],pow_3[maxn+10];

ll power(ll a,ll b){
	ll res = 1;
	while(b){
		if(b&1) res = res*a%mod;
		a = a*a%mod;
		b>>=1;
	}
	return res;
}

void init(){
	for(int i=0;i<=maxn;i++){
		pow_2[i] = power(2,i);
		pow_3[i] = power(3,i);
	}
}

int main(){
	init();
	int t;
	string s;	
	cin>>t;
	while(t--){
		ll num0=0,num1=0,sum=0;
		cin>>s;
		for(int i=0;i<s.size();i++){
			if(s[i]=='0') num0++;
			else num1++;
		}
		for(int i=0;i<s.size();i++){
			if(s[i]=='0'){
				num0--;
				continue;
			}
			else{
				num1--;
				sum = (sum + pow_2[num0] * pow_3[num1] % mod) % mod;
			}
		}
		cout<<(sum+1) % mod<<endl;
	}
	return 0;
}

法二 ： 数位DP

(待更…)