ACM-ICPC 2018 南京赛区网络预赛 G Lpl and Energy-saving Lamps（线段树）

链接：https://nanti.jisuanke.com/t/30996

题意：很简单，懒得讲了

思路：我们可以直接用线段树求出区间最左边小于某个数的数，线段树存1-n房间的灯数，维护区间最小值，查询的时候优先向左走就好了，

一个房间满了我们就用线段树将这个房间的值更新为inf，然后模拟下就好了。这样操作次数最多不大于1e5次，复杂度为 O(nlog(n))。

这道题算是一道非常基础的线段树，没什么复杂的操作，注意一次更新可能会有多个房间被填满，加个while处理下就好了。

之前那份代码忘了加个判断条件：当所有房间都被填满，那么值会不变，忘了加这个也能过数据有点水啊。

实现代码：

#include<bits/stdc++.h>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int inf = 1e9;
const int M = 2e5+10;
int sum[M<<2],n,k,q;
int a[M],b[M],x[M],y[M];

void update(int p,int c,int l,int r,int rt)
{
     if(l == r && l == p){
        sum[rt] = c;
        return ;
     }
     int m = (l + r) >> 1;
     if(p <= m) update(p,c,lson);
     else update(p,c,rson);
     sum[rt] = min(sum[rt<<1],sum[rt<<1|1]);
}

int query(int p,int l,int r,int rt)
{
    if(l == r) return l;
    int m = (l + r) >> 1;
    if(sum[rt<<1] <= p) return query(p,lson);
    if(sum[rt<<1|1] <= p) return query(p,rson);
    return -1;
}

int main()
{
    scanf("%d%d",&n,&k);
    for(int i = 1;i <= n;i ++){
        scanf("%id", &a[i]);
        update(i,a[i],1,n,1);
    }
    scanf("%d",&q);
    int mx = 0;
    for(int i = 1;i <= q;i++){
        scanf("%d",&b[i]);
        mx = max(b[i],mx);
    }
    int ans = 0, cnt = 0;
    for(int i = 1;i <= mx;i++){
        if(cnt == n){
            x[i] = cnt; y[i] = ans;
            continue;
        }
        else
        ans += k;
        while(1){   //一次更新可能有多个房间被填满
            int num = query(ans, 1, n, 1);
            if(num != -1) {
                cnt++, ans -= a[num];
                update(num, inf, 1, n, 1);
                x[i] = cnt, y[i] = ans;
            }
            else {
                x[i] = cnt, y[i] = ans;
                break;
            }
        }

    }
    for(int i = 1;i <= q;i++)
        printf("%d %d
",x[b[i]],y[b[i]]);

    return 0;
}