【一次过】Lintcode 823. 输入流

原创于 2019-02-15 10:03:51 发布 · 129 阅读

0 ·

CC 4.0 BY-SA版权

lintcode 专栏收录该内容

393 篇文章

订阅专栏

本文介绍了一种使用栈处理退格符的输入流对比算法。通过将输入流转换为最终字符串并比较，解决两个输入流是否相等的问题。算法首先解析输入流，遇到退格符则从栈中弹出字符，最后将栈内字符逆序构成最终字符串进行对比。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给出两个输入流inputA和inputB(包含退格符),如果两个输入流最后的结果相等，输出YES，否则输出NO。

样例

样例1

输入: inputA = "abcde<<" 和 inputB = "abcd<e<"
输出: "YES"
解释:
inputA和inputB最后的结果都为"abc"，故返回"YES"。

样例2

输入: inputA = "a<<bc" 和 inputB = "abc<"
输出: "NO"
解释:
inputA最后的结果为"bc",inputB最后的结果为"ab"，故返回"NO"。

注意事项

输入字符只包括小写字母和'<'。
输入流长度不超过10000。

解题思路：

利用栈来操作退格。

public class Solution {
    /**
     * @param inputA: Input stream A
     * @param inputB: Input stream B
     * @return: The answer
     */
    public String inputStream(String inputA, String inputB) {
        // Write your code here
        String res1 = helper(inputA);
        String res2 = helper(inputB);
        
        if((res1 == null && res2 == null) || res1.equals(res2))
            return "YES";
        
        return "NO";
    }
    
    private String helper(String str){
        Stack<Character> stack = new Stack<>();
        char[] chars = str.toCharArray();
        
        for(int i=0; i<chars.length; i++){
            if(chars[i] == '<'){
                if(!stack.isEmpty())
                    stack.pop();
            }else{
                stack.push(chars[i]);
            }
        }
        
        StringBuilder sb = new StringBuilder();
        while(!stack.isEmpty()){
            sb.append(stack.pop());
        }
        
        return sb.toString();
    }
}