【POI2015】【BZOJ4378】Logistyka

Description

维护一个长度为n的序列，一开始都是0，支持以下两种操作：
1.U k a 将序列中第k个数修改为a。
2.Z c s 在这个序列上，每次选出c个正数，并将它们都减去1，询问能否进行s次操作。
每次询问独立，即每次询问不会对序列进行修改。

Input

第一行包含两个正整数n,m(1<=n,m<=1000000)，分别表示序列长度和操作次数。
接下来m行为m个操作，其中1<=k,c<=n，0<=a<=10^9，1<=s<=10^9。

Output

包含若干行，对于每个Z询问，若可行，输出TAK，否则输出NIE。

Sample Input

3 8

U 1 5

U 2 7

Z 2 6

U 3 1

Z 2 6

U 2 2

Z 2 6

Z 2 1
Sample Output

NIE

TAK

NIE

TAK
HINT

鸣谢Claris

Source

鸣谢Claris

考虑建两个树状数组,都是权值BIT
一个记录出现次数一个前缀和
这样来思考:选数时候为了能选出s次,肯定尽量从大的里选
然后乱搞一下..
考虑把这些数分为大于等于s和小于s
大于等于s的最多只要取s次就好了,小于s的直接求和

需要离散化

BZOJ RE又显示成WA..一直调不出来..简直了..

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define lowbit(x)   (x&(-x))
#define GET (ch>='0'&&ch<='9')
#define MAXN 1000010
#define LL long long
using namespace std;
int n,m;
int a[MAXN],v[MAXN],cnt;
char ch[2];
struct BIT
{
    LL c[MAXN];
    inline void add(int x,int delta)
    {
        if (!x) return;
        for (int i=x;i<=cnt;i+=lowbit(i))   c[i]+=delta;
    }
    inline LL query(int x)
    {
        LL ret=0;
        for (int i=x;i;i-=lowbit(i))    ret+=c[i];
        return ret;
    }
}s[3];
void in(int &x)
{
    char ch=getchar();x=0;
    while (!GET)    ch=getchar();
    while (GET) x=x*10+ch-'0',ch=getchar();
}
struct Query
{
    int opt,x,y;
}q[MAXN];
int main()
{
    in(n);in(m);int x,y;
    for (int i=1;i<=m;i++)
    {
        scanf("%s",ch);in(x);in(y);v[i]=y;
        q[i].opt=(ch[0]=='U');q[i].x=x;q[i].y=y;
    }
    sort(v+1,v+m+2);cnt=unique(v+1,v+m+2)-(v+1);
    for (int i=1;i<=m;i++)  q[i].y=lower_bound(v+1,v+cnt+1,q[i].y)-v;
    for (int i=1;i<=m;i++)
        if (q[i].opt)
        {
            x=a[q[i].x];s[1].add(x,-1);s[2].add(x,-v[x]);
            x=a[q[i].x]=q[i].y;
            s[1].add(x,1);s[2].add(x,v[x]);
        }
        else
        {
            if (s[2].query(q[i].y-1)+(s[1].query(cnt)-s[1].query(q[i].y-1))*v[q[i].y]>=(LL)q[i].x*v[q[i].y])    puts("TAK");
            else    puts("NIE");
        }
}