ACM-ICPC 2018 南京赛区网络预赛 G.Lpl and Energy-saving Lamps

链接

https://nanti.jisuanke.com/t/30996

题解

把所有的房间按照 k k <script type="math/tex" id="MathJax-Element-1">k</script>排序，每次只需要在一个前缀区间里面找序号最小的并把它删掉，然后拿数据结构维护下就好了

代码

#include <bits/stdc++.h>
#define maxn 100010
#define inf 0x3f3f3f3f
using namespace std;
int ans1[maxn], ans2[maxn], now, N, M, rt[maxn], belong[maxn], B, a[maxn], bmin[maxn], table[maxn];
struct room
{
    int id, k;
}rm[maxn];
int read(int x=0)
{
    int c, f=1;
    for(c=getchar();!isdigit(c);c=getchar())if(c=='-')f=-1;
    for(;isdigit(c);c=getchar())x=(x<<1)+(x<<3)+c-48;
    return f*x;
}
bool cmp(room a, room b){return a.k<b.k;}
void init()
{
    memset(bmin,inf,sizeof(bmin));
    memset(a,inf,sizeof(a));
    int i;
    N=read(), M=read(), B=sqrt(N);
    for(i=1;i<=N;i++)rm[i].k=read(), rm[i].id=i;
    sort(rm+1,rm+N+1,cmp);
    for(i=1;i<=N;i++)rt[rm[i].k]=i;
    for(i=1;i<=10000;i++)if(!rt[i])rt[i]=rt[i-1];
    for(i=1;i<=N;i++)a[i]=rm[i].id;
    for(i=1;i<=N;i++)belong[i]=i/B;
    for(i=1;i<=N;i++)bmin[belong[i]]=min(bmin[belong[i]],a[i]), table[a[i]]=i;
}
int qmin(int r)
{
    int i, ans=inf;
    for(i=0;i<belong[r];i++)ans=min(ans,bmin[i]);
    for(i=r;belong[i]==belong[r] and i;i--)ans=min(ans,a[i]);
    return ans;
}
void del(int pos)
{
    int i;
    a[pos]=inf;
    bmin[belong[pos]]=inf;
    for(i=max(1,belong[pos]*B);belong[i]==belong[pos] and i<=N;i++)bmin[belong[pos]]=min(bmin[belong[pos]],a[i]);
}
void work()
{
    int day, now=0, t;
    for(day=1;day<=100000;day++)
    {
        if(ans1[day-1]==N)
        {
            ans1[day]=ans1[day-1];
            ans2[day]=ans2[day-1];
            continue;
        }
        now+=M;
        ans1[day]=ans1[day-1];
        t=qmin(rt[min(rm[N].k,now)]);
        while(t!=inf)
        {
            now-=rm[table[t]].k;
            ans1[day]++;
            del(table[t]);
            t=qmin(rt[min(rm[N].k,now)]);
        }
        ans2[day]=now;
    }
}
void show()
{
    int q=read(), x;
    while(q--)
    {
        x=read();
        printf("%d %d\n",ans1[x],ans2[x]);
    }
}
int main()
{
    init();
    work();
    show();
    return 0;
}