HDU - 6295 （暴力）

题意：给你n个点构成的树，每个点随机[1,n]的权值，问你树上存在多少回文串。

思路：权值是随机[1,n]。考虑下概率大胆猜测一下回文串长度肯定非常短。暴力一下长度小于等于3的数量就行了。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
vector<int> q[100005];
int a[100005];
int vis[100005];ll ans=0;
void dfs(int u,int fa)
{
	vis[a[u]]++;
	for(int i=0;i<q[u].size();i++)
	{
		int v=q[u][i];
		if(v==fa) continue;
		ans+=vis[a[v]];
		vis[a[v]]++;
		if(fa!=-1&&a[v]==a[fa]) ans++;
	}
	for(int i=0;i<q[u].size();i++)
	{
		int v=q[u][i];
		if(v==fa) continue;
		vis[a[v]]--;
	}
	vis[a[u]]--;
	for(int i=0;i<q[u].size();i++)
	{
		int v=q[u][i];
		if(v==fa) continue;
		dfs(v,u);
	}
}
int main()
{
	int t;scanf("%d",&t);
	while(t--)
	{
		int n;
		scanf("%d",&n);
		for(int i=1;i<=n;i++) q[i].clear();
		for(int i=1;i<=n;i++) scanf("%d",&a[i]);
		memset(vis,0,sizeof(vis));
		ans=0;
		for(int i=0;i<n-1;i++)
		{
			int x,y;
			scanf("%d%d",&x,&y);q[x].push_back(y);q[y].push_back(x);
		}
		dfs(1,-1);
		printf("%d\n",ans+n);
	}
	return 0;
}