#P13940. [EC Final 2019] Dirichlet k-th root
[EC Final 2019] Dirichlet k-th root
Description
庞数学家在上一次集训中学习了狄利克雷卷积。不过,相比深度强化学习,这对他来说太简单了。因此,他做了一些特别的事情。
如果 是从正整数到整数的两个函数,则狄利克雷卷积 定义为一个新函数:
$$(f * g)(n) =\sum_{d \mid n}f(d)g \left(\frac {n}{d}\right) .$$我们定义函数 的 次幂为
$$f^{k}=\underbrace {f * \dots * f} _{k~{\textrm {次}}}.$$在本题中,我们要求解逆问题:给定 和 ,你需要找到一个函数 ,使得 。
此外,还有一个额外的限制条件: 和 必须等于 。所有的算术运算都在 上进行,其中 ,也就是说,在狄利克雷卷积中,$(f * g)(n) =\left(\sum_{d \mid n}f(d)g \left(\frac {n}{d}\right)\right) \bmod p$。
Input Format
第一行包含两个整数 和 。
第二行包含 个整数 ()。
Output Format
如果无解,输出 。
否则,输出 ()。如果有多组解,输出任意一组即可。
5 2
1 8 4 26 6
1 4 2 5 3
Hint
由 ChatGPT 4.1 翻译
京公网安备 11011102002149号