UOJ#271. 【清华集训2016】连通子树（虚树+倍增）

传送门

题解：
注意到每种颜色个数比较少，于是建出虚树后暴力背包，用倍增维护一下虚链上的DP值即可。

所以说你只需要5个倍增数组和一些卡常技巧加上无数的小细节就可以通过这道题了。 不说了我去睡觉了。

#include <bits/stdc++.h>
using namespace std;
typedef pair <int,int> pii;
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;
} 
inline void W(int x) {
   
	static int buf[50];
	if(!x) {
   putchar('0'); return;}
	if(x<0) {
   putchar('-'); x=-x;}
	while(x) {
   buf[++buf[0]]=x%10; x/=10;}
	while(buf[0]) putchar(buf[buf[0]--]+'0');
}

const int N=1e5+50, mod=1e9+7, L=18;
inline int add(int x,int y) {
   return (x+y>=mod) ? (x+y-mod) : (x+y);}
inline int dec(int x,int y) {
   return (x-y<0) ? (x-y+mod) : (x-y);}
inline int mul(int x,int y) {
   return (long long)x*y%mod;}
inline int power(int a,int b,int rs=1) {
   for(;b;b>>=1,a=mul(a,a)) if(b&1) rs=mul(rs,a); return rs;}
inline int cinv(int x) {
   return power(x,mod-2);}
int n,Q,col[N],sum[N],f[100][6][6][6];
vector <int> edge[N];
vector <int> cont[N];
int tot,dep[N],son[N],sze[N],top[N],fa[N],dp[N],up[N][L+1],dfn[N],ind;
int dp1[N][L+1],dp2[N][L+1],dp3[N][L+1],dp4[N][L+1],dp5[N][L+1];
int vir[N],stk[N],pos[N],cnt,stk_top;
vector <int> vire[N];
int c0,c1,c2,s0,s1,s2,ans;

inline void dfs(int x,int f) {
   
	sze