[CC-SUBWAY]Subway Ride

[CC-SUBWAY]Subway Ride

题目大意：

一棵\(n(n\le5\times10^5)\)个点的含重边的树，总边数为\(m(m\le10^6)\)，每条边有一个颜色。\(q(q\le5\times10^5)\)次询问，每次询问两个结点之间的路径（不经过重复结点）最多包含多少连续的同色段。

思路：

一个显然的结论是，对于连接相同两个点的重边，如果不同的颜色数量\(\le 3\)，那么至少有一条边和路径两侧的颜色不相同，这时我们只需要发明一种新的颜色来代替这些颜色即可。因此每条边最多只需要保留\(2\)种颜色。

用\(f[i][0/1]\)表示到了第\(i\)个结点，上一条边是第\(0/1\)种颜色的答案。每次询问跑一遍这样的DP，单次的时间复杂度是\(\mathcal O(n)\)。

使用树上倍增的思想，\(f[i][j][k][l]\)表示\(i\sim anc[i][j]\)的路径上，第一条边是第\(k\)种颜色，最后一条边是第\(l\)种颜色，询问时将一整段转移即可。

时间复杂度\(\mathcal O(q\log n)\)。

源代码：

#include<map>
#include<cstdio>
#include<cctype>
#include<vector>
inline int getint() {
	register char ch;
	while(!isdigit(ch=getchar()));
	register int x=ch^'0';
	while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
	return x;
}
const int N=5e5+1,logN=19;
int tot,anc[N][logN],dep[N],f[N][2],g[N][2];
int seg[N][logN][2][2],w[N][logN][2];
std::vector<int> e[N];
inline void add_edge(const int &u,const int &v) {
	e[u].push_back(v);
	e[v].push_back(u);
}
std::map<std::pair<int,int>,std::pair<int,int>> map;
inline int lg2(const float &x) {
	return ((unsigned&)x>>23&255)-127;
}
inline void up(int &x,const int &y) {
	x=std::max(x,y);
}
void dfs(const int &x,const int &par) {
	if(par) {
		int u=x,v=par;
		if(u>v) std::swap(u,v);
		const auto &p=map[std::make_pair(u,v)];
		if(p.first) {
			w[x][0][0]=p.first;
			seg[x][0][0][0]=1;
		}
		if(p.second) {
			w[x][0][1]=p.second;
			seg[x][0][1][1]=1;
		}
	}
	anc[x][0]=par;
	dep[x]=dep[par]+1;
	for(register int i=1;i<=lg2(dep[x]);i++) {
		anc[x][i]=anc[anc[x][i-1]][i-1];
		w[x][i][0]=w[anc[x][i-1]][i-1][0];
		w[x][i][1]=w[anc[x][i-1]][i-1][1];
		for(register int j=0;j<2;j++) {
			if(!w[x][0][j]) continue;
			for(register int k=0;k<2;k++) {
				if(!w[x][i][k]) continue;
				for(register int l=0;l<2;l++) {
					if(!w[x][i-1][l]) continue;
					for(register int m=0;m<2;m++) {
						if(!w[anc[x][i-1]][0][m]) continue;
						up(seg[x][i][j][k],seg[x][i-1][j][l]+seg[anc[x][i-1]][i-1][m][k]-(w[x][i-1][l]==w[anc[x][i-1]][0][m]));
					}
				}
			}
		}
	}
	for(auto &y:e[x]) {
		if(y!=par) dfs(y,x);
	}
}
inline int query(int u,int v) {
	if(u==v) return 0;
	if(dep[u]<dep[v]) std::swap(u,v);
	f[u][0]=f[u][1]=g[u][0]=g[u][1]=0;
	f[v][0]=f[v][1]=g[v][0]=g[v][1]=0;
	for(register int i=lg2(dep[u]-dep[v]);i>=0;i--) {
		const int &p=anc[u][i];
		if(dep[p]<dep[v]) continue;
		f[p][0]=f[p][1]=g[p][0]=g[p][1]=0;
		for(register int j=0;j<2;j++) {
			for(register int k=0;k<2;k++) {
				if(!w[u][0][k]) continue;
				for(register int l=0;l<2;l++) {
					if(f[u][j]+seg[u][i][k][l]-(g[u][j]==w[u][0][k])>f[p][l]) {
						f[p][l]=f[u][j]+seg[u][i][k][l]-(g[u][j]==w[u][0][k]);
						g[p][l]=w[u][i][l];
					}
				}
			}
		}
		u=p;
	}
	if(u==v) return std::max(f[u][0],f[u][1])-1;
	for(register int i=lg2(dep[u]);i>=0;i--) {
		const int &p=anc[u][i],&q=anc[v][i];
		if(p==q) continue;
		f[p][0]=f[p][1]=g[p][0]=g[p][1]=0;
		for(register int j=0;j<2;j++) {
			for(register int k=0;k<2;k++) {
				if(!w[u][0][k]) continue;
				for(register int l=0;l<2;l++) {
					if(f[u][j]+seg[u][i][k][l]-(g[u][j]==w[u][0][k])>f[p][l]) {
						f[p][l]=f[u][j]+seg[u][i][k][l]-(g[u][j]==w[u][0][k]);
						g[p][l]=w[u][i][l];
					}
				}
			}
		}
		f[q][0]=f[q][1]=g[q][0]=g[q][1]=0;
		for(register int j=0;j<2;j++) {
			for(register int k=0;k<2;k++) {
				if(!w[v][0][k]) continue;
				for(register int l=0;l<2;l++) {
					if(f[v][j]+seg[v][i][k][l]-(g[v][j]==w[v][0][k])>f[q][l]) {
						f[q][l]=f[v][j]+seg[v][i][k][l]-(g[v][j]==w[v][0][k]);
						g[q][l]=w[v][i][l];
					}
				}
			}
		}
		u=p;
		v=q;
	}
	int p=anc[u][0];
	f[p][0]=f[p][1]=g[p][0]=g[p][1]=0;
	for(register int j=0;j<2;j++) {
		for(register int k=0;k<2;k++) {
			if(!w[u][0][k]) continue;
			if(f[u][j]+seg[u][0][k][k]-(g[u][j]==w[u][0][k])>f[p][k]) {
				f[p][k]=f[u][j]+seg[u][0][k][k]-(g[u][j]==w[u][0][k]);
				g[p][k]=w[u][0][k];
			}
		}
	}
	p=0;
	f[p][0]=f[p][1]=g[p][0]=g[p][1]=0;
	for(register int j=0;j<2;j++) {
		for(register int k=0;k<2;k++) {
			if(!w[v][0][k]) continue;
			if(f[v][j]+seg[v][0][k][k]-(g[v][j]==w[v][0][k])>f[p][k]) {
				f[p][k]=f[v][j]+seg[v][0][k][k]-(g[v][j]==w[v][0][k]);
				g[p][k]=w[v][0][k];
			}
		}
	}
	int ans=0;
	for(register int i=0;i<2;i++) {
		for(register int j=0;j<2;j++) {
			up(ans,f[anc[u][0]][i]+f[0][j]-(g[anc[u][0]][i]==g[0][j]));
		}
	}
	return ans-1;
}
int main() {
	const int n=getint(),m=getint();
	for(register int i=0;i<m;i++) {
		int u=getint(),v=getint(),c=getint();
		if(u>v) std::swap(u,v);
		auto &p=map[std::make_pair(u,v)];
		if(p.first<0) continue;
		if(p.first==c||p.second==c) continue;
		if(p.first==0) {
			add_edge(u,v);
			p.first=c;
			continue;
		}
		if(p.second==0) {
			p.second=c;
			continue;
		}
		p.first=--tot;
		p.second=0;
	}
	dfs(1,0);
	const int q=getint();
	for(register int i=0;i<q;i++) {
		printf("%d\n",query(getint(),getint()));
	}
	return 0;
}