树染色

树的染色

题意
每条边的两个端点进行染色，要求每条边两端颜色不一样，求至少需要几种颜色
分析
如果只有一个点，当然只要一种颜色就行。
如果是棵树，没有环，答案是2。
如果有环，且有节点数为奇数的环，答案是3，否则是2。
题解
说明：开始我是想先剪枝，将不成环的分支剪掉，然后遍历寻找环的个数，判断奇偶，但是太繁琐，索性我直接进行遍历染色，看能否被两种颜色染色，能则为2，不能则为3.
定义：vector<int>a[MAX]存放父节点及其子节点。
bool dp1[MAX]={0}作为标记判断是否被遍历过。
bool col[MAX]={0}记录路节点颜色。（默认颜色只有两种）
bool flag=0 作为标志判断是否能被染色成功。
void dfs(int now)遍历函数。
代码如下

#include<bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f
#define MAXI 100005
#define MAX 200005
vector<int> node[MAXI];
int n,m;
bool dp1[MAXI]={0};
bool col[MAXI]={0};
bool flag=0;
void dfs(int now){
 for(int i=0;i<node[now].size();i++){
  int son=node[now][i];
  if(!dp1[son]){
   dp1[son]=true;
   col[son]=!col[now];
   dfs(son);
  }
  else{
   if(col[son]==col[now]){
    flag=1;
    return;
   }
  }
 }
}

int main(){
    cin>>n>>m;dp1[0]=0;int t=1;
    int u,v;int maxn=0;
    for(int i=0;i<m;i++){
        cin>>u>>v;
        node[u].push_back(v);
        node[v].push_back(u);
    }
    if(n<2) {
     puts("1");
     return 0;
 } 
 col[1]=true;
 dfs(1);
    if(flag) puts("3");
    else puts("2");
    return 0;
}