leetcode算法题--树中可以形成回文的路径数

最新推荐文章于 2024-08-22 17:27:46 发布

bob62856

最新推荐文章于 2024-08-22 17:27:46 发布

阅读量255

点赞数

分类专栏： Algorithm 文章标签：算法 leetcode 职场和发展

本文链接：https://blog.csdn.net/qq_20817327/article/details/132001475

版权

Algorithm 专栏收录该内容

374 篇文章 3 订阅

订阅专栏

原题链接：https://leetcode.cn/problems/count-paths-that-can-form-a-palindrome-in-a-tree/

func countPalindromePaths(parent []int, s string) int64 {
	n := len(parent)
	g := make([][]int, n)
	for i := 1; i < n; i++ {
		p := parent[i]
		g[p] = append(g[p], i)
	}

	ans := 0
	cnt := map[int]int{0: 1} // 一条「空路径」
	var dfs func(int, int)
	dfs = func(v, xor int) {
		for _, w := range g[v] {
			x := xor ^ (1 << (s[w] - 'a'))
			ans += cnt[x] // x ^ x = 0
			for i := 0; i < 26; i++ {
				ans += cnt[x^(1<<i)] // x ^ (x^(1<<i)) = 1<<i
			}
			cnt[x]++
			dfs(w, x)
		}
	}
	dfs(0, 0)
	return int64(ans)
}