力扣 #22 括号生成（回溯法//深度遍历 && 剪枝）

#22 括号生成

数字 n 代表生成括号的对数，请你设计一个函数，用于能够生成所有可能的并且 有效的 括号组合。

示例：

输入：n = 3
输出：[
       "((()))",
       "(()())",
       "(())()",
       "()(())",
       "()()()"
     ]

思路：

• 生成括号就是典型的回溯算法。
• 也可以叫深度遍历。
• 搭配合理的剪枝
• 从空白字符串开始，递归地添加左括号或右括号。
• 如果左括号的数量大于n或者数量小于右括号的数量，剪枝。

代码：

LeetCode版代码

class Solution {
    public List<String> generateParenthesis(int n) {
        List<String> res = new ArrayList<String>();
        generate(res, "", 0, 0, n);
        return res;
    }
        //count1统计“(”的个数，count2统计“)”的个数
    public void generate(List<String> res , String ans, int leftCount, int rightCount, int n){
        //括号数量超过了n，结束递归。
        if(leftCount > n || rightCount > n) return;
        //如果左括号和右括号数量相等且都为n，将这个序列添加进结果，剪枝。
        if(leftCount == n && rightCount == n) {
            res.add(ans);
            return;
        }
        /*
        如果左括号数量不小于右括号数量，也隐含了剪枝。
        其实省略了下面这行代码
        if(leftCount < rightCOunt) return;
         */
        if(leftCount >= rightCount){
            //加一个左括号
            generate(res, ans+"(", leftCount+1, rightCount, n);
            //加一个右括号
            generate(res, ans+")", leftCount, rightCount+1, n);
        }
    }
}

idea版代码

import java.util.ArrayList;
import java.util.List;

public class number_22 {
    public static void main(String[] args) {
        List<String> l;
        l = generateParenthesis(1);
        System.out.println(l);
    }

    public static List<String> generateParenthesis(int n){
        List<String> l = new ArrayList<>();
        generate(l,"",0,0,n);
        return l;
    }

    public static void generate(List<String> res, String ans, int leftCount, int rightCount, int n){
        //如果左侧括号数量或右侧括号数量大于n，剪枝
        if(leftCount>n || rightCount>n) return;
        //如果括号匹配到n个，添加到结果中，剪枝。
        if(leftCount == n && rightCount == n) {
            res.add(ans);
            return;
        }
        
        /*
        左侧括号数量大于等于右侧括号数量，也隐含了剪枝。
        其实省略了下面这行代码
        if(leftCount < rightCOunt) return;
         */
        if(leftCount>=rightCount){
            generate(res,ans+"(",leftCount+1,rightCount,n);
            generate(res,ans+")",leftCount,rightCount+1,n);
        }
    }
}