[ BAPC 2016 ] Rock Band 暴力

题目链接：Rock Band

题意

给你一个n行m列矩阵，每行为1～m的一个排列，让你找一个以第一列为起点的n*x的子矩阵使之其中的元素在子矩阵外未出现。x应该尽可能小，最后输出子矩阵里不同元素的个数以及将子矩阵的元素从小到大输出。

题解

由题意可知，每行为1～m的一个排列，如果一个元素只在子矩阵出现，那么代表这个元素必须在子矩阵里出现n次。
我们可以从第一列开始向右遍历，统计未标记的元素个数，如果出现过的元素个数达到n个则从统计个数减去，每遍历一列就判断统计个数是否为0，如果为0代表符合条件，否则继续遍历。

代码

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;

const int maxn=1e6+10;

vector<int> g[maxn];
vector<int> tt; // 子矩阵内的所有元素
int book[maxn]; //标记数组
int main()
{
    
    
	int m,s;
	scanf("%d%d",&m,&s);
	for(int i=0;i<m;i++)
		for(int j=0;j<s;j++)
		{
    
    
			int tmp; 
			scanf("%d",&tmp);
			g[i].push_back(tmp);
		}
	int ans=0,cnt=0;
	for(int j=0;j<s;j++)
	{
    
    
		for(int i=0;i<m;i++)
		{
    
    
			if(!book[g[i][j]]) {
    
     tt.push_back(g[i][j]); cnt++;}
			book[g[i][j]]++;
			if(book[g[i][j]]==m) cnt--;
		}
		if(cnt==0)
		{
    
    
			ans=j+1;
			break;
		}
	}
	printf("%d\n",ans);
	sort(tt.begin(),tt.end());
	for(int i=0;i<tt.size();i++) printf("%d ",tt[i]);
}