Find the answer HDU - 6609 （二分+权值线段树+离散化）

题目链接

题意：给定序列 和正整数m，询问对于每个下标i ，至少需要让前面多少个数变成0才可以使得到前缀和pre[i]小于等于m.

思路：建权值线段树，维护桶内元素个数以及元素和。

我们可以二分变成0的元素个数，每次查询桶内取mid个数的最大和（假设为mmax），然后验证pre[i]-mmax<=m就可以。

PS：这题我一开始建树是动态开点，时间复杂度是O(nlogn*logINF) 没想到tle了 …

后面改成离散化建树，时间跑了2300ms左右

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 2e5+7;
const int mx = 4;
const int mod = 1e9+7;
const ll inf = 34359738370;
const int INF = 1e9+5;
//给定序列  询问对于每个下标i  至少需要使前面多少个数变成0 使得到i的前缀和小于等于m
//思路：建权值线段树 维护桶内元素和 元素个数 二分答案 每次优先贪心选择右子区间的数
int a[maxn];
ll pre[maxn];//前缀和
int n,m;
int h[maxn],size;//离散化  线段树区间是[1,size]
struct node
{
    ll sum;
    int tot;
}tree[maxn*mx];
inline int lc(int &rt) { return rt<<1;}
inline int rc(int &rt) { return rt<<1|1;}
inline void pushup(int &rt)
{
    tree[rt].sum=tree[lc(rt)].sum + tree[rc(rt)].sum;
    tree[rt].tot=tree[lc(rt)].tot + tree[rc(rt)].tot;
}
inline void updata(int rt,int l,int r,int pos)
{
    if(l == r) 
    {
        tree[rt].sum+=h[pos];
        tree[rt].tot++;
        return ;
    }
    int mid=(l+r)>>1;
    if(pos<=mid) updata(lc(rt),l,mid,pos);
    else updata(rc(rt),mid+1,r,pos);
    pushup(rt);
}
inline ll query(int rt,int l,int r,int k)
{
    if(!k) return 0;
    if(l == r) return h[l]*min(k,tree[rt].tot);//注意取min
    int mid=(l+r)>>1;
    if(k <= tree[rc(rt)].tot) return query(rc(rt),mid+1,r,k);
    else return query(lc(rt),l,mid,k-tree[rc(rt)].tot) + tree[rc(rt)].sum;
}
void init()
{
    pre[0]=0;
    memset(tree,0,sizeof(tree));
}
int main()
{
    int t;cin>>t;
    while(t--)
    {
        scanf("%d %d",&n,&m);
        init();
        for(int i=1;i<=n;i++)
        {
            scanf("%d",a+i);
            h[i]=a[i];
            pre[i]=pre[i-1]+a[i];
        }
        sort(h+1,h+n+1);
        size=unique(h+1,h+n+1)-h-1;
        for(int i=1;i<=n;i++)
        {
            int l=0,r=i-1,ans=0;
            while(l<=r)
            {
                int mid=(l+r)>>1;
                ll mmax=query(1,1,size,mid);//桶中取mid个数，最大的和能取到多少
                if(pre[i]-mmax <= m)
                {
                    ans=mid;
                    r=mid-1;
                }
                else l=mid+1;
            }
            printf("%d ",ans);
            int index=lower_bound(h+1,h+size+1,a[i])-h;
            updata(1,1,size,index);
        }
        printf("\n");
    }
}