您的位置:首页 > 其它

hdu_5634

2016-02-23 13:42 176 查看
直接用线段树就可以做,

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <set>
#include <queue>
#include <map>
#include <cmath>
#include <vector>
using namespace std;
#define N 500010
#define pi acos(-1.0)
#define inf 0x3f3f3f3f
#define pb(x) push_back((x))
typedef __int64 lll;
typedef unsigned long long ull;
int phi[10000010];
int a[1200010];
lll sum[1200010];
void phi_table(int n){//欧拉打表
phi[1]=1;
for(int i=2;i<=n;i++){
if(!phi[i])
for(int j=i;j<=n;j+=i){
if(!phi[j]) phi[j]=j;
phi[j]=phi[j]/i*(i-1);
}
}
}

void build(int root,int l,int r){//建造线段树
if(l==r){
scanf("%d",&a[root]);
sum[root]=a[root];
return ;
}
int mid=(l+r)>>1;
build(root<<1,l,mid);
build(root<<1|1,mid+1,r);
sum[root]=sum[root<<1]+sum[root<<1|1];
if(a[root<<1]==a[root<<1|1]) a[root]=a[root<<1];
}

void pushdown(int x, int l, int mid, int r){//区间拆分向下传递操作
a[x << 1] = a[x << 1 | 1] = a[x];
sum[x << 1] = (lll)a[x] * (mid - l + 1);
sum[x << 1 | 1] = (lll)a[x] * (r - mid);
a[x] = 0;
}
void gan1(int root,int l,int r,int ll,int rr){//第一个操作
if(l>=ll&&r<=rr){
if(a[root]){
a[root]=phi[a[root]];
sum[root]=(lll)a[root]*(r-l+1);
return;
}
int mid=(r+l)>>1;
gan1(root<<1,l,mid,ll,rr);
gan1(root<<1|1,mid+1,r,ll,rr);
sum[root]=sum[root<<1]+sum[root<<1|1];
if(a[root<<1]==a[root<<1|1]) a[root]=a[root<<1];
else a[root]=0;
return;
}
int mid=(l+r)>>1;
if(a[root]) pushdown(root,l,mid,r);//如果这个区间的值已经统一,那么在进行更改的时候,如果要向下操作,就要将这个统一的区间拆分成两个统一的小区间,以免给以后造成麻烦
if(ll<=mid) gan1(root<<1,l,mid,ll,rr);
if(rr>mid) gan1(root<<1|1,mid+1,r,ll,rr);
sum[root]=sum[root<<1]+sum[root<<1|1];
if(a[root<<1]==a[root<<1|1]) a[root]=a[root<<1];
else a[root]=0;
}

void gan2(int root,int l,int r,int ll,int rr,int x){//第二个操作

if(l>=ll&&r<=rr){
a[root]=x;
sum[root]=(lll)a[root]*(r-l+1);
return ;
}
int mid=(l+r)>>1;
if(a[root]) pushdown(root,l,mid,r);//如果这个区间的值已经统一,那么在进行更改的时候,如果要向下操作,就要将这个统一的区间拆分成两个统一的小区间,以免给以后造成麻烦
if(ll<=mid) gan2(root<<1,l,mid,ll,rr,x);
if(rr>mid) gan2(root<<1|1,mid+1,r,ll,rr,x);
sum[root]=sum[root<<1]+sum[root<<1|1];
if(a[root<<1]==a[root<<1|1]) a[root]=a[root<<1];
else a[root]=0;
}

lll out(int root,int l,int r,int ll,int rr){
if(l>=ll&&r<=rr) return sum[root];
lll ans=0;
int mid=(l+r)>>1;
if(a[root]) return sum[root]/(r-l+1)*(min(r,rr)-max(l,ll)+1);//如果这个区间的所有值都一样,那么直接就可以得出结果。
if(ll<=mid) ans+=out(root<<1,l,mid,ll,rr);
if(rr>mid) ans+=out(root<<1|1,mid+1,r,ll,rr);
return ans;
}

int main(){
int t,l,r,x,n,m;
phi_table(1e7);//欧拉打表
scanf("%d",&t);
while(t--){
memset(sum,0,sizeof(sum));
memset(a,0,sizeof(a));
scanf("%d%d",&n,&m);
build(1,1,n);
int tar;
while(m--){
scanf("%d",&tar);
if(tar==1){
scanf("%d%d",&l,&r);
gan1(1,1,n,l,r);
}
else if(tar==2){
scanf("%d%d%d",&l,&r,&x);
gan2(1,1,n,l,r,x);
}
else{
scanf("%d%d",&l,&r);
printf("%I64d\n",out(1,1,n,l,r));
}
}
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  线段树