【SDOI 2015】序列统计

在遭受了巨大的精神打击后,我决定了,要做一个压行的好孩子!!!

博客如果有不对之处,请一定要指出啊啊啊啊啊谢谢谢谢谢大佬!!!

我们首先可以写出一个递推式:

f [ a + c ] [ b ∗ d 取 模 m o d ] = f [ a ] [ b ] ∗ f [ c ] [ d ] f[a+c][b*d取模mod]=f[a][b]*f[c][d] f[a+c][bdmod]=f[a][b]f[c][d]

可以发现,这个格式并不是卷积。那我们怎么办呢?

我们知道,有 a i ∗ a j = a i + j a^i*a^j=a^{i + j} aiaj=ai+j,那我们可以将式子转化为原根i次方。然后我们就可以用卷积来做了。

然后我再来说一下打法:solve函数类似快速幂,每一次乘法相当于添加一个数(就像dp外层的1至n的循环)。而g数组则表示相应的余数对应的方案数。

最后,因为运算时不带取模,我们需要再加上 a [ i + m − 1 ] a[i+m-1] a[i+m1]。为什么加m-1而不是加m?因为此处保证m为质数,也就是说m的欧拉函数是m-1。(如有不懂欧拉函数的童鞋们可以去康康 )那么,我们可以保证的是,0到m-2构成的数,在m-1及之后相同的长度还有一毛一样的。而这取模m-1就是相同的余数。为什么只加一次呢?这要看我们变换的范围。

再最后,我们知道m-1并不一定是阶。这就可能有0到m-2之间有重复余数但不相加的情况。为什么可以直接输出呢?仔细看看我们的初始化,我们将输入的值为余数处理。实际上,这些不同的编号(i)代表的是同一个数。

awsl感觉自己在口胡。不知所言。

所以又到了激动人心的上代码环节。

#include<cstdio>
#include<iostream>
using namespace std;

const int mod = 1004535809, N = 50002;
int n, m, X, len, inv, lim = 1, rev[N], ori, L, pos = -1, a[N], b[N], g[N], f[N];
bool vis[N];

int qkpow(int x, int y, int p) {
	x %= p;
	int res = 1;
	while(y) {
		if(y & 1)
			res = 1ll * res * x % p;
		x = 1ll * x * x % p;
		y >>= 1;
	}
	return res;
}

int cal(const int p) {//找原根
	if(p == 2)
		return 1;
	bool flag;
	for(int i = 2; ; ++ i) {
		flag = 0;
		for(int j = 2; j * j < p; ++ j)
			if(qkpow(i, (p - 1) / j, p) == 1) {
				flag = 1;
				break;
			}
		if(! flag)
			return i;
	}
}

void Swap(int &x, int &y) {
	x ^= y ^= x ^= y;
}

void NTT(int *t, const int op) {
	for(int i = 0; i < lim; ++ i) if(i < rev[i]) Swap(t[i], t[rev[i]]);
	for(int mid = 2; mid <= lim; mid <<= 1) {
		int per = qkpow(3, (mod - 1) / mid, mod);
		if(op == -1) per = qkpow(per, mod - 2, mod);
		for(int i = 0; i < lim; i += mid) {
			int w = 1;
			for(int k = 0; k < (mid >> 1); ++ k, w = 1ll * w * per % mod) {
				int X = t[i + k], Y = 1ll * w * t[i + k + (mid >> 1)] % mod;
				t[i + k] = (X + Y) % mod;
				t[i + k + (mid >> 1)] = (X - Y + mod) % mod; 
			}
		}
	}
}

void mul(int *g, int *f) {
	for(int i = 0; i < lim; ++ i) a[i] = g[i] % mod, b[i] = f[i] % mod;
	NTT(a, 1);
	NTT(b, 1);
	for(int i = 0; i < lim; ++ i) a[i] = 1ll * a[i] * b[i] % mod;
	NTT(a, -1);
	for(int i = 0; i < lim; ++ i) a[i] = 1ll * a[i] * inv % mod;//因为在模意义之下,不能直接除以lim
	for(int i = 0; i < m - 1; ++ i)	g[i] = (a[i] + a[i + m - 1]) % mod;
}

void solve(int y) {
	inv = qkpow(lim, mod - 2, mod);
	g[0] = 1;
	while(y) {
		if(y & 1) mul(g, f);
		mul(f, f);
		y >>= 1;
	}
}

void init(const int Lim) {
	while(lim < Lim) {
		lim <<= 1;
		++ L;
	}
	for(int i = 0; i < lim; ++ i) rev[i] = (rev[i >> 1] >> 1) | ((i & 1) << (L - 1));
}

int main() {
	ios::sync_with_stdio(false);
	cin.tie(NULL);
	int num;
	cin >> n >> m >> X >> len;
	ori = cal(m);
	for(int i = 1; i <= len; ++ i) cin >> num, vis[num] = 1;
	for(int i = 0, j = 1; i < m - 1; ++ i, j = 1ll * j * ori % m) {
		if(vis[j]) f[i] = 1;//有这种形式的g^i的数,其中g为原根
		if(X == j) pos = i;
	}
	init(m - 1 << 1);
	solve(n);
	if(pos != -1) cout << g[pos] % mod;
	else cout << "0";
	putchar('\n');
	return 0;
}

谢谢!

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值