代码拉取完成,页面将自动刷新
#include <bits/stdc++.h>
#define eb emplace_back
#define ep emplace
#define fi first
#define se second
#define in read<int>()
#define lin read<ll>()
#define rep(i, x, y) for(int i = (x); i <= (y); i++)
#define per(i, x, y) for(int i = (x); i >= (y); i--)
using namespace std;
using ll = long long;
using db = double;
using pii = pair < int, int >;
using pll = pair < ll, ll >;
using vec = vector < int >;
using veg = vector < pii >;
template < typename T > T read() {
T x = 0; bool f = 0; char ch = getchar();
while(!isdigit(ch)) f |= ch == '-', ch = getchar();
while(isdigit(ch)) x = x * 10 + (ch ^ 48), ch = getchar();
return f ? -x : x;
}
template < typename T > void chkmax(T &x, const T &y) { x = x > y ? x : y; }
template < typename T > void chkmin(T &x, const T &y) { x = x < y ? x : y; }
const int N = 2e5 + 10;
const ll base = 13331;
const int mod1 = 1e9 + 7;
const int mod2 = 1e9 + 9;
pll hsh[N], pw[N], thsh[N], vhsh[N], pre[N], suf[N];
pll operator * (pll a, ll b) { return pll(a.fi * b % mod1, a.se * b % mod2); }
pll operator * (pll a, pll b) { return pll(a.fi * b.fi % mod1, a.se * b.se % mod2); }
pll operator + (pll a, pll b) { return pll((a.fi + b.fi) % mod1, (a.se + b.se) % mod2); }
pll operator + (pll a, ll b) { return pll((a.fi + b) % mod1, (a.se + b) % mod2); }
int n, siz[N], tsiz[N], psiz[N], ssiz[N];
vec G[N];
void dfs(int x, int p) {
vector < pair < pll, int > > pot; siz[x] = 0;
for(auto y : G[x]) if(y ^ p) {
dfs(y, x); siz[x] += siz[y]; pot.eb(hsh[y], siz[y]);
} sort(pot.begin(), pot.end());
hsh[x] = { '(', '(' }; for(auto v : pot) hsh[x] = (hsh[x] * pw[v.se] + v.fi);
hsh[x] = (hsh[x] * base + pii{ ')', ')' });
siz[x] += 2; thsh[x] = hsh[x]; tsiz[x] = siz[x];
}
int stk[N], top;
vector < pll > alres;
void rdfs(int x, int p) {
top = 0; for(auto y : G[x]) stk[++top] = y;
sort(stk + 1, stk + top + 1, [](auto a, auto b) { return pair < pll, int >{ thsh[a], tsiz[a] } < pair < pll, int >{ thsh[b], tsiz[b] }; });
pre[0] = { '(', '(' }; suf[top + 1] = { ')', ')' }; psiz[0] = ssiz[top + 1] = 1;
rep(i, 1, top) psiz[i] = psiz[i - 1] + tsiz[stk[i]];
per(i, top, 1) ssiz[i] = ssiz[i + 1] + tsiz[stk[i]];
rep(i, 1, top) pre[i] = (pre[i - 1] * pw[tsiz[stk[i]]] + thsh[stk[i]]);
per(i, top, 1) suf[i] = (suf[i + 1] + pw[ssiz[i + 1]] * thsh[stk[i]]);
if(G[x].size() < 4) {
pll tot = { '(', ')' };
rep(i, 1, top) {
int y = stk[i];
tot = (tot * pw[tsiz[y]] + thsh[y]);
} tot = (tot * base + pll{ ')', ')' }); alres.eb(tot);
//cerr << x << " " << tot << " " << hsh[x] << endl;
}
rep(i, 1, top) if(stk[i] ^ p) vhsh[stk[i]] = (pre[i - 1] * pw[ssiz[i + 1]] + suf[i + 1]);
for(auto y : G[x]) if(y ^ p) {
thsh[x] = vhsh[y]; tsiz[x] = n * 2 - siz[y];
rdfs(y, x);
}
thsh[x] = hsh[x]; tsiz[x] = siz[x];
}
int main() {
#ifndef ONLINE_JUDGE
freopen("1.in", "r", stdin); freopen("CF718D.out", "w", stdout);
#endif
n = in; rep(i, 2, n) { int u = in, v = in; G[u].eb(v), G[v].eb(u); }
pw[0] = { 1, 1 }; rep(i, 1, n << 1) pw[i] = pw[i - 1] * base;
dfs(1, 0); rdfs(1, 0); sort(alres.begin(), alres.end());
alres.resize(unique(alres.begin(), alres.end()) - alres.begin()); printf("%lu\n", alres.size());
return 0;
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。