版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_37632935/article/details/82353655
思路:因为只有1e5的数据。我们可以模拟一下所有情况全算出来,把答案存下来即可。但会一遍一遍的遍历超时,我们可以优化遍历的方式,因为只有1e5的房间,我们每次找的房间都是小于当前新灯泡数且最前面的房间。这个查找用一颗线段树既可以解决。
我们每次找到一个满足的房间就让当前灯泡是减去这个房间的灯泡数,然后让当前房间的灯泡数设成无穷大,更新一下线段树即可。
#include <iostream>
#include <string>
#include <string.h>
#include <algorithm>
#include <vector>
#define MAXN 1005
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;
int a[100005];
struct node{
int l,r,val;
}c[400005];
int cnt,n,m;
void build(int l,int r,int rt)
{
c[rt].l=l;c[rt].r=r;
if(l==r)
{
c[rt].val=a[l];
return ;
}
int mid=(l+r)>>1;
build(l,mid,rt<<1);
build(mid+1,r,rt<<1|1);
c[rt].val=min(c[rt<<1].val,c[rt<<1|1].val);
}
int now;
int query(int l,int r,int rt,int val)
{
if(l==r)
{
cnt++;
now-=c[rt].val;c[rt].val=INF;
return c[rt].l;
}
int mid=(l+r)>>1;
int ans=0;
if(c[rt<<1].val<=val)
{
ans=query(l,mid,rt<<1,val);
}
else if(c[rt<<1|1].val<=val)
{
ans=query(mid+1,r,rt<<1|1,val);
}
c[rt].val=min(c[rt<<1].val,c[rt<<1|1].val);
return ans;
}
int ans1[1000007],ans2[1000007];
int main()
{
cnt=0;
int t;
scanf("%d%d",&n,&m);
now=m;
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
}
build(1,n,1);
int q;scanf("%d",&q);
int i=1;
int qop=0x3f3f3f3f;
while(cnt!=n&&i<=100005)
{
int op=0;
int pos=query(1,n,1,now);
while(pos!=0)
{
op++;
pos=query(1,n,1,now);
}
if(cnt==n) qop=i;
ans1[i]=op+ans1[i-1];ans2[i]=now;
i++;now+=m;
}
while(q--)
{
int x;scanf("%d",&x);
if(x>qop)
printf("%d %d\n",ans1[qop],ans2[qop]);
else
printf("%d %d\n",ans1[x],ans2[x]);
}
return 0;
}