2017百度之星资格赛:1005. 寻找母串(卡特兰数+分块打表)

xiaoxiao2021-02-28  80

寻找母串

   Accepts: 105    Submissions: 887  Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others) Problem Description

对于一个串S,当它同时满足如下条件时,它就是一个01偏串: 1、只由0和1两种符组成; 2、在S的每一个前缀中,0的个数不超过1的个数; 3、S中0的个数和1的个数相等。

现在给定01偏串S,请计算一下S在所有长度为n的01偏串中作为子串出现的次数的总和。 由于结果比较大,结果对1e9+7取余后输出。

样例解释: 在第二个样例中,长度为4的偏串共两个1010,1100。10在1010中出现了两次,在1100中出现了1次。所以答案是3。

Input

第一行给出一个整数T(1<=T<=40),表示测试数据的数目。 每一组测试包含一个整数n和字符串S,中间用空格分开。(1<=|S|<=100000,1<=n<=1000000000)

输入保证S是一个01偏串。

Output

对于每一组数据,输出一个整数占一行,表示答案。

Sample Input Copy 2 2 10 4 10 Sample Output 1 3

可能我姿势不对。。这表怎么就打了1年

这题看起来束手无策,但其实你可以先写个暴力观察小数据,这个时候你就会发现其实这题比你想象的要简单的多

(比如你会发现其实答案之和字符串的长度|S|有关,和字符串的内容是无关的)

没错!这题有规律,答案就是

但是别高兴的太早,这个时候你又会发现其实这题比你想象的要难得多

因为这题的n范围巨大(约10亿)而求10亿的组合数是做不到的

所有先考虑化简公式看看,令x = (n-|S|)/2+1

其中F[x]是第x项卡特兰数

而卡特兰数有O(n)的递推公式

这样的话理论上可以O(n)求出所有的答案,可还是不行。。。

复杂度已经不能再优化了。。所以只能考虑打表

可是你又开不了10亿的数组,但是这题也没有说要O(1)询问呀

没错!分块打表

你只需要后台O(n)暴力出第100000个卡特兰数,第200000个卡特兰数……第500000000个卡特兰数就好了

也就是开个5000+的数组c[],其中c[i]是第100000*i个卡特兰数

然后对于每组询问(n, |S|),先计算x=(n-|S|)/2+1,然后再找到x所在的那一块(比x小离x最近的100000的倍数),

之后暴力转移就好,最多转移100000次

因为有除法所以要乘法逆元,总体复杂度O(100000log(n))

还有注意n为奇数的时候答案一定为0,因为母串要保证0和1的数量相等,所以不存在长度为奇数的母串

除此之外,n<|S|答案也为0

下面附上暴力程序+打表程序

#include<stdio.h> #include<string.h> char str[100005], jud[100005]; int Jud(int n) { int i, s1, s0; s1 = s0 = 0; for(i=1;i<=n;i++) { if(jud[i]=='1') s1++; else s0++; if(s0>s1) return 0; } if(s1!=s0) return 0; return 1; } int main(void) { int T, n, m, i, j, k, sum, ans; scanf("%d", &T); while(T--) { sum = ans = 0; scanf("%d%d", &n, &m); for(i=1;i<=m/2;i++) str[i] = '1'; for(i;i<=m;i++) str[i] = '0'; str[m+1] = 0; m = strlen(str+1); for(i=0;i<=(1<<n)-1;i++) { for(j=0;j<=n-1;j++) { jud[j+1] = '0'; if(i&(1<<j)) jud[j+1] = '1'; } jud[j+1] = 0; if(Jud(n)) { sum++; for(j=1;j<=n;j++) { for(k=1;k<=m;k++) { if(jud[j-1+k]!=str[k] || j-1+k>n) break; } if(k==m+1) ans++; } } } printf("%d\n", ans); } return 0; } /* #include<stdio.h> #define mod 1000000007 #define LL long long LL Pow(LL a, LL b) { LL sum = 1; while(b) { if(b%2) sum = sum*a%mod; a = a*a%mod; b /= 2; } return sum; } int main(void) { LL i, now = 1; freopen("ct.txt", "w", stdout); printf("{1"); for(i=1;i<=500000000;i++) { now = 2*(2*i-1)*now%mod*Pow(i+1, mod-2)%mod; if(i0000==0) printf(",%lld", now); } printf("};"); return 0; } */

转载请注明原文地址: https://www.6miu.com/read-41891.html

最新回复(0)