【BZOJ3651】网络通信 LCT

【BZOJ3651】网络通信

Description

   有一个由M 条电缆连接的 N 个站点组成的网络。为了防止垄断，由 C 个公司控制所有的电缆，规定任何公司不能控制连接同一个站点的两条以上的电缆（可以控制两条）。同时规定，每个公司不能有多余的电缆，所谓的多余，是指属于同一个公司的电缆不能形成环。  
在运作过程中，不同公司之间会进行电缆买卖。请你写一个程序判断买卖是否合法。 

Input

输入第一行有4个由空格隔开的整数 N，M，C和 T。N(1≤N≤ 8 000)表示站点数，M(0≤M≤100 000)表示连接站点的电缆数。C(1≤C≤ 100)表表示公司数量，T 表示电缆买卖次
数。后面有M行，每行三个整数Sj1,Sj2和Kj，表示连接站点Sj1和Sj2(1≤Sj1< Sj2  ≤ n)的电缆属于Kj(1≤Kj≤C)公司拥有，任意两个站点只有一条直接相连的电缆，输入状态合法。最后T(0≤T≤100 000)行，每行三个整数 Si1, Si2和  Ki,表示 Ki公司想购买站点Si1和Si2之间的电缆。 

Output

输出共 T行，表示处理的结果，有以下几种可能的结果： 
1、“No such cable.”  两个站点间没有电缆。 
2、 “Already owned.”  电缆己经是 Ki 公司控制。 
3、 “Forbidden: monopoly.” Ki 公司己经控制了两条连接 Si1  或  Si2 的电缆。 
4、 “Forbidden: redundant.” Ki 公司控制的线路会出现环。 
5、 “Sold.”  可以买卖。

Sample Input

4 5 3 5
1 2 1
2 3 1
3 4 2
1 4 2
1 3 3
1 2 3
1 2 3
1 4 3
2 3 3
2 4 3

Sample Output

Sold.
Already owned.
Forbidden: monopoly.
Forbidden: redundant.
No such cable.

题解：LCT模板题。用map维护每条边属于哪个公司，然后将每个点拆成100个（对应属于每个公司的情况），剩下的LCT即可。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <map>
#include <utility>
#define MP(A,B) make_pair(A,B)
#define p(A,B) ((A-1)*n+B)
using namespace std;
const int maxn=100010;
typedef pair<int,int> pii;
struct LCT
{
	int ch[2],rev,fa;
}s[maxn<<3];
int n,m,C,T;
map<pii,int> mp;
int d[maxn<<3];
inline bool isr(int x){return s[s[x].fa].ch[0]!=x&&s[s[x].fa].ch[1]!=x;}
inline void pushdown(int x)
{
	if(s[x].rev)
	{
		swap(s[x].ch[0],s[x].ch[1]);
		if(s[x].ch[0])	s[s[x].ch[0]].rev^=1;
		if(s[x].ch[1])	s[s[x].ch[1]].rev^=1;
		s[x].rev=0;
	}
}
inline void rotate(int x)
{
	int y=s[x].fa,z=s[y].fa,d=(x==s[y].ch[1]);
	if(!isr(y))	s[z].ch[y==s[z].ch[1]]=x;
	s[x].fa=z,s[y].fa=x,s[y].ch[d]=s[x].ch[d^1];
	if(s[x].ch[d^1])	s[s[x].ch[d^1]].fa=y;
	s[x].ch[d^1]=y;
}
void updata(int x)
{
	if(!isr(x))	updata(s[x].fa);
	pushdown(x);
}
inline void splay(int x)
{
	updata(x);
	while(!isr(x))
	{
		int y=s[x].fa,z=s[y].fa;
		if(!isr(y))
		{
			if((x==s[y].ch[0])^(y==s[z].ch[0]))	rotate(x);
			else	rotate(y);
		}
		rotate(x);
	}
}
inline void access(int x)
{
	for(int y=0;x;splay(x),s[x].ch[1]=y,y=x,x=s[x].fa);
}
inline void maker(int x)
{
	access(x),splay(x),s[x].rev^=1;
}
inline void link(int x,int y)
{
	maker(x),s[x].fa=y;
}
inline void cut(int x,int y)
{
	maker(x),access(y),splay(y),s[y].ch[0]=s[x].fa=0;
}
inline int findr(int x)
{
	while(s[x].fa)	x=s[x].fa;
	return x;
}
inline int rd()
{
	int ret=0,f=1;	char gc=getchar();
	while(gc<'0'||gc>'9')	{if(gc=='-')f=-f;	gc=getchar();}
	while(gc>='0'&&gc<='9')	ret=ret*10+gc-'0',gc=getchar();
	return ret*f;
}
int main()
{
	n=rd(),m=rd(),C=rd(),T=rd();
	int i,a,b,c,e;
	for(i=1;i<=m;i++)
	{
		a=rd(),b=rd(),c=rd();
		if(a>b)	swap(a,b);
		mp[MP(a,b)]=c,link(p(c,a),p(c,b)),d[p(c,a)]++,d[p(c,b)]++;
	}
	for(i=1;i<=T;i++)
	{
		a=rd(),b=rd(),c=rd(),e=mp[MP(a,b)];
		if(a>b)	swap(a,b);
		if(!e)	puts("No such cable.");
		else	if(e==c)	puts("Already owned.");
		else	if(d[p(c,a)]>=2||d[p(c,b)]>=2)	puts("Forbidden: monopoly.");
		else	if(findr(p(c,a))==findr(p(c,b)))	puts("Forbidden: redundant.");
		else
		{
			puts("Sold.");
			mp[MP(a,b)]=c,cut(p(e,a),p(e,b)),link(p(c,a),p(c,b)),d[p(e,a)]--,d[p(e,b)]--,d[p(c,a)]++,d[p(c,b)]++;
		}
	}
	return 0;
}