1. 程式人生 > >Codechef:Two Companies/TWOCOMP

Codechef:Two Companies/TWOCOMP

傳送門

題解: 想了半天看了一眼資料範圍。。 直接最大獨立集。

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;

const int RLEN=1<<18|1;
inline char nc() {
	static char ibuf[RLEN],*ib,*ob;
	(ib==ob) && (ob=(ib=ibuf)+fread(ibuf,1,RLEN,stdin));
	return (ib==ob) ? -1 : *ib++;
}
inline int rd() {
char ch=nc(); int i=0,f=1; while(!isdigit(ch)) {if(ch=='-')f=-1; ch=nc();} while(isdigit(ch)) {i=(i<<1)+(i<<3)+ch-'0'; ch=nc();} return i*f; } const int N=1e5+50, M=1e7+50; int n,m1,m2,sze[N],son[N],dep[N],top[N],fa[N],dfn[N],ind; struct chain {int x,y,l;} c1[N],c2[N]; vector <int>
e[N]; inline void dfs(int x,int f) { fa[x]=f; dep[x]=dep[f]+1; sze[x]=1; for(auto v:e[x]) if(v^f) { dfs(v,x); sze[x]+=sze[v]; if(sze[son[x]]<sze[v]) son[x]=v; } } inline void Dfs(int x,int f) { top[x]=f; dfn[x]=++ind; if(son[x]) Dfs(son[x],f); for(auto v:e[x]) if(v^fa[x] && v^son[
x]) Dfs(v,v); } inline int lca(int x,int y) { while(top[x]^top[y]) (dep[top[x]]>dep[top[y]]) ? (x=fa[top[x]]) : (y=fa[top[y]]); return (dep[x]>dep[y]) ? y : x; } inline bool in(int x,int y) {return (dfn[x]>=dfn[y]) && (dfn[x]<dfn[y]+sze[y]);} inline bool in(int a,int b,int c) {return in(b,a) && in(c,b);} namespace mcmf { int g[N],nt[M],vt[M],c[M],ec=1; int cur[N],lev[N],q[N],src,des; inline void add(int x,int y,int cc) { nt[++ec]=g[x]; g[x]=ec; vt[ec]=y; c[ec]=cc; nt[++ec]=g[y]; g[y]=ec; vt[ec]=x; c[ec]=0; } inline bool bfs() { for(int i=1;i<=des;i++) lev[i]=0; int r; q[r=1]=src; lev[src]=1; for(int i=1;i<=r;i++) { int u=q[i]; for(int e=g[u];e;e=nt[e]) if(c[e] && (!lev[vt[e]])) { lev[vt[e]]=lev[u]+1; q[++r]=vt[e]; if(vt[e]==des) return true; } } return false; } inline int dinic(int x,int f) { if(x==des) return f; int rs=0; for(int &e=cur[x];e;e=nt[e]) if(c[e] && (lev[vt[e]]==lev[x]+1)) { int o=dinic(vt[e],min(f-rs,c[e])); rs+=o; c[e]-=o; c[e^1]+=o; if(rs==f) return rs; } return lev[x]=0, rs; } inline int maxflow() { int rs=0; while(bfs()) { int t=0; memcpy(cur+1,g+1,sizeof(int)*des); while((t=dinic(src,2e9))) rs+=t, memcpy(cur+1,g+1,sizeof(int)*des); } return rs; } } int ans; int main() { n=rd(); m1=rd(); m2=rd(); for(int i=1;i<n;i++) { int x=rd(), y=rd(); e[x].push_back(y); e[y].push_back(x); } dfs(1,0); Dfs(1,1); mcmf::src=m1+m2+1; mcmf::des=m1+m2+2; for(int i=1;i<=m1;i++) { c1[i].x=rd(), c1[i].y=rd(), c1[i].l=lca(c1[i].x,c1[i].y); int v=rd(); ans+=v; mcmf::add(mcmf::src,i,v); } for(int i=1;i<=m2;i++) { c2[i].x=rd(), c2[i].y=rd(), c2[i].l=lca(c2[i].x,c2[i].y); int v=rd(); ans+=v; mcmf::add(i+m1,mcmf::des,v); } for(int i=1;i<=m1;i++) for(int j=1;j<=m2;j++) if(in(c1[i].l,c2[j].l,c1[i].x) || in(c1[i].l,c2[j].l,c1[i].y) || in(c2[j].l,c1[i].l,c2[j].x) || in(c2[j].l,c1[i].l,c2[j].y)) mcmf::add(i,j+m1,2e9); ans-=mcmf::maxflow(); cout<<ans<<'\n'; }