【容斥】【CF1900D】Small GCD

题目简述

求解下列式子：
$$\sum _{i=1}^n\sum _{j=1}^{i-1}(a_i,a_j)\times (n-i)$$

$$n\le 8e4，a_i\le 1e5$$

思路1

写反演写魔怔了，所以上来直接一个反演
最后的式子是这样的
$$\sum _{i=1}^n(n-i)\sum _{d|a_i}\varphi (d)cn{t}_{d,i}$$
其中 $$cn{t}_{d,i}=\sum _{j=1}^{i-1}[d|a_j]$$
显然 $i$ 那维可以省略
预处理所有 $a_i$ 的因数（调和级数），在求答案过程中顺便求 $cnt$
时间复杂度$O(m+mlogm+128n)$

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=1e5+7,inf=2e12;
int phi[N];
bool bz[N];
vector<int> p,D[N];
void init(int n)
{
	phi[1]=1;
	for(int i=2; i<=n; i++)
	{
		if(!bz[i])
		{
			p.push_back(i);
			phi[i]=i-1;
		}
		for(auto x:p)
		{
			if(x*i>n) break;
			bz[x*i]=1;
			if(i%x==0)
			{
				phi[x*i]=phi[i]*x;
				break;
			}
			else
				phi[x*i]=phi[i]*(x-1);
		}
	}
	for(int i=1; i<=n; i++)
	{
		for(int j=i; j<=n; j+=i)
		{
			D[j].push_back(i);
		}
	}
}
void O_o()
{
	int n;
	cin>>n;
	vector<int> a(n+1),cnt(N);
	for(int i=1; i<=n; i++)
	{
		cin>>a[i];
	}
	sort(a.begin()+1,a.end());
	int ans=0;
	for(int i=1; i<=n; i++)
	{
		for(auto d:D[a[i]])
		{
			ans+=(n-i)*phi[d]*cnt[d];
			cnt[d]++;
		}
	}
	cout<<ans<<"\n";
}
signed main()
{
	ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
	cout<<fixed<<setprecision(2);
	int T=1;
	init(100001);
	cin>>T;
	while(T--)
	{
		O_o();
	}
}

思路2（正常做法）

别看到 gcd 就反演，这题明显不用反演。
令 $f_d$ 为 $gcd$ 为 $d$ 的 $(a_i,a_j)$ 乘上 $(n-i)$ 之和，即
$$f_d=\sum _{i=1}^n\sum _{j=1}^{i-1}[(i,j)=d]\times (n-i)$$

直接求会有重复的问题，不好求，所以很自然的想到容斥。
令 $g_d$ 为 $gcd$ 为 $k\times d$ 的 $(a_i,a_j)$ 乘上 $(n-i)$ 之和，即
$$g_d=\sum _{i=1}^n\sum _{j=1}^{i-1}[d|(i,j)]\times (n-i)$$
那么
$$f_d=g_d-\sum _{i=d\times 2}{f}_i$$
答案就是
$$ans=\sum _{i=1}i\times f_i$$

同样需要预处理所有 $a_i$ 的因数，但不需要线性筛，时间复杂度是一样的。

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=1e5+7,inf=2e12;
vector<int> p,D[N];
void init(int n)
{
	for(int i=1; i<=n; i++)
	{
		for(int j=i; j<=n; j+=i)
		{
			D[j].push_back(i);
		}
	}
}
void O_o()
{
	int n;
	cin>>n;
	vector<int> a(n+1),cnt(N),dp(N);//dp[i]: gcd 为 i 的倍数的贡献 
	for(int i=1; i<=n; i++)
	{
		cin>>a[i];
	}
	sort(a.begin()+1,a.end());
	int ans=0;
	for(int i=1; i<=n; i++)
	{
		for(auto d:D[a[i]])
		{
			dp[d]+=cnt[d]*(n-i);
			cnt[d]++;
		}
	}
	//通过容斥，把 dp[i] 变成 gcd 为 i 的贡献 
	for(int i=N-6; i>=1; i--)
	{
		for(int j=i*2; j<=N-6; j+=i)
		{
			dp[i]-=dp[j];
		}
		ans+=i*dp[i];
	}
	cout<<ans<<"\n";
}
signed main()
{
	ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
	cout<<fixed<<setprecision(2);
	int T=1;
	init(100001);
	cin>>T;
	while(T--)
	{
		O_o();
	}
}