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

时间:2019-06-15 23:45:07

标签: java string algorithm big-o space-complexity

最近在一次采访中有人问我这个问题:

  

给出两个字符串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)空间复杂性吗?

注意:我们也可以包含多个退格字符。

2 个答案:

答案 0 :(得分:12)

为了实现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)

Source 1Source 2

答案 1 :(得分:2)

正确的templatetypedef伪代码

// 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