HDU 4407

题目链接: http://acm.hdu.edu.cn/showproblem.php?pid=4407

题目意思就是 求一个区间[l,r]中与p互素的数的和。 期中有修改的操作。。不过由于修改的操作比较少，可以独立出来考虑，最后通过容斥的思想，实际上最后只用考虑 [1,n]与p的互斥元素数目。这时候还是通过容斥思想，分解质因数后，从质因数的数目来判断是加上这一段还是减去这一段。

#include <cstdio>
#include <cstdlib>
#include <set>
#include <map>
#include <iostream>
using namespace std;

const int N = 411111;
#define ll long long


bool is_prime[N];
int prm[N],k;
set<int> se;
map<int,int> M;
int n,m;

int get_prime(int n){
	k=0;
	for(int i = 2;i < n;i++)is_prime[i]=true;
	for(int i = 4;i < n;i+=2)is_prime[i]=false;
	prm[k++]=2;
	for(int i = 3;i < n;i+=2){
		if(is_prime[i]){
			prm[k++]=i;
			for(int j = i+i;j < n;j+=i)
				is_prime[j]=false;
		}
	}
	return k;
}
int gcd(int a,int b){
	if(b==0) return a;
	return gcd(b,a%b);
}
ll sum(ll i){
	return i*(i+1)/2;
}
int p_n,p_zhi[N],cun[N];
ll tmp;
int q_n;
void dfs(int l,int r,int top){
	if(r == top){
		int num=1;
		for(int i = 1;i < r;i++){
			num *= cun[i];
		}
		tmp += sum(q_n/num)*num;
	}else{
		for(int i = l;i <= p_n;i++){
			cun[r] = p_zhi[i];
			dfs(i+1,r+1,top);
		}
	}
}
ll single(int n,int p){
	if(n<1)return 0;
	int tp = n;
	q_n = n;
	p_n=0;
	for(int i = 0;prm[i]<=p;i++){
		if(p%prm[i] == 0){
			p_zhi[++p_n]=prm[i];
			do{	
				p/=prm[i];
			}while(p%prm[i]==0);
		}
	}
	p = tp;
	
	ll ret = sum(n);
	for(int i = 1;i <= p_n;i++){ //选择 i+1个质数
		tmp=0;
		dfs(1,1,i+1);
		if(i&1)
			ret -= tmp;
		else ret += tmp;
	}
	return ret;
}

ll solve(int l,int r,int p){
	ll ans = single(r,p)-single(l-1,p);	

	set<int>::iterator x;
	for(x = se.begin(); x!=se.end();x++){
		int num = M[*x];
		if( l <= *x && *x <= r ){
			if(gcd(p,*x)==1)
				ans -= *x;
			if(gcd(p,num)==1)
				ans += num;
		}
	}
	return ans;
}

int main(){
	int t;
	get_prime(N);
	scanf("%d",&t);
	
	while(t--){
		se.clear();
		M.clear();
		scanf("%d %d",&n,&m);
		for(int i = 0,a,b,c,d;i < m;i++){
			scanf("%d %d %d",&a,&b,&c);
			if(a == 1){
				scanf("%d",&d);
				cout << solve(b,c,d) << endl;
			}else if(a == 2){
				se.insert(b);
				M[b] = c;
			}
		}
	}
	return 0;
}