题目:
http://codeforces.com/contest/940/problem/E
题意:
给一串数,能把它分成连续的几段,每段有一个分值,求分值和最小能为多少;
分值计算方法:假设这段有k个数,则分值为除去前 个数(c为常数)之后的和。
分析:
考虑若有c+1个数,则可以把c个数分为1段,1个数分为另一段;
因为若c+1个数中的最小值在两头,则答案更优,若在中间,则答案一样。
c+2个数同理;
因此,可以让每段长度均为c或1;
用优先队列处理c个数的最小值,加以dp就能解题。
代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int tmax=1e5+5;
int n,c,data[tmax];
ll sum[tmax],f[tmax];
struct node{
int x,p;
node(int x=0,int p=0) :x(x),p(p) {}
bool operator < (const node b) const
{
if(x>b.x) return true;
if(x<b.x) return false;
return p>b.p;
}
};
priority_queue<node> Q;
int main()
{
cin>>n>>c;
int i;
for(i=1;i<=n;i++)
{
scanf("%d",&data[i]);
sum[i]=sum[i-1]+(ll)data[i];
}
for(i=1;i<=n;i++)
{
f[i]=f[i-1]+data[i];
Q.push(node(data[i],i));
while(!Q.empty()&&Q.top().p<=i-c) Q.pop();
if(i<c) continue;
f[i]=min(f[i],f[i-c]+sum[i]-sum[i-c]-Q.top().x);
}
printf("%I64d",f[n]);
return 0;
}