题目
有一天Jerry给Tom出了一道题来考验他。Jerry给了Tom一个长度为2n的只包含小写字母的字符串,让Tom将这个字符串任意挑选字符,将其分成两个等长的字符串a和b(对于一个si不能同时被选到a和b中),然后a要和reverse(b)相同(a和反转后的b相同),问这样的方案数有多少?Tom有些为难,所以请你来帮帮他吧。
输入一个正整数n,和一个长度为2n的字符串
输出方案数。
示例1
输入:
2
“abba”
输出:
4
解题思路:
所谓挑选n个字符变成a和b两个字符串,是指在原字符串中抽出n个字符,这些字符的的顺序保持不变,剩下字符的顺序也保持不变,由此组成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;
}
}