L2-023 图着色问题 (25 分)
坑:
- 边的个数可能等于0
- 边数上限 v 2 v^2 v2
- 是恰好k种颜色,不能多也不能少
- 去重,先sort再unique,取unique的返回值。注意副作用(重复的元素放在后面)。
代码
#include <iostream>
#include <string>
#include <cstring>
#include <algorithm>
#include <stack>
#include <queue>
#include <vector>
#include <cmath>
#include <cstdio>
#include <bitset>
using namespace std;
const int N = 500 + 3;
const int EN = 260000;
static int v, e, k;
static int p1[EN], p2[EN];
static int color[N];
int main(){
cin >> v >> e >> k;
for(int i=0; i<e; i++){
cin >> p1[i] >> p2[i];
p1[i]--; p2[i]--;
}
int n; cin >> n;
for(int ii=0; ii<n; ii++){
bool flag = true;
for(int i=0; i<v; i++)
cin >> color[i];
for(int i=0; i<e; i++){
if(color[p1[i]] == color[p2[i]]){
flag = false;
break;
}
}
sort(color, color+v);
int num_color = unique(color, color+v)-(color);
if(num_color != k) flag = false;
if(flag) cout << "Yes" << endl;
else cout << "No" << endl;
}
return 0;
}