NOI2004 能量采集

CodeVS1937 能量采集

 

2010年NOI全国竞赛

题目描述 Description

栋栋有一块长方形的地，他在地上种了一种能量植物，这种植物可以采集太阳光的能量。在这些植物采集能量后，栋栋再使用一个能量汇集机器把这些植物采集到的能量汇集到一起。

栋栋的植物种得非常整齐，一共有n列，每列有m棵，植物的横竖间距都一样，因此对于每一棵植物，栋栋可以用一个坐标(x, y)来表示，其中x的范围是1至n，表示是在第x列，y的范围是1至m，表示是在第x列的第y棵。

由于能量汇集机器较大，不便移动，栋栋将它放在了一个角上，坐标正好是(0, 0)。

能量汇集机器在汇集的过程中有一定的能量损失。如果一棵植物与能量汇集机器连接而成的线段上有k棵植物，则能 量的损失为2k + 1。例如，当能量汇集机器收集坐标为(2, 4)的植物时，由于连接线段上存在一棵植物(1, 2)，会产生3的能量损失。注意，如果一棵植物与能量汇集机器连接的线段上没有植物，则能量损失为1。现在要计算总的能量损失。

下面给出了一个能量采集的例子，其中n = 5，m = 4，一共有20棵植物，在每棵植物上标明了能量汇集机器收集它的能量时产生的能量损失。

 

在这个例子中，总共产生了36的能量损失。

输入描述 Input Description

输入文件energy.in仅包含一行，为两个整数n和m。

输出描述 Output Description

输出文件energy.out仅包含一个整数，表示总共产生的能量损失。

样例输入 Sample Input

【样例输入1】

5 4

【样例输入2】

3 4

样例输出 Sample Output

【样例输出1】

36

【样例输出2】

20

数据范围及提示 Data Size & Hint

对于10%的数据：1 ≤ n, m ≤ 10；

对于50%的数据：1 ≤ n, m ≤ 100；

对于80%的数据：1 ≤ n, m ≤ 1000；

对于90%的数据：1 ≤ n, m ≤ 10,000；

对于100%的数据：1 ≤ n, m ≤ 100,000。

测试通过 Accepted

总耗时:  12 ms

0 /  0 数据通过测试.

运行结果

测试点#energy1.in  结果:AC    内存使用量:  256kB     时间使用量:  1ms     
测试点#energy10.in  结果:AC    内存使用量:  1004kB     时间使用量:  3ms     
测试点#energy2.in  结果:AC    内存使用量:  256kB     时间使用量:  1ms     
测试点#energy3.in  结果:AC    内存使用量:  256kB     时间使用量:  1ms     
测试点#energy4.in  结果:AC    内存使用量:  256kB     时间使用量:  1ms     
测试点#energy5.in  结果:AC    内存使用量:  256kB     时间使用量:  1ms     
测试点#energy6.in  结果:AC    内存使用量:  256kB     时间使用量:  1ms     
测试点#energy7.in  结果:AC    内存使用量:  256kB     时间使用量:  1ms     
测试点#energy8.in  结果:AC    内存使用量:  256kB     时间使用量:  1ms     
测试点#energy9.in  结果:AC    内存使用量:  256kB     时间使用量:  1ms     

Analysis

先说一个小结论：

对于一个直角坐标系上的点(x,y)，由(0,0)到(x,y)的连线上，横纵坐标都是正整数的点有gcd(x,y)个

证明：http://blog.csdn.net/fsahfgsadhsakndas/article/details/51346126

这道题中每株植物的能量为2k+1，因为不包含自己所以k=gcd(x,y)-1，

于是ans=∑[ gcd(i,j)*2-1 ]

转化一下，我们枚举i=gcd(x,y)，

记f[i]为“gcd(x,y)=i的点有多少个”，即“有多少对(x,y)使得gcd(x,y)=i”

当x和y的范围相同时，我们可以用欧拉函数算出这个值

然而现在x和y的范围分别为n和m，欧拉函数失效了

这种方法是在hzwer上看到的：

f[i]=(n/i)(m/i)-f[ki] ( k<=min(n,m) )

应用了容斥

其原理很简单，(n/i)(m/i)表示有多少对(x,y)使得d|gcd(x,y)，那么这里面就包含了gcd=d,2d,3d....[n/d]d的数对

我们只需要gcd=d的，只需把那些2d,3d的减去就行了

当i=min(n,m)时，可以发现f[min(n,m)]是正确的，从f[min(n,m)]开始先前推就能得到所有的f[i]

很神奇吧？

这个方程很好懂，但是较难懂的就是hzw说它的时间复杂度是O(nlogn)，为此我写了个小程序，

计算(1+1/2+1/3+...1/n)，将其与log2(n)作比，发现当n=10的时候(1+1/2+1/3+...1/10)/logn=1.13，

而当n=10^8时，该比值为1.39，我又输出了中间的值，发现该比值是单调递增的，

于是乎，我们可以说 在1到10^8内，近似地认为(1+1/2+1/3+...1/n)=logn

计算该方程的时间复杂度为O(nlogn)

回到题目：ans=f[x](2*x-1)，复杂度是O(n)的

Code

//CodeVS1937 能量采集 NOI2010
#include <cstdio>
#include <algorithm>
#define maxn 1000010
#define ll long long
using namespace std;
ll N, M, ans, f[maxn], m;
int main()
{
	ll i, j, x;
	scanf("%ld%ld",&N,&M);
	m=min(N,M);
	for(i=m;i;i--)
	{
		f[i]=(N/i)*(M/i);
		for(j=(i<<1);j<=m;j+=i)f[i]-=f[j];
	}
	for(i=1;i<=m;i++)ans+=f[i]*(2*i-1);
	printf("%lld\n",ans);
	return 0;
}