[HDOJ 4893] Wow! Such Sequence! [线段树]

给定一个序列，现有3种操作：

1. 查询区间[l,r]的元素的和。

2. 将区间[l,r]元素置为最近的Fibonacci数

3. 将第x个元素加上d

线段树，维护区间和，区间都变成Fibonacci数之后的和，以及是否进了变为Fibonacci数的懒操作标记即可。

最开始因为把第3个操作看成了把第x个元素置成d..调了好久...

#include <cstdio>
#include <set>

using namespace std;

struct Node {
	long long v,vv;
	bool flag;
	Node *ls,*rs;
	void down() {
		if (!flag) return;
		if (ls) {
			ls->v=ls->vv;
			ls->flag=true;
		}
		if (rs) {
			rs->v=rs->vv;
			rs->flag=true;
		}
		flag=false;
	}
	void repair() {
		v=ls->v+rs->v;
		vv=ls->vv+rs->vv;
	}
};
Node a[200100];
Node *ap,*root;
int n,m;
set <long long> c;

long long nearstFib(long long x) {
	if (x<=1) return 1;
	set<long long>::iterator tmp=c.lower_bound(x);
	long long y=*tmp;
	tmp--;
	long long z=*tmp;
	if (y-x<x-z) return y;
	return z;
}
Node *maketree(int l,int r) {
	Node *ans=ap++;
	if (l==r) {
		ans->ls=ans->rs=NULL;
		ans->v=0;
		ans->vv=1;
	} else {
		int t=(l+r)/2;
		ans->ls=maketree(l,t);
		ans->rs=maketree(t+1,r);
		ans->repair();
	}
	ans->flag=false;
	return ans;
}
void set1(Node *from,int l,int r,int i,int x) {
	if (l==r) {
		from->flag=false;
		from->v+=x;
		from->vv=nearstFib(from->v);
	} else {
		from->down();
		int t=(l+r)/2;
		if (i<=t) {
			set1(from->ls,l,t,i,x);
		} else {
			set1(from->rs,t+1,r,i,x);
		}
		from->repair();
	}
}
long long get(Node *from,int l,int r,int ll,int rr) {
	if (l==ll&&r==rr) {
		return from->v;
	}
	from->down();
	int t=(l+r)/2;
	if (rr<=t) {
		return get(from->ls,l,t,ll,rr);
	} else if (ll>t) {
		return get(from->rs,t+1,r,ll,rr);
	} else {
		return get(from->ls,l,t,ll,t)+get(from->rs,t+1,r,t+1,rr);
	}
}
void set2(Node *from,int l,int r,int ll,int rr) {
	if (l==ll&&r==rr) {
		from->flag=true;
		from->v=from->vv;
	} else {
		from->down();
		int t=(l+r)/2;
		if (rr<=t) {
			set2(from->ls,l,t,ll,rr);
		} else if (ll>t) {
			set2(from->rs,t+1,r,ll,rr);
		} else {
			set2(from->ls,l,t,ll,t);
			set2(from->rs,t+1,r,t+1,rr);
		}
		from->repair();
	}
}

int main() {
	long long x,y,z;
	int i;
	x=y=1;
	while (y<1ll<<62) {
		c.insert(y);
		z=y;
		y+=x;
		x=z;
	}
	while (scanf("%d%d",&n,&m)!=EOF) {
		ap=a;
		root=maketree(1,n);
		for (i=0;i<m;i++) {
			int k,l,r;
			scanf("%d%d%d",&k,&l,&r);
			if (k==1) {
				set1(root,1,n,l,r);
			} else if (k==2) {
				printf("%lld\n",get(root,1,n,l,r));
			} else if (k==3) {
				set2(root,1,n,l,r);
			}
		}
	}
	return 0;
}