bzoj 1135: [POI2009]Lyz 线段树+Hall定理

题意

初始时滑冰俱乐部有1到n号的溜冰鞋各k双。已知x号脚的人可以穿x到x+d的溜冰鞋。 有m次操作，每次包含两个数ri，xi代表来了xi个ri号脚的人。xi为负，则代表走了这么多人。 对于每次操作，输出溜冰鞋是否足够。
1≤n≤200,000 ， 1≤m≤500,000 ， 1≤k≤10^9 ， 0≤d≤n ) ri xi （ 1≤i≤m， 1≤ri≤n-d ， |xi|≤10^9

分析

解法比较巧妙的一道题。

Hall定理：一个二分图，左边有n个点，右边有m个点，存在一个匹配使得左边n个点全被覆盖的充要条件是对于左边点的每一个大小为k集合，至少有k个右边的点与其相连。
再回到这一题，因为连的边是一个区间，所以很显然我们只用讨论连续序列的情况。因为如果我们选了两端而没有选中间，那么选了中间的情况肯定不会更优。
那么如果要存在完美匹配，则对于任意一个脚的集合[l,r]都要满足$\sum\limits_{i=l}^ra[i]<=(r-l+1+d)*k$
把式子画一画就变成了$\sum\limits(a[i]-k)<=d*k$
那么问题就变成了单点修改和查询最大连续子段和。
用线段树维护一下即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N=200005;

int n,m,k,d;
struct tree{LL l,r,s,mx;}t[N*5];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void updata(int d)
{
    t[d].s=t[d*2].s+t[d*2+1].s;
    t[d].mx=max(t[d*2].r+t[d*2+1].l,max(t[d*2].mx,t[d*2+1].mx));
    t[d].l=max(t[d*2].l,t[d*2].s+t[d*2+1].l);
    t[d].r=max(t[d*2+1].r,t[d*2+1].s+t[d*2].r);
}

void build(int d,int l,int r)
{
    if (l==r)
    {
        t[d].l=t[d].r=t[d].mx=t[d].s=-k;
        return;
    }
    int mid=(l+r)/2;
    build(d*2,l,mid);build(d*2+1,mid+1,r);
    updata(d);
}

void ins(int d,int l,int r,int x,int y)
{
    if (l==r)
    {
        t[d].l=t[d].r=t[d].mx=t[d].s=t[d].s+y;
        return;
    }
    int mid=(l+r)/2;
    if (x<=mid) ins(d*2,l,mid,x,y);
    else ins(d*2+1,mid+1,r,x,y);
    updata(d);
}

int main()
{
    n=read();m=read();k=read();d=read();
    build(1,1,n);
    while (m--)
    {
        int r=read(),x=read();
        ins(1,1,n,r,x);
        if (t[1].mx<=(LL)d*k) puts("TAK");
        else puts("NIE");
    }
    return 0;
}