1. 程式人生 > >【ACM-ICPC 2018 沈陽賽區網絡預賽 G】Spare Tire

【ACM-ICPC 2018 沈陽賽區網絡預賽 G】Spare Tire

std += anti 數字 n+2 num names return include

【鏈接】 我是鏈接,點我呀:)
【題意】


在這裏輸入題意

【題解】


讓你求出1..n中和m互質的位置i.
讓你輸出∑ai
這個ai可以oeis一波。
發現是ai = i(i+1)
1..n中和m互質的數字的個數之前有做過一題。
然後發現是逆著做的。
刪掉不互質的。剩下的就是互質的了。
是用容斥原理搞的。
其中有ans+=n/temp和ass-=n/temp
表示的是加上temp,2
temp,3temp...ttemp這些數字
以及減去temp,2temp,3temp...t*temp這些數字
放在這一題的話其實就是
ans+=a[temp]+a[2*temp]...+a[t*temp]

然後發現是等差的下標。那麽就去推推公式吧?
然後發現真的能推出來
在get_ans2裏。自己看吧。
那麽ans就是下標和m不互質的ai加起來的和
然後ans=$∑_1^na_i$-ans.

【代碼】

#include <bits/stdc++.h>
#define LL long long
#define rep1(i,a,b) for (int i = a;i <= b;i++)
#define rep2(i,a,b) for (int i = a;i >= b;i--)
#define all(x) x.begin(),x.end()
#define pb push_back
#define lson l,mid,rt<<1
#define ri(x) scanf("%d",&x)
#define rl(x) scanf("%lld",&x)
#define rs(x) scanf("%s",x)
#define rson mid+1,r,rt<<1|1
using namespace std;

const int N = 10000;
const LL MOD = 1e9+7;

LL n,m,pri[N+10];
LL sixni;
int num;

LL Pow(LL x,LL y){ //求x^y
    LL a = 1;
    while (y){
        if (y&1) a = (a*x)%MOD;
        x=(x*x)%MOD;
        y>>=1;
    }
    return a;
}

LL get_ans2(LL t,LL x){
    LL temp1 = t*(t+1)%MOD*(2*t+1)%MOD;
    temp1 = temp1*sixni%MOD*x%MOD*x%MOD;
    temp1 = temp1 + (1+t)*t/2%MOD*x%MOD;
    return temp1;
}

LL get_ans(LL x)
{
    LL ans = 0;
    rep1(i,1,(1<<num)-1){
        LL y = 1,f = 0;
        rep1(j,1,num)
            if (i & (1<<(j-1))){
                y = y*pri[j];
                f++;
            }
        if (f&1)
            ans += get_ans2(x/y,y);
        else
            ans -= get_ans2(x/y,y);
    }

    ans = 2*(n+2)*(n+1)%MOD*n%MOD*sixni%MOD-ans;
    ans = ans%MOD;
    ans=(ans+MOD)%MOD;
    return ans;
}

void _init(){
    num = 0;
    for (LL i = 2;i*i<=m;i++)
        if ((m%i)==0){
            pri[++num] = i;
            while ((m%i)==0) m/=i;
        }
    if (m > 1) pri[++num] = m;
}

int main()
{
    sixni=Pow(6,MOD-2);
    while (~scanf("%lld%lld",&n,&m)){
        _init();
        printf("%lld\n",get_ans(n));
    }
    return 0;
}

【ACM-ICPC 2018 沈陽賽區網絡預賽 G】Spare Tire