DTOJ #3702. 月读（tsukuyomi）

【题目描述】

总而言之，月读现在有一棵大小为 $N$  ，树上每条边上有一个数字，月读有 $M$  次询问，每次询问一对 $(x,y)$  ，你需要回答从 $x$  到 $y$  的路径上的数字重新排列能否形成一个回文序列，若可行输出 $Yes$  ，否则输出 $No$  ，月读为了加快您的读入，每次询问的 $x,y$ 是通过某种方式生成的，为了加快您的输出，你只需要最后输出回答 $Yes$ 的个数和即可。

【输入格式】

第一行：两个整数 $N,M$

接下来 $N-1$ 行：三个整数 $u_i,v_i,w_i$，描述一条边 $(u_i,v_i)$ 与边上的数字 $w_i$ 。

接下来一行 $A_1,B_1$

$M$ 个询问以如下方式生成

$x_i=A_i \% n+1,y_i=B_i \% n+1$

$A_i=A_{i-1} \times 666073 \% (10^9+7)$

$B_i=B_{i-1} \times 233 \% 998244353$

【输出格式】

一行一个数字，代表回答 $Yes$ 的个数。

【样例】

样例输入

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

样例输出

1

【数据范围与提示】

对于此题有六个测试点：

A（10分） $N,M \le 2000$

B（23分） $N,M \le 10^5,u_i=v_i+1$

C（15分） $N,M \le 10^5$

D（15分） $N,M \le 10^6,w_i \le 30 $

E（17分） $N,M \le 10^6$

F（20分） $N \le 10^6,M \le 10^7$

对于所有的 $1 \le w_i \le n$

【题解】

回文序列一看就不是什么好东西。。。再认真读一遍题目，发现可以重新排列。那么这显然只需要统计每个数的出现次数是奇数还是偶数就可以了。显然这种判断奇偶性的可以用异或的方法。对于每个点存一下每种颜色的个数。询问时直接把两个节点的数组异或起来就可以了。这是最朴素的暴力。

考虑优化。维护数组，每个点只加入一次，树上主席树嘛！高兴地写着，一看数据范围：$M$ 有点小大。。。只能考虑 $O(1)$ 回答每次询问。

先从原来的思路想起。两个数组，维护的只有 $0$ 和 $1$，这显然可以用 $bitset$ 优化。

但还是过不去，让我们来考虑问题的本质：询问二进制位数。显然只有一位和 $0$ 的是合法的。我们考虑能不能只用一个数来代表一个 $bitset$。

幸运的是， $hash$ 显然可以解决这个问题。我们本来每一位的权值是 $2^{w_i}$，考虑改变底数为 $p^{w_i}$，其中 $p$ 是一个大质数。这样用自然溢出的 $hash$ 就可以把 $p$ 的 $1-n$ 次幂存下来，对于每个点求出树上前缀 $hash$ 异或和。这样对于每次询问的 $(x,y)$，只需判断 $x$ 与 $y$ 的 $hash$ 值的异或值是否为 $p^i$ 或 $0$ 即可。可以用 $hash$ 表维护。至此这题结束。

这个做法看起来真的很假。。。但我们要相信：数据是水的，出题人是懒的。。。于是信仰过了。

大质数可以采用神犇 $1124828077$ 推荐的 $13131313$，我试了试，果然没被卡 $hash$，还一遍过了。。。

【代码】

#include<bits/stdc++.h>
inline int read ( void )
{
	int x=0;char ch;bool f=false;
	while ( !isdigit(ch=getchar()) ) if ( ch=='-' ) f=true;
	for ( x=ch^48;isdigit(ch=getchar()); ) x=(x<<1)+(x<<3)+(ch^48);
	return f ? -x : x ;
}
int n,m,tot,cnt_e,h[1000010];
struct edge { int v,nxt,w; } e[2000010];
std::map<int,int> map;
inline void add ( int u,int v,int w )
{
	e[++cnt_e].nxt=h[u];e[h[u]=cnt_e].v=v;e[cnt_e].w=w;
	e[++cnt_e].nxt=h[v];e[h[v]=cnt_e].v=u;e[cnt_e].w=w;
}
typedef unsigned long long Hash;
const Hash HASH=13131313,hash_mod=10000019;
Hash Pow[1000010],hsh[1000010];
struct HASH_MAP
{
	int h[hash_mod+1],cnt;
	struct edge { Hash v;int nxt; } e[1000010];
	inline void insert ( Hash val ) { e[++cnt].nxt=h[val%hash_mod];e[h[val%hash_mod]=cnt].v=val; }
	inline bool find ( Hash val )
	{
		for ( int i=h[val%hash_mod];i;i=e[i].nxt ) if ( e[i].v==val ) return true;
		return false;
	}
}hash_map;
inline void dfs ( int u,int fr ) { for ( int i=h[u];i;i=e[i].nxt ) if ( e[i].v!=fr ) hsh[e[i].v]=hsh[u]^Pow[e[i].w],dfs(e[i].v,u); }
signed main()
{
	n=read();m=read();
	for ( int i=1,u,v,w;i<n;i++ )
	{
		u=read();v=read();w=read();
		if ( !map.count(w) ) map[w]=++tot;
		add(u,v,map[w]);
	}
	Pow[0]=1;
	for ( int i=1;i<=tot;i++ ) Pow[i]=Pow[i-1]*HASH,hash_map.insert(Pow[i]);
	dfs(1,0);
	int ans=0;
	for ( int A=read(),B=read();m--; )
	{
		int x=A%n+1,y=B%n+1;A=A*666073LL%1000000007;B=B*233LL%998244353;
		if ( !(hsh[x]^hsh[y]) or hash_map.find(hsh[x]^hsh[y]) ) ans++;
	}
	printf("%d\n",ans);
	return 0;
}