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';
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值