Luogu P1447 [NOI2010]能量采集

题目大意

给定 $n,m$，求 $$\sum _{i=1}^n\sum _{j=1}^m((2\times i,j)-1)$$

数据范围　$1⩽n,m⩽10^5$

题解

原式 $=2\times \sum _{i=1}^n\sum _{j=1}^m\times (i,j)-nm$
因而要点是求 $$S=\sum _{i=1}^n\sum _{j=1}^m\times (i,j)$$
由
$$\sum _{d\mid n}\phi (d)=n$$

得
$$S=\sum _{i=1}^n\sum _{j=1}^m\sum _{d|(i,j)}\phi (d)$$

更换枚举顺序
$$S=\sum _{i=1}^n\sum _{j=1}^m\sum _{d=1}^{(i,j)}\phi (d)[d|(i,j)]$$

$$S=\sum _{i=1}^n\sum _{j=1}^m\sum _{d=1}^{(i,j)}\phi (d)[d|i][d|j]$$

各回各家，各找各妈。
$$S=\sum _{d=1}^{min(n,m)}\phi (d)\sum _{i=1}^n[d|i]\sum _{j=1}^m[d|j]$$

化简后面两个和式，得：
$$S=\sum _{d=1}^{min(n,m)}\phi (d)\lfloor \frac{n}{d}\rfloor \lfloor \frac{m}{d}\rfloor$$

线性筛便是 $\Theta (min(n,m))$，因而可以不用除法分块，代码记得开long long。

代码

#include<bits/stdc++.h>
using namespace std;
#define maxn 100010
int prime[maxn],phi[maxn];
bool vis[maxn];
void euler_phi(int n){
    memset(phi,0,sizeof phi);
    memset(vis,0,sizeof vis);
    int tot=0;phi[1]=1;
    for(int i=2;i<=n;i++){
        if(!vis[i]){
            prime[++tot]=i;
            phi[i]=i-1;
        }
        for(int j=1;j<=tot&&i*prime[j]<=n;j++){
            vis[i*prime[j]]=1;
            if(i%prime[j]==0){
                phi[i*prime[j]]=phi[i]*prime[j];
                break;
            } phi[i*prime[j]]=phi[i]*(prime[j]-1);
        }
    }
}
int main(){
	int n,m;
	scanf("%d%d",&n,&m);
	int l=min(n,m);
	long long ans=0;
	euler_phi(l);
	for(int i=1;i<=l;i++)
		ans+=(long long)phi[i]*(n/i)*(m/i);
	printf("%lld\n",2*ans-(long long)n*m);
	return 0;
}