能量采集
题目描述
栋栋有一块长方形的地,他在地上种了一种能量植物,这种植物可以采集太阳光的能量。在这些植物采集能量后,栋栋再使用一个能量汇集机器把这些植物采集到的能量汇集到一起。
栋栋的植物种得非常整齐,一共有 nn 列,每列有 mm 棵,植物的横竖间距都一样,因此对于每一棵植物,栋栋可以用一个坐标 (x, y)(x,y) 来表示,其中 xx 的范围是 11 至 nn,yy 的范围是 11 至 mm,表示是在第 xx 列的第 yy 棵。
由于能量汇集机器较大,不便移动,栋栋将它放在了一个角上,坐标正好是 (0,0)。
能量汇集机器在汇集的过程中有一定的能量损失。如果一棵植物与能量汇集机器连接而成的线段上有 k 棵植物,则能量的损失为 2k + 1。例如,当能量汇集机器收集坐标为 (2, 4) 的植物时,由于连接线段上存在一棵植物 (1, 2),会产生 3 的能量损失。注意,如果一棵植物与能量汇集机器连接的线段上没有植物,则能量损失为 1。现在要计算总的能量损失。
下面给出了一个能量采集的例子,其中 n = 5n=5,m = 4m=4,一共有 2020 棵植物,在每棵植物上标明了能量汇集机器收集它的能量时产生的能量损失。
在这个例子中,总共产生了 36 的能量损失。
输入格式
一行两个整数n,m。
输出格式
仅包含一个整数,表示总共包含的能量损失。
输入样例
#1
5 4
#2
36
输出样例
#1
36
#2
20
解题思路
求∑∑2gcd(i,j)-1,1≤i≤n≤10^ 5,1≤j≤m≤10^5
首先将-1提出来,一共减去了n*M,这个2的系数可以简化
那么关键就在于求∑∑gcd(i,j),有一条性质是∑[d|n]φ(d)=n
代入gcd(i,j)就可以得到∑∑∑[d|i,d|j]φ(d),枚举d可以得到
原式=∑φ(d)∑[id≤n]∑[jd≤m]ij=∑φ(d)(n/d)(m/d)
用线性筛或埃式筛法求出欧拉函数即可
代码
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<cstring>
using namespace std;
long long n,m,a[1000010],p[100010],t,k=0;
bool b[100100];
void pe(int x)
{
b[1]=1;
p[1]=1;
for(int i=2;i<=x;i++)
{
if(b[i]==0)
a[t++]=i,p[i]=i-1;
for(int j=0;j<t&&i*a[j]<=x;j++)
{
b[i*a[j]]=1;
p[i*a[j]]=p[i]*p[a[j]];
if(i%a[j]==0)
{
p[i*a[j]]=p[i]*a[j];
break;
}
}
}
return;
}
int main()
{
cin>>n>>m;
if(n<m) pe(n);
else pe(m);
for(int i=1;i<=min(n,m);i++)
k+=(n/i)*(m/i)*p[i];
cout<<(k<<1)-n*m;
}