题解 [CF327C] Magic Five

luogu 题目链接

这种题都要写好久，wtcl。

下文下标从 $0$ 开始。$n$ 为原序列长度。

考虑 $k=1$ 的情况，显然要以 0 或 5 结束，前面随便选多少个，后面全不选。若前面有 $p$ 个数，方案数为 $2^p$。

若 $k\ne 1$，设 $mul=\sum _{i=0}^{k-1}{2}^{in}$，答案乘上 $mul$ 即可。

为什么呢？若原序列中第 $p$ 位为 0 或 5，那么总序列中它会出现在 $p,p+n,p+2n,...,p+(k-1)n$。总方案数显然为 $2^p+2^{p+n}+...+2^{p+(k-1)n}$，即 $2^p\ast mul$。

这里 $k$ 很大，$mul$ 可以分治计算。设 $f_p$ 表示 $\sum _{i=0}^p{2}^{in}$，则：

$$\{\begin{array}{llcc}{f}_p=f_{(p-1)/2}\ast 2^{(p+1)/2\ast n}+1& & if& p\equiv 1(\mathrm{m}\mathrm{o}\mathrm{d}2)\\ f_p=f_{p-1}+2^{pn}& & if& p\equiv 0(\mathrm{m}\mathrm{o}\mathrm{d}2)\end{array}$$

最后 $mul$ 就是 $f_{k-1}$。

//CF327C
#include <cstring> 
#include <cstdio>
typedef long long ll;

const int N = 1e5 + 10;
const ll P = 1e9 + 7;
ll mul, ans;
char s[N];
int k, n;

ll qp(ll a, ll b){
	ll ans = 1;
	while(b){
		if(b&1){
			ans = ans * a % P;
		}
		a = a * a % P;
		b >>= 1;
	}
	return ans;
}
ll calc(ll k){
	if(k == 0){
		return 1;
	}
	if(k&1){
		return calc(k/2) * (qp(2, ((k+1)/2)*n) + 1) % P;
	} else {
		return (calc(k-1) + qp(2, k*n)) % P;
	}
}

int main(){
	scanf("%s%d", s, &k);
	n = strlen(s);
	mul = calc(k-1);
	for(int i = 0; i < n; ++ i){
		if(s[i] == '0' || s[i] == '5'){
			ans += qp(2, i);
			ans %= P;
		}
	}
	printf("%lld\n", ans * mul % P);
	return 0;
}
111