剑指offer 专项突破版 15、字符串中的所有变位词

题目链接

思路

• 主要思路和上道题目一样，唯一需要注意的就是只需要一个数组就可以了，因为我们在找到变位词后数组元素会全变为0，当我们的指针整体右移后，数组的内容会回到最初用变位词初始化后，所以一个数组即可

  public class Solution {
  
      boolean isContains(int[] nums) {
          for (int num : nums) {
              if (0 != num)
                  return false;
          }
          return true;
      }
  
      public List<Integer> findAnagrams(String s, String p) {
          List<Integer> result = new ArrayList<>();
          int[] nums = new int[26];
  
          if (s.length() < p.length())
              return result;
          for (int i = 0; i < p.length(); i++) {
              nums[s.charAt(i) - 'a']++;
              nums[p.charAt(i) - 'a']--;
          }
  
          if (isContains(nums)) {
              result.add(0);
          }
  
          for (int i = p.length(); i < s.length(); i++) {
              nums[s.charAt(i) - 'a']++;
              nums[s.charAt(i - p.length()) - 'a']--;
              if (isContains(nums)) {
                  result.add(i - p.length() + 1);
              }
          }
  
          return result;
      }
  }
  

Go代码

func findAnagrams(s string, p string) (result []int) {
	if len(s) < len(p) {
		return
	}
	nums := make([]int, 26)

	for i := 0; i < len(p); i++ {
		nums[p[i]-'a']--
		nums[s[i]-'a']++
	}
	if contains(nums) {
		result = append(result, 0)
	}
	offset := len(p)
	for i := len(p); i < len(s); i++ {
		oldCh, newCh := s[i-offset], s[i]
		nums[oldCh-'a']--
		nums[newCh-'a']++
		if contains(nums) {
			result = append(result, i-offset+1)
		}
	}
	return

}

func contains(nums []int) bool {
	for i := 0; i < len(nums); i++ {
		if 0 != nums[i] {
			return false
		}
	}
	return true
}