HDU - 5406

题目链接:HDU - 5406


比较经典的费用流了。

但是直接暴力连边是会TLE的,queue可以改成stack能AC。

但是可以优化一些连边,比如 A->B , B->C ,A->C,这种,A->C的边就没有必要了。

我们可以通过B->C流过来,但是是不要费用的,所以拆点之后存在两边,一个费用为1,一个无费用为中转点。


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=2e3+10,M=1e6+10;
int n,s,s1,t,d[N],st[N],vis[N],cost;
int head[N],nex[M],w[M],flow[M],to[M],tot;
struct node{int h,d;}q[N];
int cmp(node a,node b){return a.h==b.h?a.d<b.d:a.h>b.h;}
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(){
	stack<int> q; q.push(s); memset(st,0,sizeof st); memset(d,inf,sizeof d); d[s]=0;
	while(q.size()){
		int u=q.top();	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 cost-=d[t]*f,f;
	st[x]=1;	int fl=0;
	for(int i=head[x];i&&f;i=nex[i]){
		if(flow[i]&&!st[to[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 void solve(){
	cin>>n;	t=n*2+1; s1=t+1; tot=1;	memset(head,0,sizeof head);
	for(int i=1;i<=n;i++)	scanf("%d %d",&q[i].h,&q[i].d);	
	sort(q+1,q+1+n,cmp);	add(s,s1,2,0);
	for(int i=1,mi;i<=n;i++){
		add(i,i+n,1,-1);	add(s1,i,1,0);	add(i+n,t,1,0);   add(i,i+n,1,0);
		mi=inf;
		for(int j=i+1;j<=n;j++)	if(q[j].d>=q[i].d&&q[j].d<mi){
			add(i+n,j,2,0);	mi=min(mi,q[j].d);
		}
	}
	cost=0;	while(spfa())	dfs(s,inf);
	printf("%d\n",cost);
}
signed main(){
	int T;	cin>>T;
	while(T--)	solve();
	return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章