[CF542D]Superhero's Job

[CF542D]Superhero's Job

题目大意：

定义函数

\[J(x) = \sum_{\substack{1 \leq k \leq x \\ k \mid x \\ \gcd \left( k, x / k \right) = 1}} k
\]

给定\(n(n\le10^{12})\)，求方程\(J(x)=n\)的解的个数。

思路：

对于\(\gcd(a,b)=1\)，\(J(ab)=J(a)J(b)\)。

对于\(p\in\mathbb{P}\)，\(J(p^k)=p^k+1\)。

\(10^{12}\)内，约数最多的数\(\tau(963761198400)=6720\)。

DP：\(f[i]\)表示组成第\(i\)种有用的约数的方案数。

源代码：

#include<map>
#include<cstdio>
#include<cctype>
#include<vector>
#include<algorithm>
typedef long long int64;
inline int64 getint() {
	register char ch;
	while(!isdigit(ch=getchar()));
	register int64 x=ch^'0';
	while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
	return x;
}
const int D=6721;
std::vector<int64> p;
std::map<int64,int> map;
int64 d[D],f[D];
int main() {
	const int64 n=getint();
	for(register int64 i=1;i*i<=n;i++) {
		if(n%i) continue;
		d[++d[0]]=i;
		if(i*i!=n) d[++d[0]]=n/i;
	}
	std::sort(&d[1],&d[d[0]]+1);
	for(register int i=1;i<=d[0];i++) {
		map[d[i]]=i;
		int64 x=d[i]-1;
		for(register int64 j=2;j*j<=x;j++) {
			if(x%j) continue;
			while(x%j==0) x/=j;
			if(x==1) p.push_back(j);
		}
		if(x>1) p.push_back(x);
	}
	std::sort(p.begin(),p.end());
	p.resize(std::unique(p.begin(),p.end())-p.begin());
	const int m=p.size();
	f[1]=1;
	for(register int i=0;i<m;i++) {
		for(register int j=d[0];j;j--) {
			int64 t=p[i];
			while(t<d[j]) {
				if(d[j]%(t+1)==0) {
					f[j]+=f[map[d[j]/(t+1)]];
				}
				t*=p[i];
			}
		}
	}
	printf("%lld\n",f[d[0]]);
	return 0;
}