【bzoj1283】【序列】【费用流】

Description

给出一个长度为 的正整数序列Ci，求一个子序列，使得原序列中任意长度为 的子串中被选出的元素不超过K(K,M<=100) 个，并且选出的元素之和最大。

Input

第1行三个数N,m,k。 接下来N行，每行一个字符串表示Ci。

Output

最大和。

Sample Input

10 5 3
4 4 4 6 6 6 6 6 4 4

Sample Output

30

HINT

20%的数据:n<=10。
100%的数据：N<=1000，k,m<=100。Ci<=20000。

题解：

         起点向第一个点连流量为k,费用为0的边.

         每一个点向下一个点连流量为k,费用为0的边.

         最后一个点向汇点连流量为k,费用为0的边.

         每个点x向x+m连流量为1,费用为这个点权值的边.

         如果x+m>n就像汇点连.

         跑最大费用最大流即可.

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#define N 2000
#define M 10000
#define inf 2100000000
using namespace std;
int point[N],cnt(1),next[M<<1],n,m,T,k,x;
int dis[N],pre[N],f[N],q[N*50],ans;
struct use{
   int st,en,v,c;
}e[M<<1];
void add(int x,int y,int v,int c){
   next[++cnt]=point[x];point[x]=cnt;
   e[cnt].st=x;e[cnt].en=y;e[cnt].v=v;e[cnt].c=c;
   next[++cnt]=point[y];point[y]=cnt;
   e[cnt].st=y;e[cnt].en=x;e[cnt].v=0;e[cnt].c=-c;
}
bool spfa(){
  memset(f,0,sizeof(f));
  for (int i=1;i<=T;i++) dis[i]=-inf;
  int h(0),t(1); 
  q[t]=1;dis[1]=0;f[1]=1;
  while (h<t){
   int u=q[++h];f[u]=0;
   for (int i=point[u];i;i=next[i])
     if (e[i].v&&(dis[e[i].en]<dis[u]+e[i].c)){
       dis[e[i].en]=dis[u]+e[i].c;
       pre[e[i].en]=i;
       if (!f[e[i].en]){
         f[e[i].en]=1;
         q[++t]=e[i].en;
       }
     }
  }
  return dis[T]!=-inf;
} 
void aug(){
  int mn=inf;
  for (int i=T;i!=1;i=e[pre[i]].st) mn=min(mn,e[pre[i]].v);
  for (int i=T;i!=1;i=e[pre[i]].st){
    ans+=e[pre[i]].c*mn;e[pre[i]].v-=mn;e[pre[i]^1].v+=mn;
  }
}
int main(){
  scanf("%d%d",&n,&k);m=n;n*=3;T=n+2;
  add(1,2,k,0);
  for (int i=1;i<=n;i++){
    scanf("%d",&x);
    add(i+1,i+2,k,0);
	if (i+m<=n) add(i+1,i+m+1,1,x);
	else add(i+1,T,1,x); 
  }	
  while (spfa()) aug();
  printf("%d\n",ans);
}