ZOJ Problem Set - 3635

题目大意

有n个从1..n标号的座位，按时间顺序给出每个客人来的时候是坐在第几个空座位，最后给若干个询问问第i号客人坐在哪里

分析

线段树+二分

  1 // Fast Sequence Operations II
  2 // Rujia Liu
  3 // 输入格式：
  4 // n m     数组范围是a[1]~a[n]，初始化为0。操作有m个
  5 // 1 L R v 表示设a[L]=a[L+1]=...=a[R] = v。其中v > 0
  6 // 2 L R  查询a[L]~a[R]的sum, min和max
  7 #include<cstdio>
  8 #include<cstring>
  9 #include<algorithm>
 10 using namespace std;
 11
 12 const int maxnode = 1<<17;
 13
 14 int _sum, _min, _max, op, qL, qR, v;
 15 int vist[maxnode];
 16 struct IntervalTree {
 17   int sumv[maxnode], minv[maxnode], maxv[maxnode], setv[maxnode];
 18
 19   // 维护信息
 20   void maintain(int o, int L, int R) {
 21     int lc = o*2, rc = o*2+1;
 22     if(R > L) {
 23       sumv[o] = sumv[lc] + sumv[rc];
 24       minv[o] = min(minv[lc], minv[rc]);
 25       maxv[o] = max(maxv[lc], maxv[rc]);
 26     }
 27     if(setv[o] >= 0) { minv[o] = maxv[o] = setv[o]; sumv[o] = setv[o] * (R-L+1); }
 28   }
 29
 30   // 标记传递
 31   void pushdown(int o) {
 32     int lc = o*2, rc = o*2+1;
 33     if(setv[o] >= 0) { //本结点有标记才传递。注意本题中set值非负，所以-1代表没有标记
 34       setv[lc] = setv[rc] = setv[o];
 35       setv[o] = -1; // 清除本结点标记
 36     }
 37   }
 38
 39   void update(int o, int L, int R) {
 40     int lc = o*2, rc = o*2+1;
 41     if(qL <= L && qR >= R) { // 标记修改
 42       setv[o] = v;
 43     } else {
 44       pushdown(o);
 45       int M = L + (R-L)/2;
 46       if(qL <= M) update(lc, L, M); else maintain(lc, L, M);
 47       if(qR > M) update(rc, M+1, R); else maintain(rc, M+1, R);
 48     }
 49     maintain(o, L, R);
 50   }
 51
 52   void query(int o, int L, int R) {
 53     if(setv[o] >= 0) { // 递归边界1：有set标记
 54       _sum += setv[o] * (min(R,qR)-max(L,qL)+1);
 55       _min = min(_min, setv[o]);
 56       _max = max(_max, setv[o]);
 57     } else if(qL <= L && qR >= R) { // 递归边界2：边界区间
 58       _sum += sumv[o]; // 此边界区间没有被任何set操作影响
 59       _min = min(_min, minv[o]);
 60       _max = max(_max, maxv[o]);
 61     } else { // 递归统计
 62       int M = L + (R-L)/2;
 63       if(qL <= M) query(o*2, L, M);
 64       if(qR > M) query(o*2+1, M+1, R);
 65     }
 66   }
 67 };
 68
 69 const int INF = 1000000000;
 70
 71 IntervalTree tree;
 72
 73 int main() {
 74   int n, m,num;
 75   while(scanf("%d", &n) !=EOF){
 76     memset(&tree, 0, sizeof(tree));
 77     memset(tree.setv, -1, sizeof(tree.setv));
 78     tree.setv[1] = 0;
 79     v=1;
 80     qL=1;
 81     qR=n;
 82     tree.update(1,1,n);
 83      tree.query(1,1,n);
 84     for(int i=1;i<=n;i++)
 85     {
 86         scanf("%d",&num);
 87          qL=1;qR=n;
 88         _sum = 0;
 89         int l=1,r=n,mid;
 90         while(l<r)
 91         {
 92              mid=(l+r)/2;
 93              qR=mid;
 94              tree.query(1, 1, n);
 95              if(_sum>=num)
 96                  r=mid;
 97              else
 98                  l=mid+1;
 99              _sum = 0;
100         }
101         vist[i]=l;
102         v=0;
103         qL=l;
104         qR=l;
105         tree.update(1,1,n);
106     }
107     scanf("%d",&m);
108     for(int i=1;i<=m;i++)
109     {
110         scanf("%d",&num);
111         printf("%d",vist[num]);
112         if(i!=m)
113           printf(" ");
114
115     }
116     printf("\n");
117   }
118   return 0;
119 }