Question

2296. 设计一个文本编辑器

English Version

题目描述

请你设计一个带光标的文本编辑器，它可以实现以下功能：

• 添加：在光标所在处添加文本。
• 删除：在光标所在处删除文本（模拟键盘的删除键）。
• 移动：将光标往左或者往右移动。

当删除文本时，只有光标左边的字符会被删除。光标会留在文本内，也就是说任意时候 0 <= cursor.position <= currentText.length 都成立。

请你实现 TextEditor 类：

• TextEditor() 用空文本初始化对象。
• void addText(string text) 将 text 添加到光标所在位置。添加完后光标在 text 的右边。
• int deleteText(int k) 删除光标左边 k 个字符。返回实际删除的字符数目。
• string cursorLeft(int k) 将光标向左移动 k 次。返回移动后光标左边 min(10, len) 个字符，其中 len 是光标左边的字符数目。
• string cursorRight(int k) 将光标向右移动 k 次。返回移动后光标左边 min(10, len) 个字符，其中 len 是光标左边的字符数目。

 

示例 1：

输入：
["TextEditor", "addText", "deleteText", "addText", "cursorRight", "cursorLeft", "deleteText", "cursorLeft", "cursorRight"]
[[], ["leetcode"], [4], ["practice"], [3], [8], [10], [2], [6]]
输出：
[null, null, 4, null, "etpractice", "leet", 4, "", "practi"]

解释：
TextEditor textEditor = new TextEditor(); // 当前 text 为 "|" 。（'|' 字符表示光标）
textEditor.addText("leetcode"); // 当前文本为 "leetcode|" 。
textEditor.deleteText(4); // 返回 4
              // 当前文本为 "leet|" 。
              // 删除了 4 个字符。
textEditor.addText("practice"); // 当前文本为 "leetpractice|" 。
textEditor.cursorRight(3); // 返回 "etpractice"
               // 当前文本为 "leetpractice|". 
               // 光标无法移动到文本以外，所以无法移动。
               // "etpractice" 是光标左边的 10 个字符。
textEditor.cursorLeft(8); // 返回 "leet"
              // 当前文本为 "leet|practice" 。
              // "leet" 是光标左边的 min(10, 4) = 4 个字符。
textEditor.deleteText(10); // 返回 4
               // 当前文本为 "|practice" 。
               // 只有 4 个字符被删除了。
textEditor.cursorLeft(2); // 返回 ""
              // 当前文本为 "|practice" 。
              // 光标无法移动到文本以外，所以无法移动。
              // "" 是光标左边的 min(10, 0) = 0 个字符。
textEditor.cursorRight(6); // 返回 "practi"
               // 当前文本为 "practi|ce" 。
               // "practi" 是光标左边的 min(10, 6) = 6 个字符。

 

提示：

• 1 <= text.length, k <= 40
• text 只含有小写英文字母。
• 调用 addText ，deleteText ，cursorLeft 和 cursorRight 的 总 次数不超过 2 * 104 次。

 

进阶：你能设计并实现一个每次调用时间复杂度为 O(k) 的解决方案吗？

解法

方法一：左右栈

我们可以使用两个栈 left 和 right，其中栈 left 存储光标左边的字符，另一个栈 right 存储光标右边的字符。

• 当调用 addText 方法时，我们将 text 中的字符依次入栈 left。时间复杂度 $O(|text|)$。
• 当调用 deleteText 方法时，我们将 left 中的字符出栈最多 $k$ 次。时间复杂度 $O(k)$。
• 当调用 cursorLeft 方法时，我们将 left 中的字符出栈最多 $k$ 次，然后将出栈的字符依次入栈 right，最后返回 left 栈最多 $10$ 个字符。时间复杂度 $O(k)$。
• 当调用 cursorRight 方法时，我们将 right 中的字符出栈最多 $k$ 次，然后将出栈的字符依次入栈 left，最后返回 left 栈最多 $10$ 个字符。时间复杂度 $O(k)$。

class TextEditor:
  def __init__(self):
    self.left = []
    self.right = []

  def addText(self, text: str) -> None:
    self.left.extend(list(text))

  def deleteText(self, k: int) -> int:
    k = min(k, len(self.left))
    for _ in range(k):
      self.left.pop()
    return k

  def cursorLeft(self, k: int) -> str:
    k = min(k, len(self.left))
    for _ in range(k):
      self.right.append(self.left.pop())
    return ''.join(self.left[-10:])

  def cursorRight(self, k: int) -> str:
    k = min(k, len(self.right))
    for _ in range(k):
      self.left.append(self.right.pop())
    return ''.join(self.left[-10:])


# Your TextEditor object will be instantiated and called as such:
# obj = TextEditor()
# obj.addText(text)
# param_2 = obj.deleteText(k)
# param_3 = obj.cursorLeft(k)
# param_4 = obj.cursorRight(k)

class TextEditor {
  private StringBuilder left = new StringBuilder();
  private StringBuilder right = new StringBuilder();

  public TextEditor() {
  }

  public void addText(String text) {
    left.append(text);
  }

  public int deleteText(int k) {
    k = Math.min(k, left.length());
    left.setLength(left.length() - k);
    return k;
  }

  public String cursorLeft(int k) {
    k = Math.min(k, left.length());
    for (int i = 0; i < k; ++i) {
      right.append(left.charAt(left.length() - 1));
      left.deleteCharAt(left.length() - 1);
    }
    return left.substring(Math.max(left.length() - 10, 0));
  }

  public String cursorRight(int k) {
    k = Math.min(k, right.length());
    for (int i = 0; i < k; ++i) {
      left.append(right.charAt(right.length() - 1));
      right.deleteCharAt(right.length() - 1);
    }
    return left.substring(Math.max(left.length() - 10, 0));
  }
}

/**
 * Your TextEditor object will be instantiated and called as such:
 * TextEditor obj = new TextEditor();
 * obj.addText(text);
 * int param_2 = obj.deleteText(k);
 * String param_3 = obj.cursorLeft(k);
 * String param_4 = obj.cursorRight(k);
 */

class TextEditor {
public:
  TextEditor() {
  }

  void addText(string text) {
    left += text;
  }

  int deleteText(int k) {
    k = min(k, (int) left.size());
    left.resize(left.size() - k);
    return k;
  }

  string cursorLeft(int k) {
    k = min(k, (int) left.size());
    while (k--) {
      right += left.back();
      left.pop_back();
    }
    return left.substr(max(0, (int) left.size() - 10));
  }

  string cursorRight(int k) {
    k = min(k, (int) right.size());
    while (k--) {
      left += right.back();
      right.pop_back();
    }
    return left.substr(max(0, (int) left.size() - 10));
  }

private:
  string left, right;
};

/**
 * Your TextEditor object will be instantiated and called as such:
 * TextEditor* obj = new TextEditor();
 * obj->addText(text);
 * int param_2 = obj->deleteText(k);
 * string param_3 = obj->cursorLeft(k);
 * string param_4 = obj->cursorRight(k);
 */

type TextEditor struct {
  left, right []byte
}

func Constructor() TextEditor {
  return TextEditor{}
}

func (this *TextEditor) AddText(text string) {
  this.left = append(this.left, text...)
}

func (this *TextEditor) DeleteText(k int) int {
  k = min(k, len(this.left))
  if k < len(this.left) {
    this.left = this.left[:len(this.left)-k]
  } else {
    this.left = []byte{}
  }
  return k
}

func (this *TextEditor) CursorLeft(k int) string {
  k = min(k, len(this.left))
  for ; k > 0; k-- {
    this.right = append(this.right, this.left[len(this.left)-1])
    this.left = this.left[:len(this.left)-1]
  }
  return string(this.left[max(len(this.left)-10, 0):])
}

func (this *TextEditor) CursorRight(k int) string {
  k = min(k, len(this.right))
  for ; k > 0; k-- {
    this.left = append(this.left, this.right[len(this.right)-1])
    this.right = this.right[:len(this.right)-1]
  }
  return string(this.left[max(len(this.left)-10, 0):])
}

/**
 * Your TextEditor object will be instantiated and called as such:
 * obj := Constructor();
 * obj.AddText(text);
 * param_2 := obj.DeleteText(k);
 * param_3 := obj.CursorLeft(k);
 * param_4 := obj.CursorRight(k);
 */