『狄利克雷卷积』优秀序列

$\mathrm{P}\mathrm{r}\mathrm{o}\mathrm{b}\mathrm{l}\mathrm{e}\mathrm{m}$

$\mathrm{S}\mathrm{o}\mathrm{l}\mathrm{u}\mathrm{t}\mathrm{i}\mathrm{o}\mathrm{n}$

显然这道题目和两个函数没有任何关系，仅仅是为了考察你是否会线性筛法以及用于权值的计算。

我们设$f_k(n)$表示k个数乘积为n的权值和。那么就有：$$f_k(n)=\sum _{k|n}{f}_p(k)\times f_{k-p}(\frac{n}{k})$$
然后观察一下这一个算式的特性，我们发现其实$p$在这里可以是任意的一个数。

那么我们将这里的一次操作记为一次卷积。那么我们发现在这里其实就是做k次卷积。

因此我们需要用到狄利克雷卷积算法。由于该卷积支持快速幂，我们可以使用卷积快速幂来实现。

具体做法如下：

• 根据权值记初始序列为$f$，另$g_1=1$.在这里，$f$相当于快速幂内的$a$，$k$相当于$b$，而在$g$数组中统计答案。那么我们再做$k$次快速幂以后，我们只需要输出$g[1...n]$即可。
• 当$k$是奇数时，我们对$f$和$g$同时进行一次卷积。
• 每一次都对$f$自身完成一次卷积。

我们可以用倍数法完成卷积，时间复杂度$O(n\mathrm{l}\mathrm{o}\mathrm{g}n)$.

#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;
const int N = 2e5;
const int P = 998244353;

int cnt = 0, n, k;
int phi[N], mul[N], vis[N], prime[N], f[N], g[N], res[N];

int read(void)
{
	int s = 0, w = 0; char c = getchar();
	while (c < '0' || c > '9') w |= c == '-', c = getchar();
	while (c >= '0' && c <= '9') s = s*10+c-48, c = getchar();
	return w ? -s : s;
}

void init(void)
{
	phi[1] = 1, mul[1] = 1;
	for (int i=2;i<=1e5;++i)
	{
		if (vis[i] == 0) mul[i] = -1, phi[i] = i-1, prime[++cnt] = i;
		for (int j=1;j<=cnt&&i*prime[j]<=1e5;++j)
		{
			vis[i*prime[j]] = 1;
			if (i % prime[j] == 0)
			{
				mul[i*prime[j]] = 0;
				phi[i*prime[j]] = phi[i]*prime[j]; 
				break;
			}
			mul[i*prime[j]] = -mul[i];
			phi[i*prime[j]] = phi[i]*(prime[j]-1); 
		}
	}
	g[1] = 1;
	for (int i=1;i<=n;++i) f[i] = 2 * phi[i] + mul[i];
	return; 
}

void Mul(void)
{
	memset(res,0,sizeof res);
	for (int i=1;i<=n;++i)
	    for (int j=1;i*j<=n;j++)
		    res[i*j] = 1LL * (res[i*j] + 1LL * f[i] * g[j]) % P;
	for (int i=1;i<=n;++i)
		g[i] = res[i];
	return; 
}

void mulself(void)
{
	memset(res,0,sizeof res);
	for (int i=1;i<=n;++i)
	    for (int j=1;i*j<=n;++j)
	        res[i*j] = 1LL * (res[i*j] + 1LL * f[i] * f[j]) % P;
	for (int i=1;i<=n;++i)
	    f[i] = res[i];
	return;
}

void power(int k)
{
	while (k > 0)
	{
		if (k & 1) Mul();
		mulself(), k >>= 1;
	}
	for (int i=1;i<=n;++i)
	return;
}

int main(void)
{
	n = read(), k = read();
	init();
	power(k);
	for (int i=1;i<=n;++i) printf("%d ", g[i]);
	return 0;
}