题目描述
华华刚刚帮月月完成了作业。为了展示自己的学习水平之高超,华华还给月月出了一道类似的题:
A n s = ⊕ i = 1 N ( i N m o d ( 1 0 9 + 7 ) ) Ans=\oplus_{i=1}^N(i^N\mod(10^9+7)) Ans=⊕i=1N(iNmod(109+7))
⊕ \oplus ⊕符号表示异或和,详见样例解释。
虽然月月写了个程序暴力的算出了答案,但是为了确保自己的答案没有错,希望你写个程序帮她验证一下。
输入描述:
输入一个正整数 N N N。
输出描述:
输出答案 A n s Ans Ans。
示例1
输入
3
输出
18
说明
N = 3 N=3 N=3时, 1 3 = 1 1^3=1 13=1, 2 3 = 8 2^3=8 23=8, 3 3 = 27 3^3=27 33=27,异或和为 18 18 18。
示例2
输入
2005117
输出
863466972
备注:
1 ≤ N ≤ 1.3 × 1 0 7 1≤N≤1.3×10^7 1≤N≤1.3×107
解决思路:
提示:幂函数是完全积性函数
已知:求 ⊕ i = 1 N ( i N m o d ( 1 0 9 + 7 ) ) \oplus_{i=1}^N(i^N\mod(10^9+7)) ⊕i=1N(iNmod(109+7))
求解:因为 N N N比较大,所以快速幂的 O ( n l o g n ) O(nlog_n) O(nlogn)的算法就不能通过。
由于幂函数是完全积性函数,可以通过欧拉筛线性筛出幂函数,求异或和即可。
代码
#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;
}