用于检查带退格的字符串是否相等的节省空间算法？

string algorithm java big-o space-complexity

2022-09-04 02:58:58

我最近在一次采访中被问到这个问题：

给定两个字符串 s 和 t，当两者都键入到空文本编辑器中时，如果它们相等，则返回。# 表示退格字符。

Input: S = "ab#c", T = "ad#c"
Output: true
Explanation: Both S and T become "ac".

我想出了以下解决方案，但它不节省空间：

  public static boolean sol(String s, String t) {
    return helper(s).equals(helper(t));
  }

  public static String helper(String s) {
    Stack<Character> stack = new Stack<>();
    for (char c : s.toCharArray()) {
      if (c != '#')
        stack.push(c);
      else if (!stack.empty())
        stack.pop();
    }
    return String.valueOf(stack);
  }

我想看看是否有任何更好的方法来解决这个问题，它不使用堆栈。我的意思是，我们可以在O（1）空间复杂度中解决它吗？

注意：我们也可以有多个退格字符。

答案 1

为了实现空间复杂性，请使用 Two Pointers 并从字符串的末尾开始：O(1)

public static boolean sol(String s, String t) {
    int i = s.length() - 1;
    int j = t.length() - 1;
    while (i >= 0 || j >= 0) {
        i = consume(s, i);
        j = consume(t, j);
        if (i >= 0 && j >= 0 && s.charAt(i) == t.charAt(j)) {
            i--;
            j--;
        } else {
            return i == -1 && j == -1;
        }
    }
    return true;
}

主要思想是保持计数器：如果字符是，则递增，否则递减它。如果和 - 跳过字符（递减位置）：#cnt#cnt > 0s.charAt(pos) != '#'

private static int consume(String s, int pos) {
    int cnt = 0;
    while (pos >= 0 && (s.charAt(pos) == '#' || cnt > 0)) {
        cnt += (s.charAt(pos) == '#') ? +1 : -1;
        pos--;
    }
    return pos;
}

时间复杂度：.O(n)

^{来源 1，来源 2。}

答案 2

更正了模板类型定义的伪代码

// Index of next spot to read from each string
let sIndex = s.length() - 1
let tIndex = t.length() - 1
let sSkip = 0
let tSkip = 0

while sIndex >= 0 and tIndex >= 0:
    if s[sIndex] = #:
        sIndex = sIndex - 1
        sSkip = sSkip + 1
        continue
    else if sSkip > 0
        sIndex = sIndex - 1
        sSkip = sSkip - 1
        continue

    // Do the same thing for t.
    if t[tIndex] = #:
        tIndex = tIndex - 1
        tSkip = tSkip + 1
        continue
    else if tSkip > 0
        tIndex = tIndex - 1
        tSkip = tSkip - 1
        continue

    // Compare characters.
    if s[sIndex] != t[tIndex], return false

    // Back up to the next character
    sIndex = sIndex - 1
    tIndex = tIndex - 1

// The strings match if we’ve exhausted all characters.
return sIndex < 0 and tIndex < 0