Uva 12436 Rip Van Winkle's Code(区间更新,区间查询)

题意:对于长度为250000的区间,给了你四种操作:操作A,从st到ed这段区间内的数,分别加上1,2,...,st-ed+1。操作B,从st到ed这段区间内的数,分别加上,st-ed+1,st-ed,...,1。操作C,将st到ed这段区间内的数赋值成x。操作S,查询st到ed的这段区间内的数的总和。

因为操作A和操作B都是操作的实际上都是等差数列,所以可以一起考虑。还是线段树,对于操作AB,在线段树的结点中记录了左端点要加的值add1,右端点要加的值add2,同时记录了这段区间内的公差step(都是操作AB的懒操作的标记)。对于操作C,在线段树的结点内,用一个flag表示当前区间内的数是否全部相同,并且还用一个valu表示如果在区间里的数完全相同的时候,即flag=1时,这个数是多少(都是操作C的懒操作的标记)。

容易得到,等差数列相加还是等差数列,所以可以对一个区间进行多次的AB操作。对于C操作,因为强制将区间内的赋值成x,所以之前的AB操作全部失效,即这时候应该把表示AB操作的几个变量,add1、add2和step全部赋值为零。对向子区间传递记录的值的时候,应该优先C操作的标记,因为当操作AB和操作C的标记同时存在的时候,一定是发生在先进行了操作C,再进行了操作AB,如果这时候,先向下传递AB的标记,再传递C的标记,就会导致结果不正确。

WA很久,犯的两个错误:

1.在开始的时候,只记录add1,add2,然后在向子区间传递懒操作的值,即add1和add2的时候,要分解成两个区间,设为(add1,mid1),(mid2,add2),通过(add1+add2)/2直接得到mid1,然后mid2再根据是递增或是递减相应的加一减一,实际上这里应该加上或减去公差,这也是为什么在线段树的结点里添加公差step的原因。

2.操作C的标记只有valu而没有flag,错误的认为,只要valu值不为零,就表示有要向下传递的值,实际上,操作C可以把区间全部赋值成零,这时候,valu等于零,不会向下传递,导致错误,所以多添加了一个标记flag。

PS:在有些OJ上*2有些慢T了,换成<<1就A了。。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int N=250005;
typedef long long LL;
struct node
{
    bool flag;
	int left,right;
	LL sum,add1,add2,valu,step;
	LL mid(){return left+(right-left)/2;}
	LL len(){return right-left+1;}
	void changeAB(LL a,LL b,LL k)
	{
		add1+=a;    add2+=b;    step+=k;
		sum+=(a+b)*len()/2;
	}
	void changeC(LL a)
	{
	    flag=1;     valu=a;
		add1=0;     add2=0;     step=0;
        sum=valu*len();
	}
};
struct Segtree
{
	node tree[N*4];
	void down(int ind)
	{
		if(tree[ind].flag)
		{
			tree[ind*2].changeC(tree[ind].valu);
			tree[ind*2+1].changeC(tree[ind].valu);
			tree[ind].flag=0;
		}
		if(tree[ind].add1||tree[ind].add2||tree[ind].step)
		{
		    LL add1=tree[ind].add1,add2=tree[ind].add2;
		    LL k=tree[ind].step;
		    LL mid=add1+k*(tree[ind*2].len()-1);

			tree[ind*2].changeAB(add1,mid,k);
			tree[ind*2+1].changeAB(mid+k,add2,k);
			tree[ind].add1=0;	tree[ind].add2=0;
			tree[ind].step=0;
		}
	}
	void build(LL left,LL right,LL ind)
	{
		tree[ind].left=left;	tree[ind].right=right;
		tree[ind].add1=0;		tree[ind].add2=0;
		tree[ind].sum=0;		tree[ind].valu=0;
		tree[ind].step=0;
		if(left!=right)
		{
			LL mid=tree[ind].mid();
			build(left,mid,ind*2);
			build(mid+1,right,ind*2+1);
		}
	}
	void updataAB(LL be,LL end,LL ind,LL step)
	{
		LL left=tree[ind].left,right=tree[ind].right;
		if(be<=left&&right<=end)
		{
		    LL st,ed;
		    if(step>=0) {st=left-be+1;ed=right-be+1;}
		    else {st=end-left+1;ed=end-right+1;}
		    tree[ind].changeAB(st,ed,step);
		}
		else
		{
			down(ind);
			LL mid=tree[ind].mid();
			if(be<=mid) updataAB(be,end,ind*2,step);
			if(end>mid) updataAB(be,end,ind*2+1,step);
			tree[ind].sum=tree[ind*2].sum+tree[ind*2+1].sum;
		}
	}
	void updataC(LL be,LL end,LL ind,LL valu)
	{
		LL left=tree[ind].left,right=tree[ind].right;
		if(be<=left&&right<=end) tree[ind].changeC(valu);
		else
		{
			down(ind);
			LL mid=tree[ind].mid();
			if(be<=mid) updataC(be,end,ind*2,valu);
			if(end>mid) updataC(be,end,ind*2+1,valu);
			tree[ind].sum=tree[ind*2].sum+tree[ind*2+1].sum;
		}
	}
	LL query(LL be,LL end,LL ind)
	{
		LL left=tree[ind].left,right=tree[ind].right;
		if(be<=left&&right<=end) return tree[ind].sum;
		else
		{
			down(ind);
			LL mid=tree[ind].mid();
			LL sum1=0,sum2=0;
			if(be<=mid) sum1=query(be,end,ind*2);
			if(end>mid) sum2=query(be,end,ind*2+1);
			return sum1+sum2;
		}
	}
}seg;
int main()
{

    int n;
    while(scanf("%d",&n)!=EOF)
    {
		seg.build(1,N-5,1);
		for(int i=0;i<n;i++)
		{
			char str[10];
			LL a,b,c;
			scanf("%s",str);
			if(str[0]=='A')
			{
				scanf("%lld%lld",&a,&b);
				seg.updataAB(a,b,1,1);
			}
			else if(str[0]=='B')
			{
				scanf("%lld%lld",&a,&b);
				seg.updataAB(a,b,1,-1);
			}
			else if(str[0]=='C')
			{
				scanf("%lld%lld%lld",&a,&b,&c);
				seg.updataC(a,b,1,c);
			}
			else
			{
				scanf("%lld%lld",&a,&b);
				printf("%lld\n",seg.query(a,b,1));
			}
		}
	}
	return 0;
}


你可能感兴趣的:(c,struct,tree,query,ini,Build)