1. 程式人生 > 其它 >(費用流)洛谷P4452 [國家集訓隊]航班安排

(費用流)洛谷P4452 [國家集訓隊]航班安排

技術標籤:圖論

洛谷P4452 [國家集訓隊]航班安排

思路:

又有貢獻,又有邊權,又有限制,一看就是網路流。
考慮如何建圖。
如果建圖的物件是機場,那還需要對時間進行分層,很複雜。那麼就以請求建圖,將請求拆成兩個點,出點和入點,出點是出發的時間,入點的到達的時間,兩點之間連一條流量為 1 1 1,費用為 c c c的邊,這樣控制了請求只有一次貢獻。
判斷從 0 0 0號機場飛到 a a a的時間滿不滿足,滿足可以建一個從源點到入點流量為 i n f inf inf,費用為 − f [ 0 ] [ a ] -f[0][a] f[0][a]的邊。
再建一個超級源點,連向源點,流量為 k k k,費用為 0 0

0的邊,限制 k k k次飛機。
判斷從 b b b能否按時返回 0 0 0號機場,滿足就可以建一條從出點向匯點流量為 i n f inf inf,費用為 − f [ b ] [ 0 ] -f[b][0] f[b][0]的邊。
還需要判斷兩個機場之間是否可以互達,如果可以,建一條流量為 i n f inf inf,費用為 − f [ b i ] [ a j ] -f[b_i][a_j] f[bi][aj]的邊。
由於題目已經說明 t i j ≤ t i k + t k j , f i j ≤ f i k + f k j t_{ij}\le t_{ik}+t_{kj},f_{ij}\le f_{ik}+f_{kj}
tijtik+tkj,fijfik+fkj
,所以兩點之間距離肯定最近,花費肯定最小,本身保證了最短路徑和最小費用。
這樣就可以得到我們要求的就是最大費用最大流。可以把SPFA改成求最長路;也可以把上面的權值全部取反,然後最後得到的答案取反。
為了方便處理,我將所有的機場編號都 + 1 +1 +1處理。

程式碼:

#include<bits/stdc++.h>
#define pii pair<int,int>
#define ll long long
#define cl(x,y) memset(x,y,sizeof(x))
#define loop(x,y,z) for(x=y;x<=z;x++)
#define reve(x,y,z) for(x=y;x>=z;x--) #define ct cerr<<"Time elapsed:"<<1.0*clock()/CLOCKS_PER_SEC<<"s.\n"; #define mp make_pair #define pb push_back #define fi first #define se second #define all(x) x.begin(),x.end() #define lson x<<1,l,mid #define rson x<<1|1,mid+1,r #define INF 1e18 const int NN=2e2+10; const int N=5e2+10; const int M=1e5+10; const int mod=1e9+7; const int inf=0x3f3f3f3f; const double eps=1e-8; const double pi=acos(-1); using namespace std; struct edge { int u,v,f,w; }e[M]; int head[N],len=1,dis[N],vis[N]; int maxflow,mincost; int pre[N],path[N]; int ti[NN][NN],d[NN][NN]; void add(int u,int v,int f,int w) { e[++len]={head[u],v,f,w}; head[u]=len; } void inc(int u,int v,int f,int w) { add(u,v,f,w); add(v,u,0,-w); } int spfa(int s,int t) { cl(dis,inf); cl(vis,0); cl(pre,-1); dis[s]=0; vis[s]=1; queue<int>q; q.push(s); while(!q.empty()) { int u=q.front(),i; q.pop(); vis[u]=0; for(i=head[u];i;i=e[i].u) { int v=e[i].v,w=e[i].w; if(e[i].f && dis[v]>dis[u]+w) { dis[v]=dis[u]+w; pre[v]=u; path[v]=i; if(!vis[v]) { q.push(v); vis[v]=1; } } } } return pre[t]!=-1; } void ek(int s,int t) { while(spfa(s,t)) { int mi=inf,i; for(i=t;i!=s;i=pre[i]) mi=min(mi,e[path[i]].f); for(i=t;i!=s;i=pre[i]) { e[path[i]].f-=mi; e[path[i]^1].f+=mi; } maxflow+=mi; mincost+=dis[t]*mi; } } struct node { int a,b,s,t,c; }p[N]; int main() { int n,m,i,j,k,T,s,t; scanf("%d%d%d%d",&n,&m,&k,&T); s=0; t=2*m+2; for(i=1;i<=n;i++) for(j=1;j<=n;j++) scanf("%d",&ti[i][j]); for(i=1;i<=n;i++) for(j=1;j<=n;j++) scanf("%d",&d[i][j]); for(i=1;i<=m;i++) { scanf("%d%d%d%d%d",&p[i].a,&p[i].b,&p[i].s,&p[i].t,&p[i].c); p[i].a++; p[i].b++; } for(i=1;i<=m;i++) { inc(i*2,i*2+1,1,-p[i].c); if(ti[1][p[i].a]<=p[i].s) inc(1,i*2,inf,d[1][p[i].a]); if(ti[p[i].b][1]+p[i].t<=T) inc(i*2+1,t,inf,d[p[i].b][1]); for(j=1;j<=m;j++) if(i!=j && p[i].t+ti[p[i].b][p[j].a]<=p[j].s) inc(i*2+1,j*2,inf,d[p[i].b][p[j].a]); } inc(s,1,k,0); ek(s,t); printf("%d",-mincost); return 0; }