【训练题65：数论 | 推式子优化】Convolution | 2021牛客暑期多校训练营4

题意

• Convolution | 2021牛客暑期多校训练营4
  假设 $x,y$ 的质因子分解为 $x=\prod p_i^{s_i},y=\prod p_i^{t_i}$
  我们定义运算符 $x\otimes y=\prod p_i^{|s_i-t_i|}$
  现在给你一个长度为 $n$ 的序列 $a_n$ 还有一个 $c$
  现在已知：
  $$b_i=\sum _{i=j\otimes k}{a}_i\cdot k^c$$
  其中 $1\le i,j,k\le n$ ，后文省略
  你需要求出每一个 $b_i$ (取模 $998244353$) ，输出他们的异或和即可。
• $1\le n\le 10^6$
  $0\le a_i<998244353$
  $0\le c\le 10^9$

题意

• 过程非常难想的推式子题目…
  首先，我们直接枚举 $i,j,k$ 当然会超时，考虑 $\otimes$ 运算符的本质
  其实就是
  $$x\otimes y=\frac{lcm(x,y)}{\gcd (x,y)}=\frac{x\cdot y}{\gcd (x,y{)}^2}$$
  有 $lcm$ 式子肯定不好搞，我们使用右边的式子。
  所以现在就变成了：
  $$b_i=\sum _{j=1}^n\sum _{k=1}^n[\frac{j\cdot k}{\gcd (j,k{)}^2}=i]\times a_j\times k^c$$
  很套路的，我们枚举 $\gcd (j,k)=d$ ，同时对称性构造 $j^{\prime }=\frac{j}{d}，k^{\prime }=\frac{k}{d}$ ，其中要求 $j^{\prime }\perp k^{\prime }$
  这个时候式子就变成了：
  $$b_i=\sum _{j=1}^n\sum _{k=1}^n[j^{\prime }\cdot k^{\prime }=i]\times a_j\times k^c$$
  这么一看，就是枚举 $i$ 的两个因子嘛！
  我们转换枚举方式，先枚举一个因子 $j^{\prime }$ ，再枚举其倍数 $d\cdot j^{\prime }$ ，变成了：
  b i = ∑ j ′ ∣ i ∑ d = 1 min ⁡ { n j ′ , n k ′ } a d j ′ × ( d k ′ ) c b_i=\sum_{j^\prime | i} \sum_{d=1}^{\min\{\frac{n}{j^\prime},\frac{n}{k^\prime} \}} a_{dj^\prime} \times (dk^\prime)^c bi​=j′∣i∑​d=1∑min{j′n​,k′n​}​adj′​×(dk′)c
  注意为什么 $d$ 的上限中，分子是 $n$ 不是 $i$ 呢？因为 $d=\gcd (j,k)$，明显 $1\le j,k\le n$
  注意式子中不用枚举 $k^{\prime }$ ，因为此时 $i=j^{\prime }{k}^{\prime }$ ，我们对于 $i$ 只用枚举 $j^{\prime }$ 就能获得 $k^{\prime }$ 了
  换句话说， $k^{\prime }$ 是与 $d$ 无关的，我们令 $d$ 的枚举上限为 $up$ ，所以我们自然左移：
  $$b_i=\sum _{j^{\prime }|i}(k^{\prime }{)}^c\sum _{d=1}^{up}{a}_{d{j}^{\prime }}\times d^c$$
  这样，复杂度我们变成了 $O(n^2\sqrt{n})$ （假设我们提前预处理出快速幂 $d^c$ 的话）
• 注意到，对于同一个 $i$ ，我们第二个求和的式子中，我们重算了许多东西（因为我们只会变 $up$ 的值）
  所以我们记一个辅助数组：
  $$dp[x][y]=\sum _{d=1}^y{a}_{dx}\times d^c$$
  然后原来式子就变成了：
  $$b_i=\sum _{j^{\prime }|i}(k^{\prime }{)}^c\times dp[j^{\prime }][up]$$
  注意到，我们从 $dp[j^{\prime }][x]$ 推到 $dp[j^{\prime }][x+1]$ 是非常好做的，值只增加了 $a_{j^{\prime }(x+1)}\cdot (x+1{)}^c$
  我们还注意到，对于一个 $j^{\prime }$ ，我们不会用到其他的 $dp[x\ne j^{\prime }][y]$ ，所以我们可以省略成一维数组 $dp[up]$
  由于 $up$ 不超过 $\frac{n}{j^{\prime }}$ ，所以我们获取所有 $dp[up]$ 的复杂度是 $O(n\log n)$ 的
  所以我们现在，复杂度都在与求 $b_i$ 数组，复杂度为 $O(n\sqrt{n})$
• 注意到，$b_i$ 是求出 $i$ 的一个因子 $j^{\prime }$ 然后求和
  我们当然可以先枚举因子 $j^{\prime }$ ，再枚举倍数 $k^{\prime }$ ，此时因子的倍数就是 $j^{\prime }\cdot k^{\prime }=i$，就是我们的每一个 $i$ 了
  这样复杂度就可以优化成 $O(n\log n)$ 了！

代码

• 时间复杂度：$O(n\log n)$
  为什么分析这么复杂，代码还是这么短啊！

#include <bits/stdc++.h>
#define IOS ios::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL);
using namespace std;
typedef long long ll;
void show(){std::cerr << endl;}template<typename T,typename... Args>void show(T x,Args... args){std::cerr << "[ " << x <<  " ] , ";show(args...);}

const int MAX = 1e6+50;
const int MOD = 998244353;
const int INF = 0x3f3f3f3f;
const ll LINF = 0x3f3f3f3f3f3f3f3f;
const double EPS = 1e-5;

ll qpow(ll a,ll n){/* */ll res = 1LL;while(n){if(n&1)res=res*a%MOD;a=a*a%MOD;n>>=1;}return res;}
ll qpow(ll a,ll n,ll p){a%=p;ll res = 1LL;while(n){if(n&1)res=res*a%p;a=a*a%p;n>>=1;}return res;}
ll npow(ll a,ll n){/* */ll res = 1LL;while(n){if(n&1)res=res*a;a=a*a;n>>=1;if(res<0||a<0)return 0;}return res;}
ll inv(ll a){/* */return qpow(a,MOD-2);}
ll inv(ll a,ll p){return qpow(a,p-2,p);}

ll aa[MAX];
ll bb[MAX];
ll powc[MAX];
ll dp[MAX];
int main()
{
    int n,c;n = read();c = read();
    for(int i = 1;i <= n;++i)aa[i] = read_ll();
    for(int i = 0;i <= n;++i)powc[i] = qpow(i,c);		// 预处理出 i^c

    for(int j2 = 1;j2 <= n;++j2){
        for(int up = 1;j2 * up <= n;++up){				// 更新 dp[x] 数组
            dp[up] = (dp[up-1] + aa[j2 * up] * powc[up] % MOD) % MOD;
        }
        for(int k2 = 1;j2 * k2 <= n;++k2){
            if(__gcd(j2,k2) != 1)continue;				// 要求如此
            int i = j2 * k2;
            int up = min(n / j2,n / k2);
            bb[i] = (bb[i] + dp[up] * powc[k2] % MOD) % MOD;	// 更新 b[x] 数组
        }
    }

    ll ans = 0;
    for(int i = 1;i <= n;++i)ans ^= bb[i];
    Print(ans);
    Write();
    return 0;
}