BZOJ 3994: [SDOI2015]约数个数和 莫比乌斯反演

3994: [SDOI2015]约数个数和

Time Limit: 20 Sec  Memory Limit: 128 MB
Submit: 1052  Solved: 720
[Submit][Status][Discuss]

Description

 设d(x)为x的约数个数，给定N、M，求  

Input

输入文件包含多组测试数据。

第一行，一个整数T，表示测试数据的组数。

接下来的T行，每行两个整数N、M。

Output

 T行，每行一个整数，表示你所求的答案。

Sample Input

2
7 4
5 6

Sample Output

110
121

HINT

 1<=N, M<=50000

1<=T<=50000

感觉不是很好做啊啊啊

苦思冥想最后GG。。。

去翻题解

这题解写的真是很不错的

看完之后独立推一推更是很不错的

真的很不错的

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>
using namespace std;

typedef long long ll;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return x*f;
}
void print(ll x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=50100;

int prime[N],mobius[N],f[N],M[N],cnt;

bool book[N];

void initial()
{
	register int i,j;
	mobius[1]=1;
	for(i=2;i<N;++i)
	{
		if(!book[i]){prime[++cnt]=i;mobius[i]=-1;}
		for(j=1;j<=cnt&&prime[j]*i<N;++j)
		{
			book[i*prime[j]]=1;
			mobius[prime[j]*i]=-mobius[i];
			if(i%prime[j]==0)
			{
				mobius[i*prime[j]]=0;
				break;
			}
		}
	}
	for(i=1;i<N;++i)for(j=i;j<N;j+=i)f[j]++;
	for(i=1;i<N;++i)f[i]+=f[i-1];
	for(i=1;i<N;++i)M[i]=mobius[i]+M[i-1];
}

inline void solve(int n,int m)
{
	register int i,pos;ll res=0;
	if(n>m)swap(n,m);
	for(i=1;i<=n;i=pos+1)
	{
		pos=min(n/(n/i),m/(m/i));
		res+=1ll*(M[pos]-M[i-1])*f[n/i]*f[m/i];
	}
	print(res);puts("");
}

int main()
{
	initial();int T=read();
	register int x,y;
	while(T--){x=read();y=read();solve(x,y);}
	return 0;
}
/*
2
7 4
5 6

110
121
*/