【leetcode】301 删除无效的括号(回溯)

题目链接:https://leetcode-cn.com/problems/remove-invalid-parentheses/

题目描述

删除最小数量的无效括号,使得输入的字符串有效,返回所有可能的结果。

说明: 输入可能包含了除 ( 和 ) 以外的字符。

示例 1:

输入: "()())()"
输出: ["()()()", "(())()"]

示例 2:

输入: "(a)())()"
输出: ["(a)()()", "(a())()"]

示例 3:

输入: ")("
输出: [""]

思路

class Solution {
public:
    vector<string> removeInvalidParentheses(string s) {
        /*** count-the-needed-to-delete-number ***/
        int left=0, right=0;    // 需要删除的左括号和右括号数量
        for(int i=0; i<s.size(); i++){
            if(s[i]=='(') left++;
            else if(s[i]==')'){
                if(left>0) left--;
                else right++;
            }
        }

        vector<string> result;
        string cur;
        help(result, cur, s, 0, 0, left, right);
        return result;
        
    }
    // pair代表可行解中需要()的对数
    void help(vector<string> &result, string& cur, const string& s, int pos, int pair, int left, int right){
        if(pos==s.size()){
            if(pair==0 && left==0 && right==0)  result.push_back(cur);
            return;
        }
        if(s[pos] == '('){
            /*** skip ***/
            if(cur.back() != s[pos] && left > 0)
                help(result, cur, s, pos+1, pair, left-1, right);
            /*** keep ***/
            help(result, cur += s[pos], s, pos+1, pair+1, left, right);
            cur.pop_back();
        }
        else if(s[pos] == ')'){
            /*** skip ***/
            if(cur.back() != s[pos] && right > 0)
                help(result, cur, s, pos+1, pair, left, right-1);
            /*** keep ***/
            /*** bug-places : I forget to check whether pair > 0 ***/
            if(pair>0){
                help(result, cur += s[pos], s, pos+1, pair-1, left, right);
                cur.pop_back();
            }
        }
        else{
            help(result, cur += s[pos], s, pos+1, pair, left, right);
            cur.pop_back();
        }
    }
};

【leetcode】301 删除无效的括号(回溯)_第1张图片

你可能感兴趣的:(LeetCode)