HDU 4217

点击打开题目链接

题型就是数据结构。给一个数组，然后又ｋ次操作，每次操作给定一个数ki, 从数组中删除第ki小的数，要求的是ｋ次操作之后被删除的所有的数字的和。　

简单的思路就是，用１标记该数没有被删除，０表示已经被删除，对于找到第ki小的数, 只需要找到标记数组中第一个前缀和为ki的下标，又因为用来标记的数组的前缀和是不减数列，所以可以用二分来加速。这里值得注意的是，被删除后的数，不会第二次或者多次被找到，即每个数最多被找到一次，因为如果该数被删除了，而且该数所在下标的前缀和是ki，那么一定还存在一个更小的下标，使得它的前缀和也是ki,　而我们要找的就是第一次出现前缀和为ki的下标。还需要使用I64.

附上代码：

 /*************************************************************************
     > File Name: 4217.cpp
     > Author: Stomach_ache
     > Mail: sudaweitong@gmail.com
     > Created Time: 2014年04月26日 星期六 21时51分19秒
     > Propose: HDU 4217
  ************************************************************************/
 //BIT + BinarySearch　复杂度　O(k * logn * logn)
 //单点更新，区间求值, 用１表示该数没有被删除，０表示该数已经被删除
 #include <cmath>
 #include <string>
 #include <vector>
 #include <cstdio>
 #include <fstream>
 #include <cstring>
 #include <iostream>
 #include <algorithm>
 using namespace std;

 #define X first
 #define Y second
 #define MAX_N (262144 + 5)
 typedef long long LL;
 typedef pair<int, int> pii;
 int n, k;
 int c[MAX_N];

 int
 lowbit(int x) {
         return x & (-x);
 }

 LL
 get_sum(int x) {
         LL s = ;
         while (x > ) {
                 s += c[x];
                 x -= lowbit(x);
         }

         return s;
 }

 void
 update(int x, int v) {
         while (x <= n) {
                 c[x] += v;
                 x += lowbit(x);
         }

         return ;
 }

 int
 main(void) {
         int T, cnt = ;
         scanf("%d", &T);
         while (T--) {
                 scanf("%d %d", &n, &k);
                 memset(c, , sizeof(c));
                 for (int i = ; i <= n; i++) {
                         update(i, );
                 }
                 LL ans = ;
                 for (int i = ; i < k; i++) {
                         int ki;
                         scanf("%d", &ki);
                         //只需要找到前缀和为ki对应的数，也就是第ki小的数
                         //此处为不减数列，所以使用二分来找到第一个前缀和为ki对应的数
                         int L = ki, R = n, tmp = L;
                         while (L <= R) {
                                 int mid = L + (R - L) / ;
                                 int s = get_sum(mid);
                                 if (s < ki) {
                                         L = mid + ;
                                 } else {
                                         if (s == ki) {
                                                 tmp = mid;
                                         }
                                         R = mid;
                                 }
                                 if (tmp == L && tmp == R) {
                                         break;
                                 }
                         }
                         ans += tmp;
                         update(tmp, -);
                 }
                 printf("Case %d: %I64d\n", cnt++, ans);
         }

         return ;
 }