题目:http://acm.hdu.edu.cn/showproblem.php?pid=1698
数据:case,n,q,q行x,y,z。在长度为n的hook上进行q次区间更新,把它们的价值改变。最后统计总的价值。x,y,z分别表示把[x,y]内的hook变成价值为z的hook。
开始写它时,对于样例程序运行的结果总是20,我就纳闷了,怎么改都不对,在反复研究中发现了这样一个问题(bug源于update函数):
void update(int a,int b,int root,int tg){ if(a==tree[root].l&&b==tree[root].r){ tree[root].tag=tg; tree[root].val=(tree[root].r-tree[root].l+1)*tree[root].tag; return ; } int m=tree[root].mid(); if(a>m) update(a,b,2*root+1,tg); else if(b<=m) update(a,b,2*root,tg); else { update(a,m,2*root,tg); update(m+1,b,2*root+1,tg); } tree[root].val=tree[2*root].val+tree[2*root+1].val; }
#include <iostream> #include <cstdio> using namespace std; const int maxn=1e5+5; #define l_t 2*root #define r_t 2*root+1 struct node{ int l,r,val,tag; //tag:1 2 3 -1 int mid(){ return (l+r)/2; } }tree[maxn<<2]; void update(int a,int b,int root,int tg){ if(a==tree[root].l&&b==tree[root].r){ tree[root].tag=tg; tree[root].val=(tree[root].r-tree[root].l+1)*tg; return ; } if(tree[root].tag!=-1){ tree[l_t].tag=tree[root].tag; tree[l_t].val=tree[l_t].tag*(tree[l_t].r-tree[l_t].l+1); tree[r_t].tag=tree[root].tag; tree[r_t].val=tree[r_t].tag*(tree[r_t].r-tree[r_t].l+1); tree[root].tag=-1; } int m=tree[root].mid(); if(a>m) update(a,b,r_t,tg); else if(b<=m) update(a,b,l_t,tg); else { update(a,m,l_t,tg); update(m+1,b,r_t,tg); } tree[root].val=tree[l_t].val+tree[r_t].val; } void build(int l,int r,int root){ tree[root].l=l; tree[root].r=r; tree[root].tag=-1; if(r==l){ tree[root].val=1; return ; } int mid=(l+r)/2; build(l,mid,l_t); build(mid+1,r,r_t); tree[root].val=tree[l_t].val+tree[r_t].val; } int main() { //freopen("cin.txt","r",stdin); int ca; cin>>ca; for(int k=1;k<=ca;k++){ int n,q; scanf("%d",&n); build(1,n,1); scanf("%d",&q); for(int i=0;i<q;i++){ int x,y,z; scanf("%d%d%d",&x,&y,&z); update(x,y,1,z); } printf("Case %d: The total value of the hook is %d.\n",k,tree[1].val); } return 0; }