比赛时碰到的,其实并不是什么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;
}