码迷,mamicode.com
首页 > 其他好文 > 详细

@bzoj - 4378@ [POI2015] Logistyka

时间:2019-02-10 13:45:23      阅读:204      评论:0      收藏:0      [点我收藏+]

标签:保留   次数   操作   title   正整数   看到了   uniq   typedef   题解   


@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

@solution@

如果序列中的某个数 x > s,因为我们只进行 s 次 -1 的操作,这个数能用到的部分只有 s 这么多。
所以我们可以将序列中所有大于 s 的数修改成 s。

询问能够成立的必要条件是序列所有数之和 >= c*s。
而可以发现,当我们将大于 s 的数修改成 s 过后,如上的条件就变成了充要条件。

证明的话可以使用归纳法。
首先可以发现,每次操作至少保留 c 个正数。分类讨论等于 s 的数的个数:
(1)若等于 s 的数 >= c,显然成立(直接拿这几个数进行操作)。
(2)若等于 s 的数 < c,我们保证选出的 c 个数包含这些等于 s 的数。这样操作完后依然满足所有数之和 >= c*(s-1)且所有数<=(s-1),就可以归纳证明了。

至于实现,我们并不需要真的把所有大于 s 的数修改成 s,只需要统计大于 s 的数的个数再乘上 s 即可。

可以用平衡树维护,也可以用离散化 + 线段树维护。

@accepted code@

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int MAXN = 1000000;
struct query{
    int type, x, y;
}qry[MAXN + 5];
int a[MAXN + 5], d[MAXN + 5];
int n, m, dcnt; 
void discrete() {
    sort(d+1, d+dcnt+1);
    dcnt = unique(d+1, d+dcnt+1) - d - 1;
    for(int i=1;i<=m;i++)
        if( !qry[i].type )
            qry[i].y = lower_bound(d+1, d+dcnt+1, qry[i].y) - d;
}
struct node{
    int le, ri;
    ll sum; int cnt;
}tree[4*MAXN + 5];
void pushup(int x) {
    tree[x].sum = tree[x<<1].sum + tree[x<<1|1].sum;
    tree[x].cnt = tree[x<<1].cnt + tree[x<<1|1].cnt;
}
void build(int x, int l, int r) {
    tree[x].le = l, tree[x].ri = r;
    tree[x].sum = tree[x].cnt = 0;
    if( l == r ) {
        if( l == 1 ) tree[x].cnt = n;
        return ;
    }
    int mid = (l + r) >> 1;
    build(x<<1, l, mid);
    build(x<<1|1, mid+1, r);
    pushup(x);
}
void modify(int x, int p, int k) {
    if( p > tree[x].ri || p < tree[x].le )
        return ;
    if( tree[x].le == tree[x].ri ) {
        tree[x].sum += k*d[tree[x].le];
        tree[x].cnt += k;
        return ;
    }
    modify(x<<1, p, k);
    modify(x<<1|1, p, k);
    pushup(x);
}
ll s; int c;
void query(int x, int p) {
    if( d[tree[x].ri] <= p ) s += tree[x].sum;
    else if( d[tree[x].le] > p ) c += tree[x].cnt;
    else query(x<<1, p), query(x<<1|1, p);
}
char cmd[2];
int main() {
    d[dcnt = 1] = 0;
    scanf("%d%d", &n, &m);
    for(int i=1;i<=m;i++) {
        scanf("%s%d%d", cmd, &qry[i].x, &qry[i].y);
        if( cmd[0] == ‘U‘ )
            qry[i].type = 0, d[++dcnt] = qry[i].y;
        else qry[i].type = 1;
    }
    discrete(); build(1, 1, dcnt);
    for(int i=1;i<=n;i++)
        a[i] = 1;
    for(int i=1;i<=m;i++) {
        if( !qry[i].type ) {
            modify(1, a[qry[i].x], -1);
            a[qry[i].x] = qry[i].y;
            modify(1, a[qry[i].x], 1);
        }
        else {
            s = c = 0; query(1, qry[i].y);
            puts(1LL*qry[i].x*qry[i].y <= s + 1LL*c*qry[i].y ? "TAK" : "NIE");
        }
    }
}

@details@

一开始找了一个比较复杂的结论,还涉及到线段树上二分等诡异的操作。
大概一两个月过后再去看……这什么玩意儿啊……我怎么看不懂我写了啥啊……

然后翻看题解才看到了这个比较简单的结论。

@bzoj - 4378@ [POI2015] Logistyka

标签:保留   次数   操作   title   正整数   看到了   uniq   typedef   题解   

原文地址:https://www.cnblogs.com/Tiw-Air-OAO/p/10359090.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!