题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4893
题目大意:给你N个数,初始状态每个数都为0,现在有三种操作:1,给第K个数增加a 2,询问[L,R]区间所有数的和 3,区间【l,r】内每个数都变为离他最近的斐波那契数。
解题思路:刚开始想的时候 想用mark标记该区间是否已经变为离他最近的斐波那契数,每次插入一个数的时候进行向下向上更新,但是发现无法实时更新sum值。
正确的思路是:在每个节点开一个sum数组记录所有子节点的和,再开一个fibsum数组记录所有子节点的斐波那契数之和,然后进行操作3时,只需要更新值时将sum用fibsum更新就可以了,每次插入的时候要维护每个节点fibsum的值。代码如下:
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<cstdlib>
using namespace std;
typedef __int64 LL;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define N 100005
LL sum[N<<4],fibsum[N<<4];
bool mark[N<<4];
LL fib[100];
void get_fib()
{
fib[0] = fib[1] = 1;
for(int i = 2; i <= 95; i++)
fib[i] = fib[i-1] + fib[i-2];
}
LL nearest_fib(LL x)
{
int pos = lower_bound(fib,fib+93,x)-fib;
if(pos && abs(fib[pos-1]-x) <= abs(fib[pos]-x))
return fib[pos-1];
return fib[pos];
}
void pushup(int rt)//向上更新
{
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
fibsum[rt]=fibsum[rt<<1]+fibsum[rt<<1|1];
}
void pushdown(int rt)//向下更新
{
if(mark[rt])
{
mark[rt<<1]=true;
mark[rt<<1|1]=true;
sum[rt<<1]=fibsum[rt<<1];
sum[rt<<1|1]=fibsum[rt<<1|1];
mark[rt]=false;
}
}
void build(int l,int r,int rt)//建树
{
if(l==r)
{
mark[rt]=false;
sum[rt]=0;
fibsum[rt]=1;
return;
}
int m=(l+r)>>1;
build(lson);
build(rson);
pushup(rt);
mark[rt]=false;//建树时所有节点标记为0
}
void insert(int pos,LL a,int l,int r,int rt)//单点插入值
{
if(l==r&&l==pos)
{
sum[rt]+=a;
fibsum[rt]=nearest_fib(sum[rt]);//更新fibsum【】的值
return;
}
pushdown(rt);
int m=(l+r)>>1;
if(pos<=m) insert(pos,a,lson);
else insert(pos,a,rson);
pushup(rt);
}
void update(int L,int R,int l,int r,int rt) //操作3区间更新最近值
{
if(L<=l&&R>=r)
{
mark[rt]=true;
sum[rt]=fibsum[rt];
return;
}
pushdown(rt);
int m=(l+r)>>1;
if(L<=m) update(L,R,lson);
if(R>m) update(L,R,rson);
pushup(rt);
}
LL query(int L,int R,int l,int r,int rt) //操作2区间求和
{
if(L<=l&&R>=r)
return sum[rt];
pushdown(rt);
int m=(l+r)>>1;
LL ret=0;
if(L<=m) ret+=query(L,R,lson);
if(R>m) ret+=query(L,R,rson);
return ret;
}
int main()
{
//freopen("in.txt","r",stdin);
get_fib();
int n,m;
while(~scanf("%d%d",&n,&m))
{
build(1,n,1);//建树时已经初始化,故不需要memset,而且浪费时间
int a,x,y;
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&a,&x,&y);
if(a==1)
insert(x,y,1,n,1);
else if(a==2)
printf("%I64d\n",query(x,y,1,n,1));
else
update(x,y,1,n,1);
}
}
return 0;
}