HDU 4407

http://acm.hdu.edu.cn/showproblem.php?pid=4407

把修改和询问分成两部分解决

询问求区间内与p不互素的和,和求个数一样,用容斥原理解决,只不过做容斥的时候把每一段的个数改成每一段的和,这个求和的方式我一下写搓了,导致这道题坑了很久

修改用map记录,每次扫一遍即可,原数和c互素就减掉,修改完的数和c互素就加上去,logn*m^2的复杂度

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <map>
#include <cmath>

using namespace std;

typedef __int64 ll;

ll gcd(ll a,ll b){
    return b==0?a:gcd(b,a%b);
}

ll sum(ll l,ll r,ll k){
    ll num=r/k-(l-1)/k;
    ll front;
    if(l%k==0)front=l;
    else front=(k-l%k)+l;
    ll rear=r-r%k;
    return (front+rear)*num/2;
}

ll gao(ll l,ll r,ll n){
    vector <ll> v;
    for(ll i=2;i*i<=n;i++){
        if(n%i==0){
            v.push_back(i);
            while(n%i==0)n/=i;
        }
    }
    if(n>1)v.push_back(n);
    int m=v.size();
    ll res=0; 
    for(int i=1;i<(1<<m);i++){
        int cnt=0;
        ll val=1;
        for(int j=0;j<m;j++){
            if(i&(1<<j)){
                cnt++;
                val*=v[j];
            }
        }
        if(cnt&1)res+=sum(l,r,val);
        else res-=sum(l,r,val);
    }
    return (l+r)*(r-l+1)/2-res;
}

map <ll,ll> mp;
map <ll,ll>::iterator it;

ll gan(ll x,ll y,ll p){
    ll res=gao(x,y,p);
    for(it=mp.begin();it!=mp.end();it++){
        if(it->first>y || it->first<x)continue;
        if(gcd(it->first,p)==1)res-=it->first;
        if(gcd(it->second,p)==1)res+=it->second;
    }
    return res;
}

int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        int n,m;
        scanf("%d%d",&n,&m);
        mp.clear();
        while(m--){
            int op;
            scanf("%d",&op);
            ll x,y,p;
            if(op==1){
                scanf("%I64d%I64d%I64d",&x,&y,&p);
                if(x>y)swap(x,y);
                printf("%I64d
",gan(x,y,p));
            }
            else{
                scanf("%I64d%I64d",&x,&p);
                mp[x]=p;
            }
        }
    }
    return 0;
}
View Code
原文地址:https://www.cnblogs.com/xiaohongmao/p/4097198.html