2023 睿抗机器人开发者大赛CAIP-编程技能赛-本科组（国赛）RC-u4 拆积木

题目描述：

给定一个由带编号的积木搭成的长方体。其中每块积木的厚度都一样，由若干个单位边长的相邻方块组成（相邻是指两个方块有一面重合）。现在要求将这个长方体中的积木一块一块拆掉。每块积木只能从顶端取出，并且取出时不能移动还在长方体中的其它积木。请你给出一个拆积木的顺序。当然这种顺序可能是不唯一的，我们规定当有多种选择时，总是取出编号最小的那个。

输入格式:

输入第一行给出两个正整数 N 和 M（1≤N,M≤1000），分别为长方体的高度和宽度。随后 N 行，每行给出 M 个整数，为对应位置上积木的编号。编号为不超过 106 的正整数，数字间以空格分隔。题目保证不同积木的编号是不同的。

输出格式:

在一行中输出按照题目限定可以拆卸积木的顺序。数字间以 1 个空格分隔，行首尾不得有多余空格。如果拆卸到某一步发现无法继续了，则在对应位置上输出 Impossible，然后结束。

输入样例 1（示意如上图）:

5 5
4 4 4 11 11
9 9 4 2 1
9 5 4 4 4
7 3 3 6 10
8 8 8 10 10

 输出样例 1:

11 1 2 4 6 9 5 3 7 8 10

输入样例 1（示意如上图）:

4 3
8 9 7
4 4 4
5 6 4
1 4 4

输出样例 2:

7 8 9 Impossible

思路：问可以拆卸积木的顺序，可以想到拓扑排序。对于a，b两个编号的积木，如果a的某一块在b的某一块的相邻正上方，那么就建边a->b，建好图后再拓扑排序即可。

要点：

1.要求有多种选择时，总是取出编号最小的那个，所以要用优先队列

2.注意要避免内存超限。

代码：

#include<iostream>
#include<vector>
#include<map>
#include<queue>
#include<cstring>
using namespace std;
int a[1005];
vector<int>res[1000005],dan;
int degree[1000005];
bool v[1000005];
priority_queue<int,vector<int>, greater<int> >ans;
int main(){
	int n,m;
	cin>>n>>m;
	memset(degree,0,sizeof(degree));
	memset(v,0,sizeof(v));
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			int b;
			cin>>b;
			v[b]=1;
			int flag=0;
			for(int k=0;k<res[a[j]].size();k++){
				if(res[a[j]][k]==b){
					flag=1;
					break;
				}
			}
			if(flag==1)
			{
				a[j]=b;
				continue;
			}
			if(i!=1&&a[j]!=b){
				res[a[j]].push_back(b);
				degree[b]++;
			}
			a[j]=b;
		}
	}
	int c=0;
	for(int i=1;i<=1000000;i++){
		if(v[i])
		{
			c++;
			if(degree[i]==0)
			ans.push(i);
		}
	}
	while(!ans.empty()){
		int t=ans.top();
		ans.pop();
		dan.push_back(t);
		for(int i=0;i<res[t].size();i++){
			degree[res[t][i]]--;
			if(degree[res[t][i]]==0){
				//cout<<res[t][i]<<" "<<t<<endl; 
				ans.push(res[t][i]);
			}
		}
	}
	for(int i=0;i<dan.size();i++){
		if(i==0)
		cout<<dan[i];
		else
		cout<<" "<<dan[i];
	}
	if(dan.size()!=c)
	{
		if(dan.size()==0)
		cout<<"Impossible";
		else
		cout<<" Impossible";
	}
}