bzoj 2005: [Noi2010]能量采集

题目描述

给你 $n,m\in {\mathbb{N}}^{\ast }$，求$$\sum _{i=1}^n\sum _{j=1}^m(2\gcd (i,j)-1)$$

Solution

设$$\begin{array}{rl}f& =\sum _{i=1}^n\sum _{j=1}^m\gcd (i,j)\\ & =\sum _{i=1}^n\sum _{j=1}^m\sum _{d|\gcd (i,j)}\phi (d)\\ & =\sum _{d=1}^{\min (n,m)}\phi (d)(\sum _{d|i}^{i\le n}1)(\sum _{d|j}^{j\le m}1)\\ & =\sum _{d=1}^{\min (n,m)}\lfloor \frac{n}{d}\rfloor \lfloor \frac{m}{d}\rfloor \phi (d)\end{array}$$
则 原式 $=2f-nm$。

#include<cstdio>
#include<cstdlib>
#include<cstring>

#define reg register
const int MAXN=100000;
typedef long long ll;
#define int long long

bool vis[MAXN+10];
int p[MAXN+10];
int phi[MAXN+10];
int len=0;
int n,m;
ll ans=0;

void init(){
	memset(vis,1,sizeof(vis));phi[1]=1;
	for(reg int i=2;i<=MAXN;++i){
		if(vis[i]){
			p[++len]=i;
			phi[i]=i-1;
		}
		for(reg int j=1;j<=len&&i*p[j]<=MAXN;++j){
			vis[i*p[j]]=0;
			if(i%p[j])
				phi[i*p[j]]=phi[i]*(p[j]-1);
			else{
				phi[i*p[j]]=phi[i]*p[j];
				break;
			}
		}
	}
}
int min(int x,int y){
	if(x<y) return x;
	return y;
}
signed main(){
	init();
	scanf("%d%d",&n,&m);
	for(reg int i=1;i<=min(n,m);++i)
		ans+=(n/i)*(m/i)*phi[i];
	printf("%lld",ans+ans-n*m);
}