HDU - 6437

题目链接:HDU - 6437


其他博客中基本上都是O(n+m*m)复杂度的连边。

但是其实可以优化到O(n+m)

我们对每种种类,分别看成两层图,然后第一种是第一类的,第二层是第二类的。如果选择某个节目,那么看完之后连向本身的下一个点则 -W 的费用,否则连向另一层点为 0 的费用。


AC代码:

#pragma GCC optimize("-Ofast","-funroll-all-loops")
#include<bits/stdc++.h>
//#define int long long
using namespace std;
const int inf=0x3f3f3f3f;
const int N=610,M=1e6+10;
int n,m,s,t,K,W,d[N],st[N],vis[N],cnt,ss;
int head[N],nex[M],to[M],w[M],flow[M],tot;
inline void ade(int a,int b,int c,int d){
	to[++tot]=b; nex[tot]=head[a]; w[tot]=d; flow[tot]=c; head[a]=tot;
}
inline void add(int a,int b,int c,int d){ade(a,b,c,d);	ade(b,a,0,-d);}
inline int spfa(){
	queue<int> q; q.push(s); memset(d,0x3f,sizeof d); d[s]=0; memset(st,0,sizeof st);
	while(q.size()){
		int u=q.front();	q.pop();	vis[u]=0;
		for(int i=head[u];i;i=nex[i])	if(flow[i]&&d[to[i]]>d[u]+w[i]){
			d[to[i]]=d[u]+w[i];	if(!vis[to[i]]) q.push(to[i]),vis[to[i]]=1;
		}
	}
	return d[t]<inf;
}
int dfs(int x,int f){
	if(x==t)	return f;
	int fl=0;	st[x]=1;
	for(int i=head[x];i&&f;i=nex[i]){
		if(!st[to[i]]&&flow[i]&&d[to[i]]==d[x]+w[i]){
			int mi=dfs(to[i],min(flow[i],f));
			flow[i]-=mi,flow[i^1]+=mi,fl+=mi,f-=mi;
		}
	}
	return fl;
}
inline int zkw(){
	int res=0;
	while(spfa())	if(d[t]<=0)	res-=dfs(s,inf)*d[t];
	return res;
}
void solve(){
	cin>>n>>m>>K>>W; tot=1; memset(head,0,sizeof head); t=N-5,ss=t-1; cnt=2*n;
	add(s,ss,K,0),add(ss,1,K,0),add(ss,n+1,K,0),add(n,t,K,0),add(2*n,t,K,0);
	for(int i=1;i<n;i++)	add(i,i+1,inf,0),add(i+n,i+n+1,inf,0);
	for(int i=1,a,b,c,op;i<=m;i++){
		cin>>a>>b>>c>>op;
		add(a+op*n,++cnt,1,-c),add(cnt,b+op*n,1,W),add(cnt,b+(!op)*n,1,0);
	}
	cout<<zkw()<<'\n';
}
signed main(){
	int T; cin>>T; while(T--) solve();
	return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_43826249/article/details/107721967
今日推荐