小a与黄金街道 牛客网 （欧拉函数）

链接：https://ac.nowcoder.com/acm/contest/317/D
来源：牛客网
 

小a与黄金街道

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 32768K，其他语言65536K
64bit IO Format: %lld

题目描述

小a和小b来到了一条布满了黄金的街道上。它们想要带几块黄金回去，然而这里的城管担心他们拿走的太多，于是要求小a和小b通过做一个游戏来决定最后得到的黄金的数量。
游戏规则是这样的：
假设道路长度为nn米(左端点为00，右端点为nn)，同时给出一个数kk(下面会提到kk的用法)
设小a初始时的黄金数量为AA，小b初始时的黄金数量为BB
小a从11出发走向n−1n−1，小b从n−1n−1出发走向11，两人的速度均为1m/s1m/s
假设某一时刻(必须为整数)小a的位置为xx，小b的位置为yy，若gcd(n,x)=1gcd(n,x)=1且gcd(n,y)=1gcd(n,y)=1，那么小a的黄金数量AA会变为A∗kx(kg)A∗kx(kg)，小b的黄金数量BB会变为B∗ky(kg)B∗ky(kg)
当小a到达n−1n−1时游戏结束
小a想知道在游戏结束时A+BA+B的值
答案对109+7109+7取模

输入描述:

一行四个整数n,k,A,Bn,k,A,B

输出描述:

输出一个整数表示答案

示例1

输入

复制

4 2 1 1

输出

复制

32

说明

初始时A=1,B=1A=1,B=1

第一个时刻如图所示，小a在11，小b在33，满足条件，此时A=1∗21=2,B=1∗23=8A=1∗21=2,B=1∗23=8

 

第二个时刻小a在22，小b在22，不满足条件

 

第三个时刻小a在33，小b在11，满足条件，此时A=2∗23=16,B=8∗21=16A=2∗23=16,B=8∗21=16

此时游戏结束A=2∗23=16,B=8∗21=16A=2∗23=16,B=8∗21=16

A+B=32A+B=32

示例2

输入

复制

5 1 1 1

输出

复制

2

备注:

保证3⩽n⩽108,1⩽A,B,k⩽1013

 

稍微推一下，很显然，就是求1到n - 1之间与n互质的数的和，需要用到欧拉函数，详细讲解在这儿

https://blog.csdn.net/qq_38185591/article/details/87027554

其次还要注意一个地方 a ^ b % mod 如果a很大，我们也需要进行取mod， 所以利用取模的运算规则

a ^ b % mod = (a % mod ) ^ b % mod 即可

代码如下：

#include <bits/stdc++.h>
#include <time.h>
using namespace std;

typedef long long ll;
typedef pair<ll,ll>  P;
//vector<vector<ll> >v;
const ll mod = 1e9 + 7;
const int maxn = 2e6 + 5000;
ll  n,m;
ll a[maxn];
int vis[maxn];
int xx[4] = {1,-1,1,-1};
int yy[4] = {-1,1,1,-1};
priority_queue<int,vector<int>,greater<int> >q;
ll ans ;
ll f[900000] ;
bool is_prim(ll x) {
    for(ll i = 2; i * i <= x; i++) {
        if(x % i == 0) return false;
    }
    return true;
}
ll mul(ll a,ll b) {
    ll res = 1;
    while(b) {
        if(b & 1) res *= a,res %= mod;
        a *= a;
        a %= mod;
        b >>= 1;
    }
    return res;
}
ll euler_phi(ll n) {
    ll m=(ll)sqrt(n+0.5);
    ll ans=n;
    for(ll i=2; i<=m; i++) {
        if(n%i==0) {
            ans=ans/i*(i-1);
            while(n%i==0)
                n/=i;
        }
    }
    if(n>1)
        ans=ans/n*(n-1);
    return ans % mod;
}

ll k,A,B;
int main() {
    while(cin >> n >> k >> A >> B) {
            A %= mod;

            B %= mod;
            k %= mod;
        ll sum = (n - 1) * (n - 2) / 2;
        ll poww;
        if(is_prim(n)) {
             poww = mul(k,sum);
        } else {
            ll l = euler_phi(n) * n / 2  ;
             poww = mul(k,l);
        }
        cout << (A + B) % mod  * poww % mod <<endl;
    }
    return 0;
}