牛客——华华给月月出题（快速幂，欧拉筛）

华华给月月出题

---

题目描述

华华刚刚帮月月完成了作业。为了展示自己的学习水平之高超，华华还给月月出了一道类似的题：
$Ans={\oplus }_{i=1}^N(i^N\mathrm{m}\mathrm{o}\mathrm{d}(10^9+7))$
$\oplus$符号表示异或和，详见样例解释。
虽然月月写了个程序暴力的算出了答案，但是为了确保自己的答案没有错，希望你写个程序帮她验证一下。

输入描述:

输入一个正整数$N$。

输出描述:

输出答案$Ans$。

示例1

输入

3

输出

18

说明

$N=3$时，$1^3=1$，$2^3=8$，$3^3=27$，异或和为$18$。

示例2

输入

2005117

输出

863466972

备注:

$1\le N\le 1.3\times 10^7$

解决思路：

提示：幂函数是完全积性函数

已知：求${\oplus }_{i=1}^N(i^N\mathrm{m}\mathrm{o}\mathrm{d}(10^9+7))$
求解：因为$N$比较大，所以快速幂的$O(nlo{g}_n)$的算法就不能通过。
由于幂函数是完全积性函数，可以通过欧拉筛线性筛出幂函数，求异或和即可。

代码

#include<map>
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;

typedef long long ll;
const int N = 2e7 + 10;
const int mod = 1e9 + 7;

int cnt;
bool vis[N];
int prime[N];
ll pw[N];

ll qpow(ll a, ll b, ll p) {
	ll res = 1;
	while (b) {
		if (b & 1) {
			res = res * a % p;
		}
		a = a * a % p;
		b >>= 1;
	}
	return res;
}
void init(const int n) {
	for (int i = 2; i <= n; i++) {
		if (!vis[i]) {
			prime[++cnt] = i;
			pw[i] = qpow(i, n, mod);
		}
		for (int j = 1; j <= cnt && i <= n / prime[j]; j++) {
			vis[i * prime[j]] = 1;
			pw[i * prime[j]] = pw[i] * pw[prime[j]] % mod;
			if (i % prime[j] == 0) break;
		}
	}
}

int main() {

	int n; cin >> n;

	init(n);

	ll ans = 1;
	for (int i = 2; i <= n; i++) {
		ans ^= pw[i];
	}

	cout << ans << endl;
	return 0;
}