HDU 5245 Joyful（线段树、查询区间和及修改区间每个数为数的平方）

题目链接：

HDU 5245 Joyful

题意：

查询区间和及修改查询区间，将每个数都变为数的平方。模数是9223372034707292160。

分析：

任何一个数的若干次平方后模上9223372034707292160的值都会不变？！你敢信？

用flag标记当前区间是否每个数平方取模后值是否都不变。

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <algorithm>
#include <limits.h>
#include <time.h>
#include <string>
#define lson(x) (x<<1)
#define rson(x) ((x<<1)|1)
//#define ll unsigned long long
#define IOS ios_base::sync_with_stdio(0); cin.tie(0);
using namespace std;

typedef unsigned long long ll;

const ll mod =9223372034707292160uLL; //需要用unsigned long long 存储模数
const int maxn=100500;

int T,n,m,cases=0;
ll ans,val[maxn];

struct SegTree{
int left,right,flag;
ll val;
}segtree[maxn<<2];

//快速乘法
inline ll mul(ll x,ll y)
{
ll res=0;
while(y){
if(y&1) res=(res+x)%mod;
y>>=1;
x=x*2%mod;
}
return res;
}

inline void build(int left,int right,int cur)
{
segtree[cur].left=left;
segtree[cur].right=right;
segtree[cur].flag=0;
if(left==right)
{
segtree[cur].val=val[left];
return;
}
int mid=(left+right)>>1;
build(left,mid,(cur<<1));
build(mid+1,right,(cur<<1)|1);
segtree[cur].val=segtree[(cur<<1)].val+segtree[((cur<<1)|1)].val;
}

inline void query(int a,int b,int cur)
{
int left=segtree[cur].left;
int right=segtree[cur].right;
if(a==left && b==right && segtree[cur].flag==1) { //判断当前区间是否是查询区间并且是否已标记
ans=(ans+segtree[cur].val)%mod;
return ;
}
if(left==right)
{
ans=(ans+segtree[cur].val)%mod;
ll tmp=mul(segtree[cur].val,segtree[cur].val);
if(tmp==segtree[cur].val){ //平方后数据不变
segtree[cur].flag=1; //标记flag为1
}
segtree[cur].val=tmp;
return ;
}
int mid=(left+right)>>1;
if(b<=mid) query(a,b,(cur<<1));
else if(a>mid) query(a,b,((cur<<1)|1));
else {
query(a,mid,(cur<<1));
query(mid+1,b,((cur<<1)|1));
}
segtree[cur].val=(segtree[(cur<<1)].val+segtree[((cur<<1)|1)].val)%mod;
//如果左右子区间的数据平方后都不变,标记当前区间
if(segtree[cur<<1].flag && segtree[(cur<<1)|1].flag ) segtree[cur].flag=1;
}

int main()
{
freopen("D.in","r",stdin);
scanf("%d",&T);
while(T--){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) {
scanf("%I64u",&val[i]);
}
build(1,n,1);
ans=0;
printf("Case #%d:\n",++cases);
for(int i=0;i<m;i++) {
int a,b;
scanf("%d%d",&a,&b);
query(a,b,1);
ans%=mod;
printf("%I64u\n",ans);
}
}
return 0;
}