NBUT 1120 Reimu's Teleport （线段树）

  1 #include<bits/stdc++.h>
  2 using namespace std;
  3 int ans[3];//答案
  4 struct node
  5 {
  6     bool tag;
  7     int num[3];
  8     node *ll,*rr;
  9     node()
 10     {
 11         tag=false;
 12         num[0]=num[1]=num[2]=0;
 13         ll=rr=0;
 14     }
 15 };
 16 
 17 node * built(int LL, int RR)
 18 {
 19     node *tmp=new(node);
 20     if(LL==RR)
 21     {
 22         tmp->num[1]=1;
 23         return tmp;
 24     }
 25     int mid=(LL+RR)>>1;
 26     tmp->ll=built(LL,mid);
 27     tmp->rr=built(mid+1,RR);
 28     tmp->num[1]+=tmp->ll->num[1]+tmp->rr->num[1];   //开始只需要更新向上的
 29     return tmp;
 30 }
 31 
 32 void modify( node *t,int LL,int RR,int mid)//专门修改统计数量
 33 {
 34     if(t->num[0])
 35     {
 36         t->ll->num[0]=mid+1-LL;
 37         t->ll->num[1]=t->ll->num[2]=0;
 38         t->rr->num[0]=RR-mid;
 39         t->rr->num[1]=t->rr->num[2]=0;
 40     }
 41     if(t->num[1])
 42     {
 43         t->ll->num[1]=mid+1-LL;
 44         t->ll->num[0]=t->ll->num[2]=0;
 45         t->rr->num[1]=RR-mid;
 46         t->rr->num[0]=t->rr->num[2]=0;
 47     }
 48     if(t->num[2])
 49     {
 50         t->ll->num[2]=mid+1-LL;
 51         t->ll->num[0]=t->ll->num[1]=0;
 52         t->rr->num[2]=RR-mid;
 53         t->rr->num[0]=t->rr->num[1]=0;
 54     }
 55 }
 56 
 57 void update(int ll,int rr,int LL,int RR,int c,node *t)
 58 {
 59     if(ll==LL&&rr==RR)
 60     {
 61         t->tag=true;
 62         if(c==0)
 63         {
 64             t->num[0]=RR-LL+1;
 65             t->num[1]=t->num[2]=0;
 66         }
 67         else
 68         {
 69             t->num[2]=RR-LL+1;
 70             t->num[0]=t->num[1]=0;
 71         }
 72         return;
 73     }
 74     int mid=(LL+RR)>>1;
 75 
 76     if(t->tag)
 77     {
 78         t->tag=false;
 79         t->ll->tag=t->rr->tag=true;
 80         modify(t,LL,RR,mid);
 81     }
 82     if(ll>mid)  //右边
 83         update(ll,rr,mid+1,RR,c,t->rr);
 84     else if(rr<=mid)    //左边
 85         update(ll,rr,LL,mid,c,t->ll);
 86     else
 87     {
 88         update(ll, mid, LL, mid, c,t->ll);
 89         update(mid+1,rr,mid+1,RR,c,t->rr);
 90     }
 91     t->num[0]=t->ll->num[0]+t->rr->num[0];
 92     t->num[1]=t->ll->num[1]+t->rr->num[1];
 93     t->num[2]=t->ll->num[2]+t->rr->num[2];
 94 }
 95 
 96 
 97 
 98 void query(int ll,int rr,int LL,int RR,node* t)
 99 {
100     if(ll==LL && rr==RR)
101     {
102         ans[0]+=t->num[0];
103         ans[1]+=t->num[1];
104         ans[2]+=t->num[2];
105         return;
106     }
107     int mid=((LL+RR)>>1);
108     if(t->tag)
109     {
110         t->tag=false;
111         t->ll->tag= t->rr->tag= true;
112         modify(t, LL, RR, mid);
113     }
114 
115     if(ll>mid)          query(ll,rr,mid+1,RR,t->rr);
116     else if(rr<=mid)    query(ll,rr,LL,mid,t->ll);
117     else
118     {
119         query(ll, mid, LL, mid, t->ll);
120         query(mid+1,rr,mid+1,RR,t->rr);
121     }
122 
123 }
124 
125 
126 int main()
127 {
128     //freopen("e://input.txt","r",stdin);
129     int a, t, e, r;
130     char ch;
131 
132     while(cin>>a>>t)
133     {
134         node *tree=built(1,t); //建树
135 
136         for(int i=0; i<a; i++)
137         {
138             cin>>ch;
139             scanf("%d %d", &e, &r);
140             if(ch==‘F‘)  //修改
141             {
142                 if(e>r) update(r, e, 1, t, 0, tree);
143                 else    update(e, r, 1, t, 2, tree);
144             }
145             else        //Q 查询
146             {
147                 memset(ans,0,sizeof(ans));
148                 query(e,r,1,t,tree);
149                 printf("%d %d %d\n",ans[0],ans[1],ans[2]);
150             }
151         }
152     }
153     return 0;
154 }