hihoCoder #1080 : 更为复杂的买卖房屋姿势 (线段树，多tag)

 

题意：

　　有编号为0~n的n+1个房屋，给出每个房屋的起始价格，随后给出m种修改，每次修改都要进行输出所有房屋的价格总和。修改有两种方式：（1）政府调控，编号L~R全置为同一价格（0）房屋自行涨跌，编号L~R的房屋价格全部涨/跌一部分。

 

思路：

　　只需要一个tag+浮动的价格就可以解决这个问题了。当tag=1时，表示需要对该节点的孩子进行修改。进行何种修改？看浮动价格flo是否为0，若为0，表示需要设置为同一价格，否则，表示价格需要涨跌。当先进行set后进行add时，按照set的方式下放即可；当先进行add后进行set时，set直接覆盖掉add，依然按照set的方式下放即可。

 

不知道错在哪

  1 #include <bits/stdc++.h>
  2 using namespace std;
  3 int n, m, e, l, r, d, v;
  4 bool flag;
  5 struct node
  6 {
  7     int sum;    //总和
  8     int flo;    //浮动价格
  9     bool tag;   //lazy tag
 10     node *ll,*rr;
 11 };
 12 
 13 node *create()
 14 {
 15     node *tmp=new(node);
 16     tmp->sum=tmp->flo=tmp->tag=0;
 17     tmp->ll=tmp->rr=0;
 18     return tmp;
 19 }
 20 
 21 node * create_tree(int LL,int RR)
 22 {
 23     node *tmp=create();
 24     if(LL==RR)  //没有必要设置lazy tag
 25     {
 26         int a;
 27         cin>>a;
 28         tmp->sum=a;
 29         return tmp;
 30     }
 31     int mid=((LL+RR)>>1);
 32     tmp->ll= create_tree(LL,mid);
 33     tmp->rr= create_tree(mid+1,RR);
 34     tmp->sum= tmp->ll->sum + tmp->rr->sum;
 35     return tmp;
 36 }
 37 
 38 int dfs(const int ll,const int rr,const int LL,const int RR,const int val,node *t)
 39 {
 40     if(ll==LL&&rr==RR) 
 41     {
 42         t->tag=1;
 43         if(flag==1)    //政府调控
 44         {
 45             t->flo=0;
 46             t->sum=(RR-LL+1)*val;
 47         }
 48         else        //自发涨价
 49         {
 50             t->flo=val;
 51             t->sum+=(RR-LL+1)*val;
 52         }
 53         return t->sum;
 54     }
 55 
 56     int mid=(LL+RR)/2;
 57     if(t->tag)    //下放
 58     {
 59         t->tag=0;
 60         t->ll->tag= t->rr->tag= 1;
 61         if(t->flo==0)//政府调控
 62         {
 63             t->ll->flo= t->rr->flo= 0;
 64             t->ll->sum= t->sum/(RR-LL+1)*(mid-LL+1);
 65             t->rr->sum= t->sum - t->ll->sum;
 66         }
 67         else  //需要改浮动价格的
 68         {
 69             t->ll->flo= t->rr->flo= t->flo;
 70             t->ll->sum+= (mid-LL+1)* t->flo;
 71             t->rr->sum+= (RR-mid)* t->flo;
 72         }
 73     }
 74 
 75     if(ll>mid)
 76         t->sum= t->ll->sum + dfs(ll,rr,mid+1,RR,val,t->rr);
 77     else if(rr<=mid)
 78         t->sum= t->rr->sum + dfs(ll,rr,LL,mid,val,t->ll);
 79     else
 80         t->sum=dfs(ll,mid,LL,mid,val,t->ll) + dfs(mid+1,rr,mid+1,RR,val,t->rr);
 81     return t->sum;
 82 }
 83 
 84 int main()
 85 {
 86     //freopen("input.txt", "r", stdin);
 87     cin>>n>>m;
 88     node *tree=create_tree(1,n+1);    //建树
 89     for(int i=0; i<m; i++)    //修改并输出
 90     {
 91         scanf("%d",&e);
 92         if(e)
 93         {
 94             flag=1;
 95             scanf("%d%d%d",&l,&r,&v);
 96             printf("%d\n",dfs(l+1, r+1, 1, n+1, v, tree));
 97         }
 98         else
 99         {
100             flag=0;
101             scanf("%d%d%d",&l,&r,&d);
102             printf("%d\n",dfs(l+1, r+1, 1, n+1, d, tree));
103         }
104     }
105     return 0;
106 }
WA代码

 

posted @ 2015-05-18 20:46  xcw0754  阅读(265)  评论(0)  编辑  收藏  举报