CF718D Andrew and Chemistry 题解
一道树哈希模板题。
由于度数很小,我们考虑直接搜索。
使用记搜,如果当前节点已经访问过就直接返回原来的值。
至于哈希的过程,我们考虑用 \(\text{vector}\) 来存储。
每个 \(\text{vector}\) 所存储的都是其儿子的哈希值,就可以不断递归向上合并。
为了减少常数,可以不选 \(\text{map}\) 采用 \(\text{pbds}\) 中 \(O(1)\) 的 \(\text{hash table}\)
Code
#include <bits/stdc++.h>
#include <bits/extc++.h>
using namespace std;
using namespace __gnu_pbds;
const int maxn = 100010;
int n , cnt , tot , du[maxn] , head[maxn];
set<int> q;
map<vector<int> , int> Hash;
gp_hash_table<int , int> mp[maxn];
struct edge
{
int to , nxt;
}e[maxn * 2];
inline int read()
{
int asd = 0 , qwe = 1; char zxc;
while(!isdigit(zxc = getchar())) if(zxc == '-') qwe = -1;
while(isdigit(zxc)) asd = asd * 10 + zxc - '0' , zxc = getchar();
return asd * qwe;
}
inline void add(int x , int y)
{
du[x]++ , du[y]++;
e[++cnt] = {y , head[x]} , head[x] = cnt;
e[++cnt] = {x , head[y]} , head[y] = cnt;
}
inline int dfs(int now , int fa)
{
if(mp[now][fa]) return mp[now][fa];
vector<int> tmp;
for(int i = head[now];i;i = e[i].nxt)
{
if(e[i].to == fa) continue;
tmp.push_back(dfs(e[i].to , now));
}
sort(tmp.begin() , tmp.end());
if(!Hash[tmp]) Hash[tmp] = ++tot;
return mp[now][fa] = Hash[tmp];
}
int main()
{
n = read();
for(int i = 1;i < n;i++)
{
int x = read() , y = read();
add(x , y);
}
for(int i = 1;i <= n;i++) if(du[i] < 4) q.insert(dfs(i , 0));
cout << q.size();
return 0;
}