ACM-ICPC 2018 南京赛区网络预赛 Lpl and Energy-saving Lamps （线段树：无序数组找到第一个小于val）

题意：n个房间，每个房间有ai盏旧灯，每个月可以买m盏新灯，要求：按房间顺序换灯，如果剩下的新灯数目大于ai，那么进行更换，否则跳过该房间，判断下一个房间。如果所有房间都换完灯，那么久不会再买新灯。

q次询问，每次询问该月已换的房间数以及剩余的新灯数量。

题解：线段树记录最小值，然后一遍模拟

#include<bits/stdc++.h>

using namespace std ;

#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define INF 0x3f3f3f3f
typedef long long ll;
const int maxn = 1e5+;
int n,m;
int lamp[maxn << ];
int q;
int month[maxn];
int re;
int MIN[maxn << ];
int ans1[maxn], ans2[maxn];
void pushup(int rt)
{
    MIN[rt] = min(MIN[rt<<],MIN[rt<<|]);
}

void build(int l , int r , int rt)
{
    if(l == r)
    {
        scanf("%d",&MIN[rt]);
        return ;
    }

    int mid = (l+r) >>  ;
    build(lson);
    build(rson);
    pushup(rt);
}

int query(int l , int r , int rt , int val)
{
    if(l==r)
    {
        re-=MIN[rt];
        MIN[rt]=INF;
        return ;///找到了
    }
    int mid = (l + r) >> ;
    int pos =  ;
    if(MIN[rt<<] <= val) ///在左子树找到
    {
        pos = query(lson,val) ;
        pushup(rt);
        return pos;

    }
    if(MIN[rt<<|]<=val)///在右子树找到
    {
        pos = query(rson,val);
        pushup(rt);
        return pos;
    }
    return ;
}
int main ( )
{
    int n,m;
    scanf("%d%d",&n,&m);
    build(,n,);
    int Maxmonth = ;

    scanf("%d",&q);
    for(int i= ; i<=q ; i++)
    {
        scanf("%d",&month[i]);
        Maxmonth=max(month[i],Maxmonth);
    }
    int cur=;
    int cnt=;///换了多少
    int fa=;///是否换完
    while(cur<=Maxmonth)
    {
        if(!fa)
        re+=m;
        while(query(,n,,re)&&!fa)
        cnt++;

        if(!fa)
        {
            ans1[cur]=cnt;
            ans2[cur]=re;
        }
        else
        {
            ans1[cur]=n;
            ans2[cur]=re;
        }
        cur++;
        if(cnt>=n)
        fa=;
    }
    for(int i= ; i<=q ; i++)
    printf("%d %d\n",ans1[month[i]],ans2[month[i]]);

}