BZOJ 1283 序列, 费用流,建图

题意:长度为n的序列a,第i个物品价值为a[i].选出一个子序列,满足任意一个长度为m的区间中,被子序列选中的个数<=k. 

n<=1e3,m,k<=100.问选出的子序列价值和的最大值.

若k==1, 显然可以直接dp[i]=max(dp[i-1],dp[i-m]+a[i]). .子问题为k==1的情况. (或者i->i+m连接(cap=1,cost=a[i])的边.)

可以证明,原问题的解集等价于:由k组互不相交的子问题的解组成的解集.

原问题的任意一个解P={b1,b2,...bm}都可以由k个子问题的解P0,p1,p2..pk-1组成.令pi={b[j]| j≡i (mod k)}.即可.

任意k组互不相交的子问题的解的并均为原问题的解.

当pi={c[1],c[2],..c[x]} 因为 c[i]-c[i-1]>=m 所以任意一个长度为m的区间[y,y+m-1]内包含p[i]的个数<=1.

令P=p0∪p1..∪pk-1 则任意[y,y+m-1]区间内,P的元素个数<=k.

现在将求原问题一个最优解 转化为 求k个最优不重叠的子问题的解

建图: 连接S->1->2->..i->i+1..->T. 容量为k,费用为0. 对i->i+m连接一条容量为1,费用为a[i]的边.(i+m>n 则连接i,T)的边.

这样建图,每组最大费用流对应着一个子问题解.求流量为k的最大费用流即可.

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> ii;
const int N=2e5+5,M=6e3+5,inf=0x3f3f3f3f;
int n,k,m,a[N],tot,s,t,head[N];
struct edge{
	int u,to,nxt,cap,cost;
	edge(){}
	edge(int u,int to,int nxt,int cap,int cost):u(u),to(to),nxt(nxt),cap(cap),cost(cost){}
}e[M];
void init(){
	tot=0;
	s=0,t=n+1;
	memset(head,-1,sizeof(head));
}
void add(int u,int v,int cap,int cost){
	e[tot]=edge(u,v,head[u],cap,cost);
	head[u]=tot++;
}
void Add(int u,int v,int cap,int cost){
	add(u,v,cap,cost);
	add(v,u,0,-cost);
}
int pre[N],inq[N],f[N],c[N];
int SPFA(){
	memset(pre,-1,sizeof(pre));
	queue<int> q;
	for(int i=0;i<=t;i++)	c[i]=inf,inq[i]=0,f[i]=0;
	q.push(s),c[s]=0,inq[s]=1,f[s]=inf;
	while(!q.empty()){
		int u=q.front();q.pop();
		inq[u]=0;
		//cout<<u<<' '<<c[u]<<'\n';
		for(int i=head[u];i!=-1;i=e[i].nxt){
			int v=e[i].to,w=e[i].cost;
			if(e[i].cap>0&&c[v]>c[u]+w){
				c[v]=c[u]+w;
				f[v]=min(f[u],e[i].cap);
				pre[v]=i;
				if(!inq[v])inq[v]=true,q.push(v);
			}
		}
	}
	return f[t];
}
int MCMF(){
	int res=0,tmp;
	while(tmp=SPFA()){
		for(int i=pre[t];i!=-1;i=pre[e[i].u]){
			int sum=tmp*e[i].cost;
			res+=sum;
			e[i].cap-=tmp,e[i^1].cap+=tmp;
		}
	}
	return res;
}
int main(){
	//ios::sync_with_stdio(false);cin.tie(0);
	cin>>n>>m>>k;
	init();
	Add(s,1,k,0);
	for(int i=1;i<=n;i++){
		cin>>a[i];
		if(i==n)	Add(i,t,k,0);
		else	Add(i,i+1,k,0);
		Add(i,min(i+m,t),1,-a[i]);
	}	
	int res=MCMF();
	cout<<-res<<'\n';
	return 0;
}