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

最新推荐文章于 2021-08-03 20:26:40 发布

原创最新推荐文章于 2021-08-03 20:26:40 发布 · 185 阅读

0 ·

CC 4.0 BY-SA版权

线段树同时被 2 个专栏收录

22 篇文章

订阅专栏

-------------数据结构-------------

14 篇文章

订阅专栏

本文详细解析了使用离散化权值线段树解决特定类型问题的算法思路，包括如何通过从大到小删除元素来保持序列和不超过预设阈值，同时最小化删除元素的数量。文章提供了完整的代码实现，并强调了处理重复元素的重要性。

一.题目链接：

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;
}