bzoj3441(线段树的奇怪题)

来源:互联网 发布:算法分析专业 编辑:程序博客网 时间:2024/06/11 02:26

比赛时碰到的,其实并不是什么imba的题,但其他人都沉迷30行的代码,没人用我的解法,所以就写了。

题面

语文不好,题意不会说了。

显然每个(马猴)水缸的喝水次数是可以算出来的,即为tim[i]。
朴素的思考,一轮轮的喝,那么我只要知道每一轮喝了多少次就可以了。
根据(马猴)水缸的特点,某个水缸在某一轮没被喝,以后也不会被喝了。

假设前k-1轮喝了ans次,第k-1轮喝水t次,喝水的序列为a[1],a[2]…a[t]。
那么即是要找出a序列里所有不能在第k轮喝水的水缸,并一个个从序列里删除。

对于水缸a[i],若不满足(tim[a[i]]>=ans+i-1),(tim[a[i]]-i>=ans-1),则a[i]不能在第k轮喝水。

故要维护序列中的tim[a[i]]-i最小值,不断删除对应元素,直到Min(tim[a[i]]-i)>=ans-1。
显然删除了a[i]后,对于j>i的每个j,tim[a[j]]-j 要+1。

上面操作就一棵线段树可以解决了(Splay也是妥妥的)。
根据这题特点,删除应先删除位置小的,故推荐使用左子树优先的二叉查找(在树上走的时候一定要记得下放标记)。

#include <iostream>#include <fstream>#include <algorithm>#include <cmath>#include <ctime>#include <cstdio>#include <cstdlib>#include <cstring>using namespace std;#define mmst(a, b) memset(a, b, sizeof(a))#define mmcp(a, b) memcpy(a, b, sizeof(b))#define imin(a, b) (((a)<(b)) ? (a) : (b))typedef long long LL;const int N=400100;const LL oo=1e12;int n,m;LL ans,len;LL tim[N],a[N],x;LL t[4*N],lazy[4*N];void build(int ro,int l,int r){    if(l==r)    {        t[ro]=tim[l]-l;        return;    }    int mid=(l+r)/2;    build(ro*2,l,mid);    build(ro*2+1,mid+1,r);    t[ro]=imin(t[ro*2],t[ro*2+1]);}void down(int ro){    t[ro*2]+=lazy[ro];    t[ro*2+1]+=lazy[ro];    lazy[ro*2]+=lazy[ro];    lazy[ro*2+1]+=lazy[ro];    lazy[ro]=0;}void add(int ro,int l,int r,int zuo,int you){    if(l>=zuo&&r<=you)    {        t[ro]++;        lazy[ro]++;        return;    }    int mid=(l+r)/2;    down(ro);    if(zuo<=mid)    add(ro*2,l,mid,zuo,you);    if(you>mid)     add(ro*2+1,mid+1,r,zuo,you);    t[ro]=imin(t[ro*2],t[ro*2+1]);}void Cut(){    int tu=t[1],ro=1,l=1,r=n;    while(l!=r)    {        down(ro);        int mid=(l+r)/2;        if(t[ro*2]==tu)        {            ro=ro*2;            r=mid;        }        else        {            ro=ro*2+1;            l=mid+1;        }    }    t[ro]=oo;    while(ro!=1)    {        ro=ro/2;        t[ro]=imin(t[ro*2],t[ro*2+1]);    }    add(1,1,n,l,n);}int main(){    cin>>n>>m>>x;    for(int i=1;i<=n;i++)    scanf("%lld",&tim[i]);    for(int i=1;i<=n;i++)    scanf("%lld",&a[i]);    for(int i=1;i<=n;i++)    {        if(tim[i]>x)        tim[i]=0;        else        {            if(a[i]==0)            tim[i]=oo;            else            tim[i]=(x-tim[i])/a[i]+1;        }       }    build(1,1,n);    len=n;    for(int i=1;i<=m;i++)    {        LL tu=t[1];        while(tu<ans)        {            Cut();            len--;            tu=t[1];        }        ans+=len;    }    cout<<ans<<endl;    return 0;}
阅读全文
0 0