咔咔咔过了!
这道题用的是单哈希映射,但是验证了15次
并且用unsigned long long 避免了乘法取模,他这个相当于是对ull的最大值取模,但是哈希映射要求是用素数取模,不是很理解,这个题没有用素数也可以过,
2
64
−
1
=
(
2
32
−
1
)
(
2
32
+
1
)
2^{64}-1=(2^{32}-1)(2^{32}+1)
264−1=(232−1)(232+1)
ull可以取64位1,ll只能取63位,最高符号位。
对于dp来说,就是在用换根,dp如果表示所有节点到本节点的距离的数量的哈希那么子节点和父节点的关系可以表示成为:
这里的数量表示每个距离出现的次数,比如说2出现了3次,那么在2上应该映射成3,然后去做哈希
代码:
#include<iostream>
#include<cstdio>
#include<random>
#include<vector>
#include<map>
using namespace std;
typedef unsigned long long ull;
const int length = 2e5 + 5;
vector<vector<int>> edge(length);
mt19937 R(1e9 + 7);
int V = 1e9 + 7;
int reco[length];
ull dp[length];
int base;
ull a = 0xffffffffffffffff;
long long b = 0xffffffffffffffff;
void dfs(int cur, int fa)
{
for (int j : edge[cur])
{
if (j != fa)
{
dfs(j, cur);
dp[cur] += dp[j]*base;
}
}
}
void dfs1(int cur, int fa)
{
for (int j : edge[cur])
{
if (j != fa)
{
dp[j] = (dp[cur] - dp[j]*base)*base + dp[j];
//这个时候的含义就是所有节点到j节点的距离
dfs1(j, cur);
}
}
}
int main(void)
{
int n;
scanf_s("%d", &n);
for (int i = 0; i < n-1; i++)
{
int a;
scanf_s("%d", &a);
reco[a]++;
}
for (int i = 0; i < n - 1; i++)
{
int a, b;
scanf_s("%d%d", &a, &b);
edge[a].push_back(b);
edge[b].push_back(a);
}
int t = 15;
vector<int> ans(n + 5, 0);
for (int i = 0; i < t; i++)
{
map<ull, int> mp;
base = R() % V + 1;
vector<ull> pw(n + 5, 0);
pw[0] = 1;
for (int i = 1; i <=n; i++)
{
pw[i] = pw[i - 1] * base;
}
ull tmp = 0;
for (int i = 0; i < n; i++)
{
tmp += reco[i] * pw[i];
}
for (int i = 0; i < n; i++)
{
ull ans = tmp + pw[i];
mp[ans] = 1;
}
for (int i = 1; i <= n; i++)
dp[i] = 1;
dfs(1, -1);
dfs1(1, -1);
for (int i = 1; i <= n; i++)
{
if (mp[dp[i]])
ans[i]++;
}
}
int res = 0;
for (int i = 1; i <= n; i++)
{
if (ans[i] == t)
res++;
}
printf("%d\n", res);
for (int i = 1; i <= n; i++)
{
if (ans[i] == t)
printf("%d ", i);
}
}