codeforces 464B Restore Cube

题目链接

给8个点， 判断这8个点能否组成一个正方体， 如果能， 输出这8个点。 同一个点的x, y, z可以交换。

每一个点有6种排列方式， 一个8个点， 暴力枚举出所有排列方式然后判断能否组成正方体。 判断的方法看代码。

#include <iostream>
#include <vector>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <map>
#include <set>
#include <string>
#include <queue>
#include <stack>
#include <bitset>
using namespace std;
#define pb(x) push_back(x)
#define ll long long
#define mk(x, y) make_pair(x, y)
#define lson l, m, rt<<1
#define mem(a) memset(a, 0, sizeof(a))
#define rson m+1, r, rt<<1|1
#define mem1(a) memset(a, -1, sizeof(a))
#define mem2(a) memset(a, 0x3f, sizeof(a))
#define rep(i, n, a) for(int i = a; i<n; i++)
#define fi first
#define se second
typedef pair<int, int> pll;
const double PI = acos(-1.0);
const double eps = 1e-8;
const int mod = 1e9+7;
const int inf = 1061109567;
const int dir[][2] = { {-1, 0}, {1, 0}, {0, -1}, {0, 1} };
int a[8][3];
ll dis[8];
ll get_dis(int x, int y) {
ll ret = 0;
for(int i = 0; i<3; i++) {
ret += 1LL*(a[x][i]-a[y][i])*(a[x][i]-a[y][i]);
}
return ret;
}
void output() {
puts("YES");
for(int i = 0; i<8; i++) {
for(int j = 0; j<3; j++) {
cout<<a[i][j]<<" ";
}
cout<<endl;
}
}
int check() {
for(int center = 0; center<8; center++) {
int cnt = 0;
for(int i = 0; i<8; i++) {
if(i == center)
continue;
dis[cnt++] = get_dis(i, center);
}
sort(dis, dis+7);
if(dis[0] == 0)
return 0;
if(dis[0]!=dis[1]||dis[1]!=dis[2])
return 0;
if(dis[3]!=dis[4]||dis[4]!=dis[5])
return 0;
if(dis[3]!= 2*dis[0])
return 0;
if(dis[6] != 3*dis[0])
return 0;
}
return 1;
}
int dfs(int d) {
if(d == 8) {
if(check()) {
output();
return 1;
}
return 0;
}
sort(a[d], a[d]+3);
do {
if(dfs(d+1))
return 1;
} while(next_permutation(a[d], a[d]+3));
return 0;
}
int main()
{
for(int i = 0; i<8; i++) {
for(int j = 0; j<3; j++) {
cin>>a[i][j];
}
}
if(!dfs(0)) {
puts("NO");
}
return 0;
}