1301: [蓝桥杯2016决赛]随意组合

题意

传送门

分析

当时想到的就是排列组合，就是从第一组挑一个数，第二组挑一个数。
用的是搜索，而且用了两个dfs，同时为了避免重复，四个数一定是从大到小排列的。

#include <iostream>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <vector>
#include <map>
#include <queue>
#include <cstdio>
#include <string>
#include <stack>
#include <set>
#include<cmath>
#include <iomanip>
#define IOS ios::sync_with_stdio(false), cin.tie(0)
using namespace std;
int a[4]={2,3,5,8};//第一组数 
int b[4]={1,4,6,7};//第二组数 
bool book_1[4];//标记第一组中的某个数是否已选用 
bool book_2[4]; //标记第二组数是否已选用 
int c[5];//最后选择完第一组数的排列 
int d[5];//最后选择完第二组数的排列  
int ans=0;
//两个dfs会交叉搜 
void dfs_1(int step);
void dfs_2(int step){
	
	for(int i=0;i<4;i++){
		if(book_2[i]==false){
			d[step]=b[i];
			book_2[i]=true;
			dfs_1(step+1);
			book_2[i]=false;
		}
	}
}
void dfs_1(int step){
	if(step==5){
		if((c[1]*10+d[1])*(c[1]*10+d[1])+(c[2]*10+d[2])*(c[2]*10+d[2])+(c[3]*10+d[3])*(c[3]*10+d[3])+(c[4]*10+d[4])*
		(c[4]*10+d[4])==(d[1]*10+c[1])*(d[1]*10+c[1])+(d[2]*10+c[2])*(d[2]*10+c[2])+(d[3]*10+c[3])*(d[3]*10+c[3])
		+(d[4]*10+c[4])*(d[4]*10+c[4])&&(c[1]*10+d[1])<(c[2]*10+d[2])&&(c[2]*10+d[2])<(c[3]*10+d[3])&&(c[3]*10+d[3])<(c[4]*10+d[4])){
			ans++;
			for(int i=1;i<=4;i++){
				cout<<c[i]<<d[i]<<" ";
			}
			cout<<endl;
			return ;
		}
	}
	for(int i=0;i<4;i++){
	   if(book_1[i]==false){
	   	c[step]=a[i];
	   	book_1[i]=true;
	   	dfs_2(step);
	   	book_1[i]=false;
	   }
	}
}
int main(){
	dfs_1(1);
	cout<<ans<<endl;
  //cout<<24<<endl;
	return 0;
}