hdu-4614Vases and Flowers 线段树区间赋值

题目来源：http://acm.hdu.edu.cn/showproblem.php?pid=4614

题目是说给你一个区间然后初始值为0就是代表没有花

给你两种操作1 代表从A开始插F只花 花盆为空就查  不为空就往下一个推 直到花插完为止 如果一朵花都能插  就输出Can not put...

如果能就输出第一个插花位置 和最后一个插花位置, 多出来的花扔掉

2 代表把区间A,B 的花全部扔掉 要你输出扔掉的花有多少

题目就这样 ,我也不知道为啥一开始WA了十几次  花了一个晚上 最终换思路重写一遍过掉

后来发现AC的代码 如果插花F=0 竟然是要顺推把那个位置占掉  虽然我也不知道测试数据F 会不会等于0   =0=

思路就是先找出最左边的第一个能插花的位置first  再找出最右边的位置last

然后用了lazy标记用于区间操作 cover( 1, first ,last ,1);

其他也就没什么了

#include <iostream>
#include <cstdio>
#include <cstring>
#define MAX 100010
using namespace std;
typedef struct
{
int l;
int r;
int sum;
int cov;
} Tree;
Tree tree[MAX*10];
int N,M,ch,A,B,K,last,first;
void update(int i)
{
if(tree[i].cov!=-1)
{
tree[i].sum=(tree[i].r-tree[i].l+1)*tree[i].cov;
tree[i*2].cov=tree[i].cov;
tree[i*2+1].cov=tree[i].cov;
tree[i].cov=-1;
}
}
int build(int i,int l,int r)
{
int mid;
tree[i].l=l;
tree[i].r=r;
tree[i].sum=0;
tree[i].cov=-1;
if(l==r)
return 0;
mid=(l+r)/2;
build(i*2,l,mid);
build(i*2+1,mid+1,r);
tree[i].sum=tree[i*2].sum+tree[i*2+1].sum;
}
int cover(int i,int x,int y,int now)
{
int mid;
update(i);
if((x<=tree[i].l)&&(y>=tree[i].r))
{
tree[i].cov=now;
return 0;
}
mid=(tree[i].l+tree[i].r)/2;
if(x<=mid) cover(i*2,x,y,now);
if(y>mid) cover(i*2+1,x,y,now);
update(i*2);
update(i*2+1);
tree[i].sum=(tree[i*2].sum+tree[i*2+1].sum);
}
int cal(int i,int x,int y)
{
int mid;
int ans;
update(i);
if ((x<=tree[i].l)&&(y>=tree[i].r))
return (tree[i].sum);
ans=0;
mid=(tree[i].l+tree[i].r)/2;
if (x<=mid) ans=(ans+cal(i*2,x,y));
if (y>mid) ans=(ans+cal(i*2+1,x,y));
update(i*2);
update(i*2+1);
tree[i].sum=(tree[i*2].sum+tree[i*2+1].sum);
return ans;
}
int findfirst(int i,int x) //firs比较好找
{
int mid=(tree[i].l+tree[i].r)/2;
if(tree[i].l==tree[i].r)
return tree[i].l;
if(x>mid)
findfirst(i*2+1,x);
else
{
int sum=cal(i*2,x,tree[i*2].r);
if(sum==tree[i*2].r-x+1)
findfirst(i*2+1,mid+1);
else
findfirst(i*2,x);
}

}
int findlast(int i,int x,int k)
{
int mid=(tree[i].l+tree[i].r)/2;
if(tree[i].r==tree[i].l)
return tree[i].l;
if(mid<x) //如果x> mid 直接右转
findlast(i*2+1,x,k);
else
{
int sum=cal(i*2,x,tree[i*2].r);
int canputl=tree[i*2].r-x+1-sum;
int canputr=tree[i*2+1].r-tree[i*2+1].l+1- cal(i*2+1,mid+1,tree[i*2+1].r);
if(canputl==0)
findlast(i*2+1,mid+1,k);
else if(canputr==0) //这个是重点
findlast(i*2,x,k);
else
{
if(canputl<k)
findlast(i*2+1,mid+1,k-canputl);
else if(canputl>=k)
findlast(i*2,x,k);
}
}
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&N,&M);
build(1,0,N-1);
while(M--)
{
scanf("%d",&ch);
switch(ch)
{
case 1:
{
scanf("%d%d",&A,&K);
if(A>=N||cal(1,A,N-1)==N-A)
printf("Can not put any one.\n");
else
{
first=findfirst(1,A);
last=findlast(1,A,K);
printf("%d %d\n",first,last);
cover(1,first,last,1);
}
break;

}
case 2:
{
scanf("%d%d",&A,&B);
printf("%d\n",cal(1,A,B));
cover(1,A,B,0);
break;
}
}
}
printf("\n");
}
}