PAT甲级 1154 Vertex Coloring (25分) map/C++

1154 Vertex Coloring (25分)

题目大意：给定一个图，并给每个顶点上色，要求相连的两个相邻顶点的颜色不能相同。

给定一串颜色序列，判断是否符合要求，如果符合要求，则输出一共用了多少种颜色；不符合要求则输出No。

1. 使用map来存放同种颜色的顶点，map< int , vector < int > > 即 <color，vertices >，再判断vertices中是否有相连的顶点，如果相连，则不符合要求；
2. 输入要使用scanf（），否则会超时（测试点2）。
3. 柳神的代码效率很高，又学到了QWQ（传送门）

#include<iostream>                  //输入输出流头文件
#include<map>                       //map映射容器
#include<vector>                    //变长数组容器
using namespace std;                //标准命名空间
int n,m,k;
int a[10000][10000]={0};//邻接矩阵
void checkColor(map<int,vector<int>> color);
int main(){                         //主函数
#ifdef ONLINE_JUDGE                 //如果有oj系统（在线判定），则忽略文件读入，否则使用文件作为标准输入
#else    
   freopen("1.txt", "r", stdin);   //从1.txt输入数据
#endif 
   cin>>n>>m;
   for(int i=0;i<m;i++){
	   int x,y;
	    scanf("%d %d",&x,&y);
	   a[x][y]=a[y][x]=1;
   }
   map <int ,vector<int>> color;
   cin>>k;
   for(int i=0;i<k;i++){
	   for(int j=0;j<n;j++){
		  int c;
		  scanf("%d",&c);
		  color[c].push_back(j);
	   }
	   checkColor(color);
	   color.clear();
   }
    return 0;                       //返回0，如果不返回0，PAT会报错
}
void checkColor(map<int,vector<int>> color){
	map<int,vector<int>>::iterator iter=color.begin();
	for(;iter!=color.end();iter++){
		for(int i=0;i<iter->second.size();i++){
			for(int j=i+1;j<iter->second.size();j++){
				if(a[iter->second[i]][iter->second[j]]==1) {cout<<"No"<<endl;return ;}
			}
		}
	}
	cout<<color.size()<<"-coloring"<<endl;
}