1. 程式人生 > >【L2-023】圖著色問題 【圖的遍歷】

【L2-023】圖著色問題 【圖的遍歷】

7-11 圖著色問題(25 分)
圖著色問題是一個著名的NP完全問題。給定無向圖G=(V,E),問可否用K種顏色為V中的每一個頂點分配一種顏色,使得不會有兩個相鄰頂點具有同一種顏色?

但本題並不是要你解決這個著色問題,而是對給定的一種顏色分配,請你判斷這是否是圖著色問題的一個解。

輸入格式:

輸入在第一行給出3個整數V(0 < V ≤500)、E(≥0)和K(0 < K ≤ V),分別是無向圖的頂點數、邊數、以及顏色數。頂點和顏色都從1到V編號。隨後E行,每行給出一條邊的兩個端點的編號。在圖的資訊給出之後,給出了一個正整數N(≤20),是待檢查的顏色分配方案的個數。隨後N行,每行順次給出V個頂點的顏色(第i個數字表示第i個頂點的顏色),數字間以空格分隔。題目保證給定的無向圖是合法的(即不存在自迴路和重邊)。

輸出格式:

對每種顏色分配方案,如果是圖著色問題的一個解則輸出Yes,否則輸出No,每句佔一行。

輸入樣例:

6 8 3
2 1
1 3
4 6
2 5
2 4
5 4
5 6
3 6
4
1 2 3 3 1 2
4 5 6 6 4 5
1 2 3 4 5 6
2 3 4 2 3 4
輸出樣例:

Yes
Yes
No
No

分析 :列舉一遍所有點,然後列舉所有與點相連的點。(即列舉所有相連的點,看其顏色是不是相同)
程式碼

#include<bits/stdc++.h>
using namespace std;
#define LL long long
const int N = 1e5+11; const int M = 1e6+11; const int mod = 1e9+7; const int inff = 0x3f3f3f3f; vector<int>ve[N]; int n; int color[N]; int flag; void solve(){ for(int i=1;i<=n;i++){ for(int j=0;j<ve[i].size();j++){ int v=ve[i][j]; if(color[i]==color[v]) flag=0
; } } } int main(){ int m,k;scanf("%d%d%d",&n,&m,&k); while(m--){ int a,b;scanf("%d%d",&a,&b); ve[a].push_back(b); ve[b].push_back(a); } int q;scanf("%d",&q); while(q--){ map<int,int>mp; int f=1; for(int i=1;i<=n;i++) { scanf("%d",&color[i]); mp[color[i]]=1; } if(mp.size()!=k ) puts("No"); // 注意 題意是必須要用到k種顏色 else{ flag=1; solve(); if(flag) puts("Yes"); else puts("No"); } } return 0; }