并查集+线段树 BZOJ 3038: 上帝造题的七分钟2

3038: 上帝造题的七分钟2

Time Limit: 3 Sec   Memory Limit: 128 MB
Submit: 422   Solved: 181
[ Submit][ Status]

Description

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

Input

第一行一个整数n，代表数列中数的个数。
第二行n个正整数，表示初始状态下数列中的数。
第三行一个整数m，表示有m次操作。
接下来m行每行三个整数k,l,r，k=0表示给[l,r]中的每个数开平方(下取整)，k=1表示询问[l,r]中各个数的和。

Output

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

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

HINT

1：对于100%的数据，1<=n<=100000，1<=l<=r<=n，数列中的数大于0，且不超过1e12。

2：数据不保证L<=R 若L>R，请自行交换L,R，谢谢！

思路：由于每次操作是对一段数取根号，所以对于每个long long以内的数来讲，最多只会取6次根号就变成了1，然后就一直是1了。那些变成了1的位置我们能够直接删掉。 怎么删掉呢？我们借助并查集，表示当前位置开始到后面最近的一个还不是1的位置。当某一个位置变为了1，我们讲这个位置跟l+1合并就行了。具体可以看下代码。线段树主要是用来查询区间和的。

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
using namespace std;
const int maxn=100000+5;
int tree_size;
#define LL unsigned long long
LL sum[maxn<<2];

int p[maxn];
int find(int x) { return p[x]==x?x:p[x]=find(p[x]); }

LL a[maxn];

int n;

void update(int x,LL v)
{
    x+=tree_size; sum[x]=v;
    x>>=1;
    while(x>0) {
        sum[x]=sum[x<<1]+sum[x<<1|1];
        x>>=1;
    }
}

LL query(int l,int r)
{
   l+=tree_size,r+=tree_size;
   LL ret=0;
   while(l<=r) {
        if(l&1)  ret+=sum[l++];
        if(~r&1) ret+=sum[r--];
        l>>=1,r>>=1;
   }
   return ret;
}


void input()
{
    for(int i=1;i<=n;++i) scanf("%lld",&a[i]);
    for(int i=1;i<=n+1;++i) p[i]=i;
    tree_size=1;
    while(tree_size*2<n) tree_size<<=1;
    tree_size = 2*tree_size-1;
    for(int i=1;i<=n;++i)
        update(i,a[i]);
}

inline void swap(int&a,int&b) { a^=b; b^=a; a^=b; }
void solve()
{
    int m;scanf("%d",&m);
    while(m--) {
        int k,l,r; scanf("%d%d%d",&k,&l,&r);
        if(r<l) swap(l,r);
        if(k==1) printf("%lld\n",query(l,r));
        else {
            while(l<=r) {
                if(l!=find(l)) { l=find(l); continue; }
                a[l]=floor(sqrt((double)a[l])+1e-8);
                update(l,a[l]);
                if(a[l]==1) p[l]=find(l+1);
                l=find(l+1);
            }
        }
    }
}

int main()
{
    while(scanf("%d",&n)==1)
    {
        input();
        solve();
    }
}