HDU 4614 Vases and Flowers
HDU 4614 Vases and Flowers
线段树
题意
两个数n,m, n表示花瓶数量0~n-1，开始全为空。接下来有m组，每组有数k，a，b;当k=1时， 从a位置开始插花，b为花数量，当花瓶有花时，跳过当前，直到找到空花瓶再插入，输出插入第一支花的位置和最后一支花的位置，花可以没插完，当一支都没有插入则输出Can not put any one.；当k=2时，清空【a,b】的花瓶，并输出在被清空的花的数量。
思路
线段树用1表示空位，0表示有花。
清空实质就是统计a,b之间有多少花，并把区间a,b强制置1。
插花要有点技巧，设置两个查询函数：查询l,r之间空位的数量，查询从第开头数p个空位到哪。
这样从a开始插b朵花：先查询a到末尾与多少空位x，若x==0输出没法插花；x小于b就把b换成x，实际插x朵；否则插b朵。然后查询开头到a有多少空位k，从开头数k+1个空位到哪，数k+b个空位到哪。这两个数就是实际插花的左右端点。在将这个区间全置0即可。
应该有线段树上二分的写法，可以直接查询从a往右数b个空瓶到哪，可是我不会写。
代码
注意题目从0开始编号，注意转换与边界处理。
#includeelse if(R<=mid) res=update(L, R, col, lson);
else res=update(L, R, col, lson)+update(L, R, col, rson);
pushup(rt);
return res;
}
int query(int L, int R, int l, int r, int rt)
{
if(L<=l&&r<=R) return stree[rt].sum;
pushdown(l, r, rt);
int mid=(l+r)>>1;
if(R<=mid) return query(L, R, lson);
else if(midreturn query(L, R, rson);
else return query(L, R, lson)+query(L, R, rson);
}
int query_pos(int p, int l, int r, int rt)
{
if(l==r) return l;
else if(stree[rt].sum==p) return stree[rt].rpos;
int mid=(l+r)>>1;
pushdown(l, r, rt);
if(stree[rt<<1].sum>=p)
return query_pos(p, lson);
else
return query_pos(p-stree[rt<<1].sum, rson);
}
int main()
{
int T;scanf("%d", &T);
while(T--)
{
int n, m;scanf("%d%d", &n, &m);
build(1, n, 1);
while(m--)
{
int a, b, c;scanf("%d%d%d", &a, &b, &c);
if(a==1)
{
b++;
int sum=query(b, n, 1, n, 1);
if(sum==0) puts("Can not put any one.");
else
{
int pre=(b==1?0:query(1, b-1, 1, n, 1));
c=min(c, sum);
int l=query_pos(pre+1, 1, n, 1);
int r=query_pos(pre+c, 1, n, 1);
update(l, r, 0, 1, n, 1);
printf("%d %d\n", l-1, r-1);
}
}
else
{
printf("%d\n", update(b+1, c+1, 1, 1, n, 1));
}
}
printf("\n");
}
//system("pause");
return 0;
}