hdu4027Can you answer these queries?(线段树）

链接

算是裸线段树了，因为没个数最多开63次 ，开到不能再看就标记。查询时，如果某段区间被标记直接返回结果，否则继续向儿子节点更新。

注意用——int64

注意L会大于R 这点我很纠结。。您出题人故意的吗 WAn次。。

  1 #include <iostream>
  2 #include<cstdio>
  3 #include<cstring>
  4 #include<algorithm>
  5 #include<stdlib.h>
  6 #include<vector>
  7 #include<cmath>
  8 #include<queue>
  9 #include<set>
 10 using namespace std;
 11 #define N 100010
 12 #define LL __int64
 13 #define INF 0xfffffff
 14 const double eps = 1e-8;
 15 const double pi = acos(-1.0);
 16 const double inf = ~0u>>2;
 17 LL s[N<<2],f[N<<2];
 18 LL a[N];
 19 void up(int w)
 20 {
 21     s[w] =s[w<<1]+s[w<<1|1];
 22     f[w] = (f[w<<1]&&f[w<<1|1]);
 23 }
 24 void build(int l,int r,int w)
 25 {
 26     if(l==r)
 27     {
 28         s[w] = a[l];
 29         if(a[l]==1||a[l]==0) f[w] = 1;
 30         return ;
 31     }
 32     int m = (l+r)>>1;
 33     build(l,m,w<<1);
 34     build(m+1,r,w<<1|1);
 35     up(w);
 36 }
 37 void update(int a,int b,int l,int r,int w)
 38 {
 39     if(a<=l&&b>=r)
 40     {
 41         if(l==r)
 42         {
 43             LL k = sqrt(s[w]*1.0);
 44             if(k*k>s[w])
 45             k--;
 46             s[w] = k;
 47             if(s[w]<=1)
 48             f[w] = 1;
 49            // up(w);
 50             return ;
 51         }
 52         if(f[w])
 53         {
 54             up(w);
 55             return ;
 56         }
 57         else
 58         {
 59             int m = (l+r)>>1;
 60             if(a<=m)
 61             update(a,b,l,m,w<<1);
 62             if(b>m)
 63             update(a,b,m+1,r,w<<1|1);
 64             up(w);
 65             return ;
 66         }
 67     }
 68     int m  = (l+r)>>1;
 69     if(a<=m)
 70     update(a,b,l,m,w<<1);
 71     if(b>m)
 72     update(a,b,m+1,r,w<<1|1);
 73     up(w);
 74 }
 75 LL query(int a,int b,int l,int r,int w)
 76 {
 77     if(a<=l&&b>=r)
 78     {
 79         return s[w];
 80     }
 81     int m = (l+r)>>1;
 82     LL res = 0;
 83     if(a<=m) res+=query(a,b,l,m,w<<1);
 84     if(b>m) res+=query(a,b,m+1,r,w<<1|1);
 85     return res;
 86 }
 87 int main()
 88 {
 89     int i,n,q;
 90     int k,x,y,kk=0;
 91     while(scanf("%d",&n)!=EOF)
 92     {
 93         memset(f,0,sizeof(f));
 94         for(i = 1; i <=n ;i++)
 95         {
 96             scanf("%I64d",&a[i]);
 97         }
 98         build(1,n,1);
 99         scanf("%d",&q);
100         printf("Case #%d:\n",++kk);
101         while(q--)
102         {
103             scanf("%d%d%d",&k,&x,&y);
104             if(x>y) swap(x,y);
105             if(k)
106             {
107                 printf("%I64d\n",query(x,y,1,n,1));
108             }
109             else
110             {
111                 update(x,y,1,n,1);
112             }
113         }
114         puts("");
115     }
116     return 0;
117 }
View Code

 

 

posted @ 2014-05-29 14:41  _雨  阅读(162)  评论(0)  编辑  收藏  举报