POJ 1091 跳蚤(容斥原理)

这道题的目标是寻找满足条件的 a1、a2、…、an,满足a1*x1+a2*x2+...+an*xn+M*xn+1=1
也就是gcd(a1,a2,...,an,m)=1
利用容斥原理就可以找出满足该条件的a1、a2、…、an。
详见代码:

#define _CRT_SECURE_NO_WARNINGS
#include<cstdio>
#include<iostream>
#include<vector>
#include<cmath>
using namespace std;
typedef long long ll;
int n, m;
vector<int> x;
void getprime(int m)
{
	//得到m的质因子
	x.clear();
	for (int i = 2; i*i <= m; ++i)
	if (m%i == 0)
	{
		x.push_back(i);
		while (m%i == 0) m /= i;
	}
	if (m != 1) x.push_back(m);
}
int main(){
	while (scanf("%d%d", &n, &m)==2){
		getprime(m);//分解m
		ll ans = pow(m, n);//ans为m^n
		for (int i = 1; i< (1 << x.size()) ; ++i){//m素因子个数为t,则i<2^t,即m素因子集合的子集个数
			int k = 1, cnt = 0;
			for (int j = 0; j<x.size(); ++j)
				if (i&(1 << j)) 
					k *= x[j], ++cnt;//k为取出的这个子集的值的积，cnt表示子集中元素的个数
			if (cnt & 1) //等价于cnt%2==1,取出个数为奇数时减，取出个数为偶数时加
				ans -= pow(m / k, n);
			else 
				ans += pow(m / k, n);
		}
		printf("%lld\n", ans);
	}
	return 0;
}