1. 给定一个字符串数组,将字母异位词组合在一起。字母异位词指字母相同,但排列不同的字符串。
示例:
输入: [“eat”, “tea”, “tan”, “ate”, “nat”, “bat”],
输出:
[
[“ate”,“eat”,“tea”],
[“nat”,“tan”],
[“bat”]
]
说明:
所有输入均为小写字母。
不考虑答案输出的顺序。
import (
"sort";
"strings"
)
func groupAnagrams(strs []string) [][]string {
var ans [][]string
mymap := make(map[string]int)
for _, v := range strs {
orderedString := getOrderedString(v)
index, ok := mymap[orderedString]
if ok {
ans[index] = append(ans[index], v)
} else {
mymap[orderedString] = len(ans)
ans = append(ans, []string{v})
}
}
return ans
}
func getOrderedString(str string) string {
letters := strings.Split(str, "")
sort.Strings(letters)
return strings.Join(letters, "")
}
2. 返回与给定先序遍历 preorder 相匹配的二叉搜索树(binary search tree)的根结点。
(回想一下,二叉搜索树是二叉树的一种,其每个节点都满足以下规则,对于 node.left 的任何后代,值总 < node.val,而 node.right 的任何后代,值总 > node.val。此外,先序遍历首先显示节点的值,然后遍历 node.left,接着遍历 node.right。)
示例:
输入:[8,5,1,7,10,12]
输出:[8,5,10,1,7,null,12]
import (
"sort"
)
func bstFromPreorder(preorder []int) *TreeNode {
length := len(preorder)
if length == 0 {
return nil
}
array := preorder[1:]
Node := new(TreeNode)
val := preorder[0]
Node.Val = val
right := sort.Search(length - 1, func(i int) bool { return array[i] > val})
Node.Left = bstFromPreorder(array[:right])
Node.Right = bstFromPreorder(array[right:])
return Node
}
总结:
- go里面0和false不能隐式转换
func Search(n int, f func(int) bool) int
Search函数采用二分法搜索找到[0, n)区间内最小的满足f(i)==true的值i。也就是说,Search函数希望f在输入位于区间[0, n)的前面某部分(可以为空)时返回假,而在输入位于剩余至结尾的部分(可以为空)时返回真;Search函数会返回满足f(i)==true的最小值i。如果没有该值,函数会返回n。注意,未找到时的返回值不是-1,这一点和strings.Index等函数不同。Search函数只会用区间[0, n)内的值调用f。
一般使用Search找到值x在插入一个有序的、可索引的数据结构时,应插入的位置。这种情况下,参数f(通常是闭包)会捕捉应搜索的值和被查询的数据集。