[BZOJ1283]序列-【费用流】

题意

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

分析

利用最大流代表一个可行方案，考虑如下建图：

建立源、汇S,T。对序列每个元素建一个点ai

S向a1连一条容量为k的边，费用为0

Ai向ai+1连一条容量为k，费用为0（表示不选ai到a_(i+1)）

An向T连一条容量为k，费用为0

Ai向a_(i+m)连一条容量为1，费用为ai。（表示选择了ai）

Ai向T连一条容量为1，费用为ai（表示选择了ai）
可以证明，转换后的可行方案是原问题的可行方案

参考代码

#include
using namespace std;
#define in read()
#define ll long long 
#define db double 
#define pc(x) putchar(x)
#define gc getchar()
const int inf=0x7fffffff;
const int mod =1e9+7;
inline int read(){	int x=0,f=1;char c=gc;
	while((c<'0'||c>'9')&&(c!='-'))c=gc;if(c=='-')f=-1,c=gc;
	for(;c>='0'&&c<='9';c=gc)x=x*10+c-'0';	return x*f;
}
const int N=1e3+10;
struct Edge{
	int v,nxt,c,f;
}e[N*10];
int cnt=1,first[N];
int S,T,n,m,k;
inline void  add(int u,int v,int c,int f){
	e[++cnt].v=v;e[cnt].nxt=first[u];first[u]=cnt;e[cnt].c=c;e[cnt].f=f;
}
inline void ins(int u,int v,int c,int f){
	add(u,v,c,f);add(v,u,0,-f);
}
namespace D{
	int vis[N],dis[N];
	int spfa(int s,int t){
		memset(vis,0,sizeof(vis));
		memset(dis,-1,sizeof(dis));
		dis[t]=0;vis[t]=1;
		deque<int>q;
		q.push_back(t);
		while(!q.empty()){
			int now=q.front();q.pop_front();vis[now]=0;  
			for(int i=first[now];i;i=e[i].nxt){
			//	cerr<
				if(e[i^1].c && dis[now]-e[i].f > dis[e[i].v] ){
					dis[e[i].v]=dis[now]-e[i].f;
					if(!vis[e[i].v]){
						vis[e[i].v]=1;
						if(!q.empty()&&dis[e[i].v] > dis[q.front()])q.push_front(e[i].v);else q.push_back(e[i].v);
					}
				}
			}
		}
		return dis[s]!=-1;
	}
	int dfs(int x,int t,int f,int &ans){
		vis[x]=1; 
		if(x==t) return f; 
		int used=0,w; 
		for(int i=first[x];i;i=e[i].nxt){
			int v=e[i].v,c=e[i].c,ff=e[i].f;
			if(!vis[v] && c && dis[x]-ff==dis[v]){
				w=dfs(v,t,min(c,f-used) , ans);
				if(w)ans+= w*ff, e[i].c-=w;e[i^1].c+=w;used+=w;
				if(used==f)break;
			}
		}
		return used;
	}
	int flow(int s,int t,int &ans){
		int ret=0;
		while(spfa(s,t)){ 
			vis[t]=1;
			while(vis[t]){
				memset(vis,0,sizeof(vis));
				ret+=dfs(s,t,inf,ans);
			}
		}
		return ret;
	}
} 
int main(){
 	n=in;m=in;k=in;int ans=0;
 	S=0;T=n+1;
	ins(S,1,k,0); 
	for(int i=1;i<=n;i++){
		int x=in;
	 	ins(i,i+1,k,0);
		if(i<=n-m)ins(i,i+m,1,x);
		if(i>n-m)ins(i,T,1,x);
	}
	D::flow(S,T,ans);
	cout<<ans;
	return 0;
}