1. 程式人生 > >BZOJ 2561 最小生成樹

BZOJ 2561 最小生成樹

void sca open ems ostream cap sizeof 雙向 sap

第一眼瞎那啥貪心,然後覺得不太對勁,就滾去看題解,發現是網絡流OTZ

模擬Kruskal的過程發現,若<u,v>要在最小生成樹中出現,權值則小於<u,v>的邊不能讓u,v聯通,轉換成最小割模型,最大生成樹同理。

跑兩遍最大流。

註意邊要建雙向的啊,被這點坑死了。然後板子不要瞎那啥亂打。

順便發現果然ISAP跑得很快。

技術分享
//Twenty
#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include
<cstring> #include<queue> #include<vector> #define INF 0x3f3f3f3f typedef long long LL; using namespace std; const int maxn=1e6+299; int ecnt=1,ans,n,m,u,v,w,fir[maxn],pre[maxn]; struct Edge { int u,v,w,no,vv; Edge(){} Edge(int u,int v,int w):u(u),v(v),w(w){} }ee[maxn];
struct edge { int from,to,flow,cap,nxt; edge(){} edge(int from,int to,int cap,int flow,int nxt):from(from),to(to),cap(cap),flow(flow),nxt(nxt){} }e[maxn]; int d[maxn],c[maxn],cur[maxn]; void add(int from,int to,int cap) { e[++ecnt]=edge(from,to,cap,0,fir[from]); fir[from]=ecnt; e[
++ecnt]=edge(to,from,0,0,fir[to]); fir[to]=ecnt; } queue<int>que; void bfs(int s,int t) { for(int i=1;i<=n;i++) d[i]=n; d[t]=0; que.push(t); while(!que.empty()) { int x=que.front(); que.pop(); for(int i=fir[x];i;i=e[i].nxt) { edge tp=e[i]; if(d[e[i].to]==n&&e[i].flow==e[i].cap) { int to=e[i].to; d[to]=d[x]+1; que.push(to); } } } } int cal(int s,int t) { int fl=1e9; for(int i=t;i!=s;i=e[pre[i]].from) fl=min(fl,e[pre[i]].cap-e[pre[i]].flow); for(int i=t;i!=s;i=e[pre[i]].from) { e[pre[i]].flow+=fl; e[pre[i]^1].flow-=fl; } return fl; } int maxflow(int s,int t) { int res=0; bfs(s,t); for(int i=1;i<=n;i++) cur[i]=fir[i],c[d[i]]++; for(int x=s;d[x]<n;) { if(x==t) { res+=cal(s,t); x=s; } int ok=0; for(int &i=cur[x];i;i=e[i].nxt) if(d[e[i].to]+1==d[x]&&e[i].cap>e[i].flow) { pre[x=e[i].to]=i; ok=1; break; } if(!ok) { cur[x]=fir[x]; int M=n; for(int i=fir[x];i;i=e[i].nxt) { if(e[i].cap>e[i].flow&&(M>d[e[i].to]+1)) M=d[e[i].to]+1; } if(!(--c[d[x]])) break; c[d[x]=M]++; if(x!=s) x=e[pre[x]].from; } } return res; } int main() { //freopen(".in","r",stdin); //freopen(".out","w",stdout); scanf("%d%d",&n,&m); for(int i=1;i<=m;i++) { scanf("%d%d%d",&u,&v,&w); ee[i]=Edge(u,v,w); } scanf("%d%d%d",&u,&v,&w); for(int i=1;i<=m;i++) { if(ee[i].w<w) { add(ee[i].u,ee[i].v,1); add(ee[i].v,ee[i].u,1); } } ans+=maxflow(u,v); memset(fir,0,sizeof(fir)); memset(pre,0,sizeof(pre)); ecnt=1; for(int i=1;i<=m;i++) { if(ee[i].w>w) { add(ee[i].u,ee[i].v,1); add(ee[i].v,ee[i].u,1); } } ans+=maxflow(u,v); printf("%d\n",ans); return 0; }
View Code

BZOJ 2561 最小生成樹