Jerry的考验

题目详情

---

有一天Jerry给Tom出了一道题来考验他。Jerry给了Tom一个长度为2 * n的只包含小写字母的字符串，让Tom将这个字符串任意挑选字符，将其分成两个等长的字符串a和b(对于一个si不能同时被选到a和b中)，然后a要和reverse(b)相同(a和反转后的b相同)，问这样的方案数有多少？Tom有些为难，所以请你来帮帮他吧。
输入一个正整数n，和一个长度为2*n的字符串
输出方案数。
示例1
输入：
2
“abba”
输出：
4

解题思路

---

在原字符串中抽出n个字符的字符串a，字符顺序保持不变，剩下的字符都给字符串b，比较a和reverse(b)是否相同(a和反转后的b是否相同)

例如 “abcdef”，挑选第2、3、5个字符，则分成 “bce” 和 “adf” 两个串。

接下来是整理的思路解析：整体框架是dfs，枚举每个字符属于a还是属于b，搜索过程中需要利用a和b的对称性做加速处理，否则会超时。

比方说
xcccddcccxdd

从左往右枚举a字符串的构成，如果令第一个x属于a，根据对称性，倒数第三个字符x一定是属于b；如此推导出末尾的dd一定属于a，中间位置的dd一定属于b，而且是b的头两个字符；然后左边ccc一定a，右边ccc一定是b，由此得出1种方案。令第一个x属于b也可以用同样的方式得到1种方案。

用这个思路直接写代码不太好写，可以通过枚举二进制，固定左半边的选择情况，然后对于每一个case，通过dfs搜索右半边有多少种合法组合，搜索过程中利用对称性进行剪枝。

对于字符全部相同case如"aaaaaaaa"，因为过程中无法剪枝，会退化成2^(2*n)。对于这种case，答案就是 C(2n,n) ，预判一下直接返回即可。

参考代码

---

//摘自csdn@小菜鸟_____

package solution106;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;

class Solution {

	//{0,1,2,3...,n-1}减arr2
     public static int[] getCha(int[] arr2,int n){
        int[] arr1 = new int[2*n];
        
        for(int key = 0;key<2*n;key++){
            arr1[key] = key;
        }

        int[] arr3 = new int[n];
        int k = 0;
        boolean flag = false;
        for(int i= 0;i<arr1.length;i++){
            for(int j = 0;j<arr2.length;j++){
                if(arr1[i] == arr2[j]){
                    flag = true;
                    break;
                }
            }
            if(!flag){
                arr3[k] = arr1[i];
                k++;
            }
            flag = false;
        }
        return arr3;
    }

	//字符串反转
    public static String reverse(String str){
        String new_s1 = "";
        for (int i = str.length() - 1; i >= 0; i--) {
            new_s1 += str.charAt(i);
        }
        return new_s1;
    }

    //从a乘到b,当字符串中所有字符都相同时运行这个直接计算count
    public static int getSameCount(int a, int b) {
        if(a==1){
            return 1;
        }else{
            if(a > b) {
                return a * getSameCount(a - 1, b);
            }else{
                return a;
            }
        }
    }
	
	//深度优先算法实现组合，参考 https://zhuanlan.zhihu.com/p/62833882
	//如combine(4,2)
	//结果就是 [[1, 2], [1, 3], [1, 4], [2, 3], [2, 4], [3, 4]]
    public static List<List<Integer>> combine(int n, int k) {
        List<List<Integer>> res = new LinkedList<>();
        boolean[] f = new boolean[n];
        List<Integer> tmp = new ArrayList<>(k);

        dfs(k, 0, n, f, tmp, res);

        return res;
    }
    private static void dfs(int left, int s, int n, boolean[] f, List<Integer> tmp, List<List<Integer>> res) {
        if(0 == left) {
            res.add(tmp);
            return;
        }

        for(int i = s; i < n; i++) {
            if(f[i])
                continue;

            f[i] = true;
            List<Integer> t = new ArrayList<>(tmp);
            t.add(i+1);
            dfs(left-1, i+1, n, f, t, res);

            f[i] = false;
        }
    }
    
    public long solution(int n, String s) {
      
        int count = 0;
        if(s.length() == 2 && s.substring(0,n).equals(s.substring(n,2*n))){
        	//如果字符串中的所有字符都一样并且字符串长度是2就运行这个
            count = 2;
        }else if(s.substring(0,n).equals(s.substring(n,2*n))){
        	//如果字符串中的所有字符都一样就运行这个
            count = getSameCount(2*n,n+1)/2;
        }else {
        	List result = combine(2*n,n);

        	char[] array = s.toCharArray();

        	String a = "";
        	String b = "";
        	//两个for循环用于遍历result
            for (int i = 0; i < result.size(); i++) {

                List temp = (List) result.get(i);
                int[] arr2 = new int[n];

                for (int j = 0; j < n; j++) {
                	//获取字符串a
                    a = a + array[(int) temp.get(j) - 1];
                    //获取字符串a每个字符对应array的下标
                    arr2[j] = (int) temp.get(j) - 1;
                }
				
				//获取字符串b对应的array下标
                int[] arr3 = getCha(arr2, n);

				//拼接字符串b
                for (int k = 0; k < n; k++) {
                    b = b + array[arr3[k]];
                }
				
				//判断a是否与reverse(b)相等
                if (a.equals(reverse(b))) {
                    count++;
                }
                a = "";
                b = "";
            }
        }
        return count;
    }
}

原文链接

测试结果

---

注：没有服务器的同学，可以参加阿里云高校计划，每天8点免费领取ECS服务器。
活动链接：https://developer.aliyun.com/adc/college/