L2-023 图着色问题 (25 分) 坑点详细分析

L2-023 图着色问题 (25 分) 坑点详细分析！！！

题目描述

图着色问题是一个著名的NP完全问题。给定无向图$G=(V,E)$，问可否用$K$种颜色为V中的每一个顶点分配一种颜色，使得不会有两个相邻顶点具有同一种颜色？

但本题并不是要你解决这个着色问题，而是对给定的一种颜色分配，请你判断这是否是图着色问题的一个解。

输入格式

输入在第一行给出3个整数$V$$(0<V\le 500)$、$E$$(\ge 0）$和$K$$(0<K\le V)$，分别是无向图的顶点数、边数、以及颜色数。顶点和颜色都从1到V编号。随后E行，每行给出一条边的两个端点的编号。在图的信息给出之后，给出了一个正整数$N$$(\le 2)$，是待检查的颜色分配方案的个数。随后$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

主体思路

此问题实际是一个二分图的问题，跟利用染色法判定图是否为二分图类似！

1. 利用邻接表存储图，注意这是个无向图！(这个不会就去看y总的算法基础课！y总yyds!)
2. 采用深搜的策略，看一条边的两个顶点所染的颜色是否相同，相同返回false，深搜结束若没有返回false则本路径染色符合要求，循环的去检查其余路径是否满足要求，若都满足则该颜色分配符合要求！

（详细请看代码！！！）

坑点分析

1. 注意这是个无向图，存边的时候需要将a->b和b->a这两条边都存进去！
2. 注意要判断询问时候颜色数量是否等于初始给定的染色数。(利用set)

   	   set<int> hs;
       for(int i = 1; i <= n; i ++ )
       {
           int x;
           scanf("%d", &x);
           color[i] = x;
           hs.insert(x);
       }
       if(hs.size() != k)
       {
           puts("No");
           continue;
       }

3、记得$N$开大一点，不开大过不去（超坑的！！！）
       N = 2e5 + 10(第五个测试点不会过)

       N = 2e6 + 10(AC)

       N = 2e7 + 10(0分，内存超限！)

c++代码

#include <iostream>
#include <cstring>
#include <set>

using namespace std;

const int N = 2e6 + 10; //坑点3开到2e5 + 10会过不去，开到2e7 + 10会内存超限

int n, m, k, q;
int h[N], ne[N], e[N], idx;//数组模拟邻接表4件套
int color[N];//记录每个节点的颜色
bool st[N];//记录该节点是否已经遍历到

//用数组模拟链表
void add(int a, int b)
{
	//假设存储的是(a->b)
    e[idx] = b;
    ne[idx] = h[a];
    h[a] = idx ++ ;
}

//深度优先遍历
bool dfs(int u)
{
    st[u] = true;//该节点已经被遍历到
    for(int i = h[u]; i != -1; i = ne[i])
    {
        int j = e[i];//将u节点的邻接节点取出
        if(!st[j])//如果j节点未被遍历到
        {
            if(!dfs(j)) return false;
        }
        else if(color[u] == color[j]) return false;//如果当前节点与它的邻接节点所染颜色相同的话，则一定不是正确的染色方式，返回false
    }
    return true;
}

int main()
{
    scanf("%d%d%d", &n, &m, &k);
    memset(h, -1, sizeof h);//初始话头节点，规定-1指向空
    //存储图
    for(int i = 0; i < m; i ++ )
    {
        int a, b;
        scanf("%d%d", &a, &b);
        add(a, b), add(b, a);
    }
    
    scanf("%d", &q);
    
    while(q -- )
    {
    	//判断询问时候颜色数量是否等于初始给定的染色数
        set<int> hs;
        for(int i = 1; i <= n; i ++ )
        {
            int x;
            scanf("%d", &x);
            color[i] = x;
            hs.insert(x);
        }
        if(hs.size() != k)
        {
            puts("No");
            continue;
        }
        //清空st数组
        memset(st, false, sizeof st);
        bool flag = true;
        //检查所有节点是否发生冲突，发生冲突输出No，反之输出Yes
        for(int i = 1; i <= n; i ++ )
        {
            if(!dfs(i))
            {
                flag = false;
                break;
            }
        }
        if(flag) puts("Yes");
        else puts("No");
    }
    
    return 0;
}