【解题报告】HDU 6099 Find the answer （离散化、树状数组、二分）

最新推荐文章于 2020-06-09 19:27:52 发布

Jiengup

最新推荐文章于 2020-06-09 19:27:52 发布

阅读量245

点赞数

分类专栏：题解文章标签：算法二分离散化

本文链接：https://blog.csdn.net/qq_42772300/article/details/98074287

版权

题解专栏收录该内容

25 篇文章 0 订阅

订阅专栏

原题地址

在这里插入图片描述

题目大意

给定一个长度为n的序列和一个整数m，要求你对这个序列里的每一个位置i输出一个值，值为在前1到i-1个数中最少将多少个数变成0之后满足所有的1到i个数加和小于等于m。

思路

首先很容易想到的是，我们每当不满足前1到i个数的和小于等于m时，我们要尽量取前1到i个数中较大的数变为0，这样可以使得我们取数的个数是最小的，换一种角度想，设第i个位置的值为a[i]，即我们要在前i-1个数中取尽量多的数让这些数加和小于等于m-a[i]。我们想到可以用二分解决这个问题，于是我们就想到离散化数据，然后二分答案，每次将a[i]放到它应该在的位置（即保证我们每次都取前面出现过的尽量小的数加起来），让其满足条件就行。

tips：这题提供了一种离散化解决问题的思路，很巧妙

//#include <bits/stdc++.h>
#include <iostream>
#include <assert.h>
#include <cstdio>
#include <algorithm>
#include <map>
#include <queue>
#include <cstring>
#include <stack>
#include <list>
#define lson rt<<1
#define rson rt<<1|1
const int maxn = 2e5+10;
const long long inf = 0x3f3f3f3f3f3f3f3f;
using namespace std;
typedef long long ll;
int n;
ll m;
ll a[maxn];
int b[maxn];
ll sum[maxn];
int num[maxn];
struct node
{
    ll d;
    int id;
    bool operator < (const node &b)const
    {
        if(d == b.d)
            return id < b.id;
        return d < b.d;
    }
}nd[maxn];

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

ll get_sum(int x)
{
    ll ret = 0;
    for(int i=x; i>=1; i-=lowbit(i))
    {
        ret += sum[i];
    }
    return ret;
}

int get_num(int x)
{
    int ret = 0;
    for(int i=x; i>=1; i-=lowbit(i))
    {
        ret += num[i];
    }
    return ret;
}

void add_sum(int x, ll v)
{
    for(int i=x; i<=n; i+=lowbit(i))
    {
        sum[i] += v;
    }
}

void add_num(int x)
{
    for(int i=x; i<=n; i+=lowbit(i))
    {
        num[i] += 1;
    }
}

int main()
{
    int t;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d %lld", &n, &m);
        for(int i=1; i<=n; i++)
        {
            scanf("%lld", &a[i]);
            nd[i].d = a[i];
            nd[i].id = i;
            num[i] = sum[i] = 0;
        }
        sort(nd+1, nd+n+1);
        for(int i=1; i<=n; i++)
        {
            b[nd[i].id] = i;
        }
        for(int i=1; i<=n; i++)
        {
            ll d = m - a[i];
            int l = 0, r = n;
            int ans = 0;
            while(l < r)
            {
                int mid = (l + r + 1) >> 1;
                if(get_sum(mid) <= d)
                {
                    l = mid;
                    ans = get_num(mid);
                }
                else
                    r = mid-1;
            }
            //cout<<l<<" !!!"<<endl;
            printf("%d ", i - ans-1);
            add_sum(b[i], a[i]);
            add_num(b[i]);
            //ut<<get_sum(n)<<" "<<get_num(n)<<endl;
        }
        printf("\n");
    }
    return 0;
}

/*
2
7 15
1 2 3 4 5 6 7
5 100
80 40 40 40 60




0 0 0 0 0 2 3
0 1 1 2 3
*/