如果题目要求是输出所有可行的染色方案:
#include<bits/stdc++.h>
using namespace std;
int n,m;
int mapp[105][105];
int color[105];
bool judge(int x,int c)
{
for(int i=1;i<=n;i++)
{
if(mapp[i][x]==1&&color[i]==c)//若出现了与其相连的点与此颜色一样,则此颜色不可取
return false;
}
return true;
}
void dfs(int x)
{
if(x==n+1)
{
for(int i =1;i<=n;i++)
{
cout<<color[i];
if(i==n)
cout<<endl;
else
cout<<" ";
}
return ;
}
for(int i=1;i<=n;i++)//遍历颜色
{
if(judge(x,i)==1)//判断这个点能否取这个颜色
{
color[x]=i;
dfs(x+1);
color[x]=0;
}
}
}
int main()
{
cin>>n>>m;
int a,b;
for(int i=1;i<=m;i++)
{
cin>>a>>b;
mapp[a][b]=1;//表示这两个点相邻,就不能为同一种颜色
mapp[b][a]=1;
}
dfs(1);
}
如果题目要求是最少的颜色数:
例题:
问题描述
n个人参加某项特殊考试。
为了公平,要求任何两个认识的人不能分在同一个考场。
求是少需要分几个考场才能满足条件。
输入格式
第一行,一个整数n(1<n<100),表示参加考试的人数。
第二行,一个整数m,表示接下来有m行数据
以下m行每行的格式为:两个整数a,b,用空格分开 (1<=a,b<=n) 表示第a个人与第b个人认识。
输出格式
一行一个整数,表示最少分几个考场。
样例输入
5
8
1 2
1 3
1 4
2 3
2 4
2 5
3 4
4 5
样例输出
4
样例输入
5
10
1 2
1 3
1 4
1 5
2 3
2 4
2 5
3 4
3 5
4 5
样例输出
5
因为这个问题的数据量很小,所以我们想到了回溯,在已知的状态下,然后去判断下一个学生是否可以在之前存在的教室,如果不可以的话,那么就新开一个教室。
代码如下:
#include<bits/stdc++.h>
using namespace std;
int mapp[105][105]; //是否存在关系,存在关系就是1,不存在关系就是0
int cun[105][105]; //第一维度表示的是考场,二位度里面放的是这个考场里面的学生
//cun[1][1] = 2,cun[1][2] = 0;表示考场1里面第一个存在的人是2,然后后面一位是0,也就是不存在人了,那么这时的cun[1] = 1;表示的是考场里面人的数量
int cnt[105]; //每一个考场学生的数量
int res=99999999;
int n,m;
void solve(int id,int num)//id表示学生,num表示当前已经分的考场数
{
if(num>=res) //当现在安排的数量已经大于了最小的教室数量的话,返回
return;
if(id>n) //当已经安排完所有的学生了时
{
res=min(res,num);
return;
}
for(int i=1;i<=num;i++) //首先看看之前的房间里面能不能放进去,遍历之前每一个房间
{
int jishu=0; //得到的是和这个人不认识的人数
for(int j=1;j<=cnt[i];j++)
{
if(mapp[id][cun[i][j]]==0)
jishu++;
}
if(jishu==cnt[i]) //如果这里面的学生都和现在将要插入的学生的不认识
{
++cnt[i];
cun[i][cnt[i]]=id; //将这个学生存到这个考场中
solve(id+1,num);
cun[i][cnt[i]]=0;
cnt[i]--;//可以放进去,但是也可以选择不放对吧,所以回溯
}
}
//或者直接重新开一个房间
cnt[num+1]++;
cun[num+1][cnt[num+1]]=id; //没有的话就把它放到下一个教室里
solve(id+1,num+1);
cun[num+1][cnt[num+1]]=0;
cnt[num+1]-- //回溯;
}
int main()
{
cin>>n>>m;
while(m--)
{
int a,b;
cin>>a>>b;
mapp[a][b]=mapp[b][a]=1;
}
solve(1,0);
cout<<res<<endl;
return 0;
}