洛谷 P2819 图的m着色问题

题目：

题目背景

给定无向连通图G和m种不同的颜色。用这些颜色为图G的各顶点着色，每个顶点着一种颜色。如果有一种着色法使G中每条边的2个顶点着不同颜色，则称这个图是m可着色的。图的m着色问题是对于给定图G和m种颜色，找出所有不同的着色法。

题目描述

对于给定的无向连通图G和m种不同的颜色，编程计算图的所有不同的着色法。

输入输出格式

输入格式：

 

第1行有3个正整数n，k 和m，表示给定的图G有n个顶点和k条边，m种颜色。顶点编号为1，2，…，n。接下来的k行中，每行有2个正整数u,v，表示图G 的一条边(u,v)。

 

输出格式：

 

程序运行结束时，将计算出的不同的着色方案数输出。

 

输入输出样例

输入样例#1： 复制

5 8 4
1 2
1 3
1 4
2 3
2 4
2 5
3 4
4 5

输出样例#1： 复制

48

说明

n<=100;k<=2500;

在n很大时保证k足够大。

保证答案不超过20000。

思路：

一开始看到这个题目，想着是从1进行搜索，然后根据相连的边判定选的颜色是否合理，写完之后发现不对，调试也调不好，感觉可能是回溯的时候没有完全回溯。

然后看题解发现是按照顶点升序进行搜索的。发现这样确实是比较好写一些，回溯也比较清晰，于是就写了一个。

建边是用的邻接矩阵。因为给出的边比较少。而且判定的时候也比较好判定。

代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
const int maxn=105;
int n,k,m;
int head[maxn];
int ans=0;
int col[maxn];
int edge[maxn][maxn];
void dfs (int num)
{
    if(num==n+1)
    {
        ans++;
        return;
    }
    for (int i=1;i<=m;i++)
    {
        int ok=1;
        for (int j=1;j<num;j++)
        {
            if(edge[num][j]==1&&col[j]==i)
            {
                ok=0; break;
            }
        }
        if(ok)
        {
            col[num]=i;
            dfs(num+1);
            col[num]=-1;
        }
    }
}
int main()
{
    memset (edge,0,sizeof(edge));
    memset (col,-1,sizeof(col));
    scanf("%d%d%d",&n,&k,&m);
    for (int i=0,j=0;i<k;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        edge[x][y]=edge[y][x]=1;
    }
    dfs(1);
    printf("%d\n",ans);
    return 0;
}