拼多多2021笔试真题集 -- 4. 多多的骰子组合

多多的骰子组合
多多君拼团购买了N个骰子，为了方便后面进行活动，多多君需要将这些骰子进行分类。

两个骰子为同类的定义是：
1
将其中一个骰子通过若干次上下、左右或前后翻转后，其与另一个骰子对应的6面数字均相等。

现在多多君想知道不同种类的骰子的数量分别有多少。
时间限制：C/C++ 2秒，其他语言4秒
空间限制：C/C++ 256M，其他语言512M
输入描述：
第一行1个整数N，表示骰子的数量。
（1 <= N <= 1,000）
接下来N行，每行6个数字（1～6，且各不相同）
其中第i行表示第i个骰子当前上、下、左、右、前、后这6面的数字。
输出描述：
共2行:
第一行1个整数M，表示不同种类的骰子的个数
第二行M个整数，由大到小排序，表示每个种类的骰子的数量
示例1
输入例子：
2
1 2 3 4 5 6
1 2 6 5 3 4
输出例子：
1
2
例子说明：
第二个骰子相当于是第一个骰子从左向右旋转了一面得到，属于同类。
示例2
输入例子：
3
1 2 3 4 5 6
1 2 6 5 3 4
1 2 3 4 6 5
输出例子：
2
2 1
例子说明：
第三个骰子无法通过任何旋转变换成第一个或第二个骰子。
示例3
输入例子：
10
2 5 1 3 4 6
5 4 3 2 1 6
1 4 6 2 3 5
1 5 6 3 4 2
6 4 2 1 5 3
3 6 4 5 2 1
1 6 3 4 2 5
5 1 4 2 6 3
6 2 3 1 5 4
5 3 6 1 4 2
输出例子：
9
2 1 1 1 1 1 1 1 1
例子说明：
只有第4个骰子(1 5 6 3 4 2)与第8个骰子(5 1 4 2 6 3)属于同一类。

一种可能的变换方式:

1. 首先从右向左翻转1次
   (1 5 6 3 4 2) -> (1 5 4 2 3 6)
2. 然后从上向下翻转2次
   (1 5 4 2 3 6) -> (6 3 4 2 1 5) -> (5 1 4 2 6 3)

题解

模拟题，考虑一个骰子各种角度的结果就好了。

AC代码

#include<bits/stdc++.h>
using namespace std;
int U[6],D[6],L[6],R[6],B[6];
void init()
{
	U[0] = 5, D[0] = 4, L[0] = 2, R[0] = 3, B[0] = 1;
//	U[0][1] = 2, D[0][1] = 3, L[0][1] = 4, R[0][1] = 5, B[0][1] = 1;
// 4 5 3 2
//3 2 5 4
	U[1] = 4, D[1] = 5, L[1] = 2, R[1] = 3, B[1] = 0;
	U[2] = 0, D[2] = 1, L[2] = 5, R[2] = 4, B[2] = 3;
	U[3] = 0, D[3] = 1, L[3] = 4, R[3] = 5, B[3] = 2;
	U[4] = 0, D[4] = 1, L[4] = 2, R[4] = 3, B[4] = 5;
	U[5] = 0, D[5] = 1, L[5] = 3, R[5] = 2, B[5] = 4;
}
vector<string>get_nums(string arr)
{
	vector<string>res;
	res.push_back(arr);
	for(int i=0;i<6;i++)
	{
		string s = "";
		s += arr[U[i]];
		s += arr[D[i]];
		s += arr[L[i]];
		s += arr[R[i]];
		s += arr[i];
		s += arr[B[i]];
		res.push_back(s);

		s = "";
		s += arr[L[i]];
		s += arr[R[i]];
		s += arr[D[i]];
		s += arr[U[i]];
		s += arr[i];
		s += arr[B[i]];
		res.push_back(s);

		s = "";
		s += arr[D[i]];
		s += arr[U[i]];
		s += arr[R[i]];
		s += arr[L[i]];
		s += arr[i];
		s += arr[B[i]];
		res.push_back(s);

		s = "";
		s += arr[R[i]];
		s += arr[L[i]];
		s += arr[U[i]];
		s += arr[D[i]];
		s += arr[i];
		s += arr[B[i]];
		res.push_back(s);

	}
	return res;
}
set<string>vis;
bool check(string s1, string s2)
{
	vis.clear();
	vector<string> nums1 = get_nums(s1);
	vector<string> nums2 = get_nums(s2);
	for(int i=0;i<nums1.size();i++)
	{
		vis.insert(nums1[i]);
	}
	for(int j=0;j<nums2.size();j++)
	{
		if(vis.find(nums2[j])!=vis.end())
		{
			return true;
		}
	}
	return false;
}
int n;
string arr[1005];
vector<vector<string> >all;
vector<int>result;
int main()
{
	init();
	cin>>n;
	for(int i=0;i<n;i++)
	{
		arr[i] = "";
		for(int j=0;j<6;j++)
		{
			int x;
			cin>>x;
			arr[i] += ('0' + x);
		}
	}
	vector<string>single;
	single.push_back(arr[0]);
	all.push_back(single);
	for(int i=1;i<n;i++)
	{
		int m = all.size();
		bool flag = true;
		for(int j=0;j<m;j++)
		{
			if(check(all[j][0], arr[i]))
			{
				all[j].push_back(arr[i]);
				flag = false;
				break;
			}
		}
		if(flag)
		{
			single.clear();
			single.push_back(arr[i]);
			all.push_back(single);			
		}
	}
	for(int i=0;i<all.size();i++)
	result.push_back(all[i].size());
	sort(result.begin(),result.end());
	cout<<result.size()<<endl;
	for(int i=result.size()-1;i>=0;i--)
	cout<<result[i]<<" ";
	return 0;
}