杭电多校第三场 1007 Find the answer（权值线段树）

                                            Find the answer

                                   Time Limit: 4000/4000 MS (Java/Others)    Memory Limit: 65536/65536 K (Java/Others)
                                                      Total Submission(s): 4521    Accepted Submission(s): 508

                                                             Statistic | Submit | Clarifications | Back

题意：

给定一个序列，对于每个数a[i] 求出最少删除 i 前面多少个数使得前缀和小于等于m 。

题解：

对于w[i]，先求一下前缀和sum，如果sum<=m, ans=0;

当sum>m时，我们就要删除1 ~ i-1的元素，为了减少删除的个数，优先删除大的元素，但是暴力不可以，所以构建权值线段树

对 w[i] 离散化，权值线段树维护权值线段树维护以i 为右端点的前缀和以及数的个数 每次查询最多可以取多少个数可以使得前缀和小于等于m，利用二分的思想在线段树上查找大的值优先所以优先查找右子树

/*
Niubility_n
*/
#include
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int maxn=2e5+10;

int a[maxn],b[maxn],n,w,cnt;
struct node
{
    int num;
    ll sum;
} Tree[maxn<<2];
void build(int l,int r,int root)
{
    Tree[root].sum=Tree[root].num=0;
    if(l==r)return ;
    int mid=(r+l)>>1;
    build(l,mid,root<<1);
    build(mid+1,r,root<<1|1);
}
void update(int x,int l,int r,int root)
{
    Tree[root].sum+=b[x];
    Tree[root].num++;
    if(l==r)return;
    int mid=(l+r)>>1;
    if(x<=mid)update(x,l,mid,root<<1);
    else update(x,mid+1,r,root<<1|1);
}
int query(ll sum,int l,int r,int root)
{
    if(sum == 0) return 0;
    if(l == r)
    {
        return (sum + b[l] - 1)/ b[l] ;//上取整，当前叶子节点的数目
    }
    int mid = (l+r) >> 1,ans = 0;
    if(sum <= Tree[root<<1|1].sum)
        ans += query(sum,mid+1,r,root<<1|1);
    else ans += Tree[root<<1|1].num + query(sum-Tree[root<<1|1].sum,l,mid,root<<1);

    return ans;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&w);
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&a[i]);
            b[i]=a[i];
        }
        sort(b+1,b+n+1);
        cnt=unique(b+1,b+n+1)-b-1;
        build(1,cnt,1);
        for(int i=1; i<=n; i++)
            a[i]=lower_bound(b+1,b+cnt+1,a[i])-b;
        ll sum=0;
        for(int i=1; i<=n; i++)
        {
            sum+=b[a[i]];
            int ans=0;
            if(sum>w)ans=query(sum-w,1,cnt,1);
            printf("%d ",ans);
            update(a[i],1,cnt,1);
        }
        printf("\n");
    }
}