2021牛客暑期多校训练营4-H题

最新推荐文章于 2024-09-20 20:16:39 发布

Nep_Genesis

最新推荐文章于 2024-09-20 20:16:39 发布

阅读量149

点赞数

文章标签： c++

本文链接：https://blog.csdn.net/Nep_Genesis/article/details/119188586

版权

本文介绍了如何通过转化条件并应用快速幂和动态规划解决ACM竞赛中的Convolution问题，将LCM和GCD转换为乘积形式，然后高效计算b_i的表达式，最终求解XOR答案。时间复杂度为O(n log n)。

摘要由CSDN通过智能技术生成

H题：Convolution

原题链接:https://ac.nowcoder.com/acm/contest/11255/H

题目大意

请添加图片描述

思路

先对条件进行变换。
$lcm(x,y)=\Pi_{i}p_i^{max(a_i,b_i)}$
$gcd(x,y)=\Pi_{i}p_i^{min(a_i,b_i)}$
$a_i-b_i|=max(a_i,b_i)-min(a_i,b_i)$
$=>x\otimes y=\frac{lcm(x,y)}{gcd(x,y)}=\frac{xy}{gcd^2(x,y)}$
对于序列b
$b_i=\sum_{1\leq j,k\leq n,j\otimes k=i}a_jk^c$
设 $g=gcd(j,k),x=\frac{j}{g},y=\frac{k}{g}$ 可以化简为
$b_i=\sum_{1\leq x,y\leq n,xy=i}a_{xg}{(yg)}^c$
$=>b_i=\sum_{1\leq x,y\leq n,xy=i}y^ca_{xg}{g}^c$
放个dp数组预处理 $a_{xg}g^c$ 然后循环计算 $b_i$ 的值。最后所有 $b_i$ Xor求答案。
时间复杂度 $O(n\ log\ n)$
注：快速幂的值要预处理，否则也会超时。

代码实现

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int mod=998244353;
const int N=1e6+7;
ll n,C,a[N],b[N],ans;
ll dp[N],p[N];
ll gcd(ll a,ll b){
	if(a%b==0) return b;
	return gcd(b,a%b);
}
ll fmx(ll x,ll y)
{
	ll ret=1;
	while(y)
	{
		if(y&1) ret=ret*x%mod;
		x=x*x%mod;
		y>>=1;
	}
	return ret;
}
int main()
{
	scanf("%lld%lld",&n,&C);
	for(int i=1;i<=n;i++)
		scanf("%lld",&a[i]);
	for(int x=1;x<=n;x++){
		p[x]=fmx(x,C);
	}
	for(int i=1;i<=n;i++){
        for(int j=1;i*j<=n;j++)
		dp[j]=(dp[j-1]+p[j]*a[i*j])%mod;
		for(int j=1;i*j<=n;j++)
		if(gcd(i,j)==1)
		b[i*j]=(b[i*j]+dp[min(n/i,n/j)]*p[j]%mod)%mod;
	}
	for(int i=1;i<=n;i++)
	ans^=b[i];
	printf("%lld\n",ans);
	return 0;
}