将字母异位词组合在一起

给你一个字符串数组,请你将 字母异位词 组合在一起。可以按任意顺序返回结果列表。

字母异位词 是由重新排列源单词的所有字母得到的一个新单词。
输入: strs = [“eat”, “tea”, “tan”, “ate”, “nat”, “bat”]
输出: [[“bat”],[“nat”,“tan”],[“ate”,“eat”,“tea”]]

输入: strs = [“”]
输出: [[“”]]

输入: strs = [“a”]
输出: [[“a”]]

package main 
import "fmt"

/*
给你一个字符串数组,请你将 字母异位词 组合在一起。可以按任意顺序返回结果列表。

字母异位词 是由重新排列源单词的所有字母得到的一个新单词。
输入: strs = ["eat", "tea", "tan", "ate", "nat", "bat"]
输出: [["bat"],["nat","tan"],["ate","eat","tea"]]

输入: strs = [""]
输出: [[""]]

输入: strs = ["a"]
输出: [["a"]]
*/

func BiJiao(inputStr string, inputStr1 string) bool {
   mapCnt := make(map[rune]int)
   for _, temp := range(inputStr) {
      mapCnt[temp]++
   }
   for _, temp2 := range(inputStr1) {
      if v, ok := mapCnt[temp2]; ok {
          if v > 0 {
             mapCnt[temp2]--
          } else {
             return false
          }
      } else {
         return false
      }
   }

   for _,v := range(mapCnt) {
      if v != 0 {
         return false
      }
   }
   return true
}

func GuiLei(inputStr []string) ([][]string) {
   var resultStr [][]string
    if len(inputStr) == 1 {
       resultStr = append(resultStr, inputStr)
       return resultStr
    }

    flag := make([]int , len(inputStr))
    for i:=0; i < len(inputStr); i++ {
       flag[i] = 0
    }
    mapStrVec := make(map[string][]string)
    for i:=0; i < len(inputStr); i++ {
       if flag[i] != 1 {
          mapStrVec[inputStr[i]] = []string {inputStr[i]}
          flag[i] =1
       } else {
          continue
       }
       
       for j := i+1; j < len(inputStr); j++ {
          if flag[j] == 1 {
             continue
          }
          if BiJiao(inputStr[i], inputStr[j]) {
             flag[j] = 1
             mapStrVec[inputStr[i]] = append(mapStrVec[inputStr[i]], inputStr[j])
             
          }
       }
    }

    for _, v := range mapStrVec {
      resultStr = append(resultStr, v)
    }
    return resultStr
}


func main() {
   fmt.Println(GuiLei([]string{"eat", "tea", "tan", "ate", "nat", "bat"}))
}

你可能感兴趣的:(go面试算法题,go面试算法题目)