📑 题目:97. 交错字符串

🚀 本题 LeetCode 传送门

题目大意

给定三个字符串 s1、s2、s3,请你帮忙验证 s3 是否是由 s1 和 s2 交错 组成的。两个字符串 s 和 t 交错 的定义与过程如下,其中每个字符串都会被分割成若干 非空 子字符串:

  • s = s1 + s2 + … + sn
  • t = t1 + t2 + … + tm
  • |n - m| <= 1
  • 交错 是 s1 + t1 + s2 + t2 + s3 + t3 + … 或者 t1 + s1 + t2 + s2 + t3 + s3 + …

提示:a + b 意味着字符串 a 和 b 连接。

解题思路

  • 深搜或者广搜暴力解题。笔者用深搜实现的。记录 s1 和 s2 串当前比较的位置 p1 和 p2。如果 s3[p1+p2] 的位置上等于 s1[p1] 或者 s2[p2] 代表能匹配上,那么继续往后移动 p1 和 p2 相应的位置。因为是交错字符串,所以判断匹配的位置是 s3[p1+p2] 的位置。如果仅仅这么写,会超时,s1 和 s2 两个字符串重复交叉判断的位置太多了。需要加上记忆化搜索。可以用 visited[i][j] 这样的二维数组来记录是否搜索过了。笔者为了压缩空间,将 i 和 j 编码压缩到一维数组了。i * len(s3) + j 是唯一下标,所以可以用这种方式存储是否搜索过。具体代码见下面的实现。

代码

  1. package leetcode
  2. func isInterleave(s1 string, s2 string, s3 string) bool {
  3. if len(s1)+len(s2) != len(s3) {
  4. return false
  5. }
  6. visited := make(map[int]bool)
  7. return dfs(s1, s2, s3, 0, 0, visited)
  8. }
  9. func dfs(s1, s2, s3 string, p1, p2 int, visited map[int]bool) bool {
  10. if p1+p2 == len(s3) {
  11. return true
  12. }
  13. if _, ok := visited[(p1*len(s3))+p2]; ok {
  14. return false
  15. }
  16. visited[(p1*len(s3))+p2] = true
  17. var match1, match2 bool
  18. if p1 < len(s1) && s3[p1+p2] == s1[p1] {
  19. match1 = true
  20. }
  21. if p2 < len(s2) && s3[p1+p2] == s2[p2] {
  22. match2 = true
  23. }
  24. if match1 && match2 {
  25. return dfs(s1, s2, s3, p1+1, p2, visited) || dfs(s1, s2, s3, p1, p2+1, visited)
  26. } else if match1 {
  27. return dfs(s1, s2, s3, p1+1, p2, visited)
  28. } else if match2 {
  29. return dfs(s1, s2, s3, p1, p2+1, visited)
  30. } else {
  31. return false
  32. }
  33. }