P4841 城市规划 FFT+生成函数

\(\color{#0066ff}{ 题目描述 }\)

刚刚解决完电力网络的问题, 阿狸又被领导的任务给难住了. 刚才说过, 阿狸的国家有n个城市, 现在国家需要在某些城市对之间建立一些贸易路线, 使得整个国家的任意两个城市都直接或间接的连通. 为了省钱, 每两个城市之间最多只能有一条直接的贸易路径. 对于两个建立路线的方案, 如果存在一个城市对, 在两个方案中是否建立路线不一样, 那么这两个方案就是不同的, 否则就是相同的. 现在你需要求出一共有多少不同的方案. 好了, 这就是困扰阿狸的问题. 换句话说, 你需要求出n个点的简单(无重边无自环)无向连通图数目. 由于这个数字可能非常大, 你只需要输出方案数mod 1004535809(479 * 2 ^ 21 + 1)即可.

\(\color{#0066ff}{输入格式}\)

一行，四个整数，N、M、x、|S|，其中|S|为集合S中元素个数。第二行，|S|个整数，表示集合S中的所有元素。

\(\color{#0066ff}{输出格式}\)

一行，一个整数，表示你求出的种类数mod 1004535809的值。

\(\color{#0066ff}{输入样例}\)

\(\color{#0066ff}{输出样例}\)

\(\color{#0066ff}{数据范围与提示}\)

对于 20%的数据, n <= 10

对于 40%的数据, n <= 1000

对于 60%的数据, n <= 30000

对于 80%的数据, n <= 60000

对于 100%的数据, n <= 130000

\(\color{#0066ff}{ 题解 }\)

直接构造出g，求个ln（指数生成函数）

最初系数/i!是好算，最后要乘回来

#include<bits/stdc++.h>

#define LL long long

LL in() {

	char ch; LL x = 0, f = 1;

	while(!isdigit(ch = getchar()))(ch == '-') && (f = -f);

	for(x = ch ^ 48; isdigit(ch = getchar()); x = (x << 1) + (x << 3) + (ch ^ 48));

	return x * f;

}

using std::vector;

const int maxn = 1e6 + 20;

const int mod = 1004535809;

int len, r[maxn];

LL ksm(LL x, LL y) {

	LL re = 1LL;

	while(y) {

		if(y & 1) re = re * x % mod;

		x = x * x % mod;

		y >>= 1;

	}

	return re;

}

void FNTT(vector<int> &A, int flag) {

	A.resize(len);

	for(int i = 0; i < len; i++) if(i < r[i]) std::swap(A[i], A[r[i]]);

	for(int l = 1; l < len; l <<= 1) {

		int w0 = ksm(3, (mod - 1) / (l << 1));

		for(int i = 0; i < len; i += (l << 1)) {

			int w = 1, a0 = i, a1 = i + l;

			for(int k = 0; k < l; k++, a0++, a1++, w = 1LL * w0 * w % mod) {

				int tmp = 1LL * A[a1] * w % mod;

				A[a1] = ((A[a0] - tmp) % mod + mod) % mod;

				A[a0] = (A[a0] + tmp) % mod;

			}

		}

	}

	if(!(~flag)) {

		std::reverse(A.begin() + 1, A.end());

		int inv = ksm(len, mod - 2);

		for(int i = 0; i < len; i++) A[i] = 1LL * A[i] * inv % mod;

	}

}

vector<int> operator * (vector<int> A, vector<int> B) {

	int tot = A.size() + B.size() - 1;

	for(len = 1; len <= tot; len <<= 1);

	for(int i = 0; i < len; i++) r[i] = (r[i >> 1] >> 1) | ((i & 1) * (len >> 1));

	FNTT(A, 1), FNTT(B, 1);

	vector<int> ans;

	for(int i = 0; i < len; i++) ans.push_back(1LL * A[i] * B[i] % mod);

	FNTT(ans, -1);

	ans.resize(tot);

	return ans;

}

vector<int> operator - (const vector<int> &A, const vector<int> &B) {

	vector<int> ans;

	for(int i = 0; i < (int)std::min(A.size(), B.size()); i++) ans.push_back(A[i] - B[i]);

	for(int i = B.size(); i < (int)A.size(); i++) ans.push_back(A[i]);

	for(int i = A.size(); i < (int)B.size(); i++) ans.push_back(-B[i]);

	return ans;

}

vector<int> inv(const vector<int> &A) {

	if(A.size() == 1) {

		vector<int> ans;

		ans.push_back(ksm(A[0], mod - 2));

		return ans;

	}

	int n = A.size(), _ = (n + 1) >> 1;

	vector<int> B = A, ans;

	B.resize(_);

	B = inv(B);

	ans.push_back(2);

	ans = B * (ans - A * B);

	ans.resize(n);

	return ans;

}

vector<int> getd(const vector<int> &A) {

	vector<int> ans;

	ans.resize(A.size() - 1);

	for(int i = 1; i < (int)A.size(); i++) ans[i - 1] = 1LL * i * A[i] % mod;

	return ans;

}

vector<int> geti(const vector<int> &A) {

	vector<int> ans;

	ans.resize(A.size() + 1);

	for(int i = 1; i < (int)ans.size(); i++) ans[i] = 1LL * A[i - 1] * ksm(i, mod - 2) % mod;

	return ans;

}

vector<int> ln(const vector<int> &A) {

	vector<int> B = getd(A), C = inv(A);

	return geti(B * C);

}

int main() {

	int n = in();

	vector<int> a;

	static int fac[maxn];

	fac[0] = 1;

	for(int i = 1; i <= n; i++) fac[i] = 1LL * fac[i - 1] * i % mod;

	for(int i = 0; i <= n; i++) a.push_back(1LL * ksm(2, (((LL)i * (i - 1)) / 2)) * ksm(fac[i], mod - 2) % mod);

	a = ln(a);

	printf("%d\n", (int)(1LL * a[n] * fac[n] % mod));

	return 0;

}

巴特西