Find the answer （HDU - 6609，权值线段树）

一.题目链接：

HDU-6609

二.题目大意：

有 n 个数，到第 i 个数时，如果 sum > m，要求在 [1, i - 1] 中删除一些数，使得 sum ≤ m.

求最少删除数的个数.

三.分析：

首先，删除的话肯定是要从最大数开始删.

即：将数组顺序排列，从大往小删除.

离散化后的权值线段树则可以实现这一点.

即：查询从数组 [1, i - 1]中，在满足 sum < m 的情况下，最多选几个数.

注意：数有重复！！！

ps：这题貌似要用错误的格式才能 A 掉，不然 PE.

四.代码实现：

#include <set>
#include <map>
#include <ctime>
#include <queue>
#include <cmath>
#include <stack>
#include <bitset>
#include <vector>
#include <cstdio>
#include <sstream>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#define eps 1e-8
#define lc k * 2
#define rc k * 2 + 1
#define pi acos(-1.0)
#define ll long long int
using namespace std;

const int M = (int)2e5;
const ll mod = (ll)1e6 + 3;
const int inf = 0x3f3f3f3f;

int a[M + 5];
int b[M + 5];

struct node
{
    ll sum;
    int num;
} tree[M * 4 + 5];

int discrete(int n)
{
    memcpy(b, a, sizeof(a));
    sort(b + 1, b + n + 1);
    return unique(b + 1, b + n + 1) - (b + 1);
}

int tofind(int x, int len)
{
    return lower_bound(b + 1, b + len + 1, x) - b;
}

int query(int k, int l, int r, int v)
{
    if(l == r)
        return v / (tree[k].sum / tree[k].num);
    int mid = (l + r) >> 1;
    if(v <= tree[lc].sum)
        return query(lc, l, mid, v);
    else
        return tree[lc].num + query(rc, mid + 1, r, v - tree[lc].sum);
}

void update(int k, int l, int r, int pos, int v)
{
    tree[k].num++;
    tree[k].sum += v;
    if(l == r)
        return;
    int mid = (l + r) >> 1;
    if(pos <= mid)
        update(lc, l, mid, pos, v);
    else
        update(rc, mid + 1, r, pos, v);
}

int main()
{
    int q;
    scanf("%d", &q);
    int n, m, cnt;
    ll sum;
    while((q--) > 0)
    {
        memset(tree, 0, sizeof(tree));
        scanf("%d %d", &n, &m);
        for(int i = 1; i <= n; ++i)
            scanf("%d", &a[i]);
        int len = discrete(n);
        sum = 0;
        for(int i = 1; i <= n; ++i)
        {
            sum += a[i];
            cnt = (sum > m ? i - 1 - query(1, 1, n, m - a[i]) : 0);
            printf("%d ", cnt);
            update(1, 1, n, tofind(a[i], len), a[i]);
        }
        printf("\n");
    }
    return 0;
}