首页 > > 程序设计 > C/C++ >

2492 上帝造题的七分钟 2

2018-06-17 22:16:46来源：未知阅读 ()

时间限制: 1 s

空间限制: 64000 KB

题目等级 : 大师 Master

题目描述 Description

　　XLk觉得《上帝造题的七分钟》不太过瘾，于是有了第二部。

　　"第一分钟，X说，要有数列，于是便给定了一个正整数数列。
　　第二分钟，L说，要能修改，于是便有了对一段数中每个数都开平方(下取整)的操作。
　　第三分钟，k说，要能查询，于是便有了求一段数的和的操作。
　　第四分钟，彩虹喵说，要是noip难度，于是便有了数据范围。
　　第五分钟，诗人说，要有韵律，于是便有了时间限制和内存限制。
　　第六分钟，和雪说，要省点事，于是便有了保证运算过程中及最终结果均不超过64位有符号整数类型的表示范围的限制。
　　第七分钟，这道题终于造完了，然而，造题的神牛们再也不想写这道题的程序了。"
　　——《上帝造题的七分钟·第二部》
　　所以这个神圣的任务就交给你了。

输入描述 Input Description

　　第一行一个整数n，代表数列中数的个数。
　　第二行n个正整数，表示初始状态下数列中的数。
　　第三行一个整数m，表示有m次操作。
　　接下来m行每行三个整数k,l,r，k=0表示给[l,r]中的每个数开平方(下取整)，k=1表示询问[l,r]中各个数的和。
　　UPD：注意数据中有可能l>r，所以遇到这种情况请交换l和r。

输出描述 Output Description

　　对于询问操作，每行输出一个回答。

样例输入 Sample Input

10
1 2 3 4 5 6 7 8 9 10
5
0 1 10
1 1 10
1 1 5
0 5 8
1 4 8

样例输出 Sample Output

19
7
6

数据范围及提示 Data Size & Hint

　　对于30%的数据，1<=n,m<=1000，数列中的数不超过32767。
　　对于100%的数据，1<=n,m<=100000，1<=l,r<=n，数列中的数大于0，且不超过1e12。
　　注意l有可能大于r，遇到这种情况请交换l,r。

来源：Nescafe 20

树状数组+路径压缩

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<cmath>
 5 #include<queue>
 6 #include<algorithm>
 7 #define lli long long int 
 8 using namespace std;
 9 const lli MAXN=1000001;
10 inline void read(lli &n)
11 {
12     char c='+';lli x=0;bool flag=0;
13     while(c<'0'||c>'9'){c=getchar();if(c=='-')flag=1;}
14     while(c>='0'&&c<='9'){x=x*10+c-48;c=getchar();}
15     flag==1?n=-x:n=x;
16 }
17 lli tree[MAXN];
18 lli a[MAXN];
19 lli f[MAXN];
20 lli n,m,tmp;
21 lli lb(lli &x)
22 {
23     return x&-x;
24 }
25 void add_tree(lli pos,lli v)
26 {
27     while(pos<=n)
28     {
29         tree[pos]+=v;
30         pos+=lb(pos);
31     }
32 }
33 lli interval_sum(lli pos)
34 {
35     lli ans=0;
36     while(pos)
37     {
38         ans+=tree[pos];
39         pos-=lb(pos);
40     }
41     return ans;
42 }
43 lli find(lli x)
44 {
45     if(f[x]==x)
46         return f[x];
47     else f[x]=find(f[x]);
48 }
49 int main()
50 {
51     read(n);
52     f[n+1]=n+1;
53     for(lli i=1;i<=n;i++)
54     {
55         read(a[i]);
56         f[i]=i;
57         add_tree(i,a[i]);
58     }
59     read(m);
60     for(lli i=1;i<=m;i++)
61     {
62         lli how,l,r;
63         read(how);read(l);read(r);
64         if(l>r)swap(l,r);
65         if(how==0)// 开方 
66         {
67             for(l=find(l);l<=r;l=find(l+1))//压缩路径 
68             {
69                 tmp=a[l];
70                 a[l]=sqrt(a[l]);
71                 tmp-=a[l];//  
72                  
73                 if(a[l]==1)
74                     f[l]=find(l+1);
75                 add_tree(l,-tmp);
76             }
77         }
78         else// 询问 
79             printf("%lld\n",interval_sum(r)-interval_sum(l-1));
80     }
81     return 0;
82 }