[蓝桥杯2016决赛]随意组合
题目描述
小明被绑架到X星球的巫师W那里。其时,W正在玩弄两组数据 (2 3 5 8) 和 (1 4 6 7)
他命令小明从一组数据中分别取数与另一组中的数配对,共配成4对(组中的每个数必被用到)。
小明的配法是:{(8,7),(5,6),(3,4),(2,1)}。巫师凝视片刻,突然说这个配法太棒了!
因为:每个配对中的数字组成两位数,求平方和,无论正倒,居然相等:
87^2 + 56^2 + 34^2 + 21^2 = 12302
78^2 + 65^2 + 43^2 + 12^2 = 12302
小明想了想说:“这有什么奇怪呢,我们地球人都知道,随便配配也可以啊!”
{(8,6),(5,4),(3,1),(2,7)}
86^2 + 54^2 + 31^2 + 27^2 = 12002
68^2 + 45^2 + 13^2 + 72^2 = 12002
巫师顿时凌乱了。。。。。
请你计算一下,包括上边给出的两种配法,巫师的两组数据一共有多少种配对方案具有该特征。
配对方案计数时,不考虑配对的出现次序。
就是说:{(8,7),(5,6),(3,4),(2,1)}与{(5,6),(8,7),(3,4),(2,1)}是同一种方案。
输出
输出一个整数表示答案
package pro;
import java.util.ArrayList;
public class Main {
public static void main(String[] args) {
int arr1[] = {2,3,5,8};
int arr2[] = {1,4,6,7};
int vis[] = new int[4];
int res[] = new int[4];
ArrayList<int[]> list = new ArrayList<>();
ArrayList<int[]> list2 = new ArrayList<>();
dps(0,list,vis,arr1,res);
vis = new int[4];
dps(0,list2,vis,arr2,res);
test(list,list2);
}
static void dps(int i,ArrayList<int[]> list,int vis[],int arr[],int res[]) {
if(i==4) {
list.add(res);
}
for(int j = 0; j < 4; j++) {
if(vis[j]==0) {
vis[j] = 1;
res[i] = arr[j];
dps(i+1,list,vis,arr,res);
vis[j] = 0;
}
}
}
static void test(ArrayList<int[]> list,ArrayList<int[]> list2) {
int[] a;
int cnt = 0;
for(int i = 0; i < list.size(); i++) {
a = list.get(i);
for(int j = 0; j < list2.size(); j++) {
int[] b = list2.get(j);
int n = (int)(Math.pow(a[0]*10+b[0], 2)+Math.pow(a[1]*10+b[1], 2)+Math.pow(a[2]*10+b[2], 2)+Math.pow(a[3]*10+b[3], 2));
int m = (int)(Math.pow(b[0]*10+a[0], 2)+Math.pow(b[1]*10+a[1], 2)+Math.pow(b[2]*10+a[2], 2)+Math.pow(b[3]*10+a[3], 2));
if(n==m)
cnt++;
}
}
System.out.println(cnt/24);
}
}
解析:使用两个全排列,将得到的两个全排列进行重组,就是组合的所有情况,然后判断去重.
2021-02-01