Description
Input
Output
这题是区间修改,单点查询,可以用线段树或树状数组处理。
线段树做的核心代码:
int sum[N],add[N],ans[N]; void PushUp(int rt){ sum[rt]=sum[rt<<1]+sum[rt<<1|1]; } void PushDown(int rt,int m){ if (add[rt]!=0){ sum[rt<<1]+=add[rt]*(m-(m>>1)); sum[rt<<1|1]+=add[rt]*(m>>1); add[rt<<1]+=add[rt]; add[rt<<1|1]+=add[rt]; add[rt]=0; } } void update(int L,int R,int c,int l,int r,int rt){ if (L<=l&&R>=r){ add[rt]+=c; sum[rt]+=(l-r+1)*c; return; } int m=(l+r)>>1; PushDown(rt,r-l+1); if (L<=m) update(L,R,c,l,m,rt<<1); if (R>m) update(L,R,c,m+1,r,rt<<1|1); PushUp(rt); } void solve(int l,int r,int rt){ if (l==r){ ans[l]=sum[rt]; return; } int m=(l+r)>>1; PushDown(rt,r-l+1); solve(l,m,rt<<1); solve(m+1,r,rt<<1|1); } void Run(){ int n,a,b; while (~scanf("%d",&n)&&n){ memset(sum,0,sizeof(sum)); memset(add,0,sizeof(add)); for (int i=1;i<=n;i++){ scanf("%d %d",&a,&b); update(a,b,1,1,n,1); } solve(1,n,1); printf("%d",ans[1]); for (int i=2;i<=n;i++) printf(" %d",ans[i]); puts(""); } }
树状数组做的核心代码:
int num[N]; int tree[N]; int lowbit(int x){ return x&(-x); } void update(int x,int add){ while (x>0){ tree[x]+=add; x-=lowbit(x); } } int query(int x,int n){ int sum=0; while (x<=n){ sum+=tree[x]; x+=lowbit(x); } return sum; } void Run(int n){ memset(tree,0,sizeof(tree)); int a,b; for (int i=1;i<=n;i++){ scanf("%d %d",&a,&b); update(a-1,-1); update(b,1); } printf("%d",query(1,n)); for (int i=2;i<=n;i++) printf(" %d",query(i,n)); puts(""); }
原文:http://www.cnblogs.com/micrari/p/3523291.html