Codeforces 948C Producing Snow(优先队列+思维)

题目链接：http://codeforces.com/contest/948/problem/C

题目大意：给定长度n(n<=1e5)，第一行v[i]表示表示第i堆雪的体积，第二行t[i]表示第1~i天的雪将要消融的体积，一堆雪如果消融到体积为0则消失，求每天消融的雪的体积。

解题思路：用优先队列，第i天就将v[i]+sum[i-1]放入优先队列中，然后初始消融量ans=t[i]*q.size()，假设每堆雪都够消融，然后根据优先队列找到q.top()<=sum[i]即不够消融的，减掉差值。

代码

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
#include<functional>
#include<algorithm>
using namespace std;
typedef long long LL;
const int N=1e5+5;

LL t[N],sum[N],v[N];                             //sum[i]是t[i]的前缀和

int main(){
    priority_queue<LL,vector<LL>,greater<LL> >q;
    int n,m;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%lld",&v[i]);

    }
    for(int i=1;i<=n;i++){
        scanf("%lld",&t[i]);
        sum[i]=t[i]+sum[i-1];
    }
    for(int i=1;i<=n;i++){
        q.push(v[i]+sum[i-1]);                  //加上sum[i-1]，可以在减的时候将前一段没有消融的sum[i-1]抵消掉
        LL ans=t[i]*q.size();                   //先假设没有一堆雪不够消融t[i]
        while(!q.empty()&&q.top()<=sum[i]){     //找到不够t[i]的，减掉差值（不够的部分）
            ans+=q.top()-sum[i];
            q.pop();
        }
        printf("%lld%c",ans,i==n?'
':' ');
    }
    return 0;
}