XDOJ_1144_排序+類比

標籤：stream   div   http   using   ace   names   for   php   class   

http://acm.xidian.edu.cn/problem.php?id=1144

 

不一定每一次都是合并k個，第一次先合并不足k的次數，後面都合并k個。

但是不知道為什麼我這樣寫的一直WA，後來在優先隊列中加上諾幹個0，使得每次都是合并k個就過了，很奇怪。

 

#include<iostream>#include<cstring>#include<cstdio>#include<algorithm>#include<queue>using namespace std;priority_queue< long long,vector<long long>,greater<long long> > q;int n,k;int main(){    while(~scanf("%d%d",&n,&k))    {        while(!q.empty())   q.pop();        long long x,ans = 0;        for(int i = 1;i <= n;i++)        {            scanf("%lld",&x);            q.push(x);        }        while((n-1)%(k-1))        {            q.push(0);            n++;        }        while(q.size() != 1)        {            long long t = 0,sum = 0;            while(t++ < k)            {                sum += q.top();                q.pop();            }            ans += sum;            q.push(sum);        }        printf("%lld\n",ans);    }    return 0;}

 

XDOJ_1144_排序+類比