bzoj4878 [Lydsy1705月赛]挑战NP-Hard dfs

Description

---

天才大学生quailty热衷于解决NP-Hard问题，你如果AC 了这道题，就可以成为他真正的粉丝。图染色问题：给定
无向图G和一个正整数k。对于图中的每个点，选择一个在[1,k]之间的整数作为其颜色。你需要保证对于每条边，
其两端点的颜色均不相同。简单k路径问题：给定无向图G和一个正整数k。请找到一条经过了恰好k条边的简单路径
。即，你需要找到一个长度为k+1的序列v_1,v_2,…,v_{k+1}，满足1<=v_i<=n，且任意两个v均不相同，同时v_i
与v_{i+1}之间存在一条边。现在给定无向图G和一个正整数k，quailty知道你没有他的水平，所以你只需解决上面
的任意一个问题就可以成为他的粉丝。

Solution

---

路径问题可以变成最长路，如果最长路>=k就是可行的
染色问题可以贪心地每次选择相邻节点未选择的最小颜色染（类似mex函数，如果存在颜色编号>k就是不可行的

然而有更加优秀的做法，先dfs一次求出染色方案，若不能染色就找到颜色为k+1的点沿着颜色递减走，这样一定能走一条长度为k的路径。正确性证明可以考虑归纳法（懒
需要注意的是图并不一定联通（我大概在这个问题上死过好多次了

Code

---

#include <stdio.h>
#include <string.h>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define drp(i,st,ed) for (int i=st;i>=ed;--i)
#define fill(x,t) memset(x,t,sizeof(x))

const int N=4005;
const int E=40005;

struct edge {int y,next;} e[E];

int ls[N],edCnt;
int col[N],rec[N],id;

int read() {
    int x=0,v=1; char ch=getchar();
    for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
    for (;ch>='0'&&ch<='9';x=x*10+ch-'0',ch=getchar());
    return x*v;
}

void add_edge(int x,int y) {
    e[++edCnt]=(edge) {y,ls[x]}; ls[x]=edCnt;
    e[++edCnt]=(edge) {x,ls[y]}; ls[y]=edCnt;
}

void dfs(int now) {
    id++;
    for (int i=ls[now];i;i=e[i].next) rec[col[e[i].y]]=id;
    for (int i=1;;i++) if (rec[i]!=id) {
        col[now]=i;
        break;
    }
    for (int i=ls[now];i;i=e[i].next) if (!col[e[i].y]) dfs(e[i].y);
}

void solve() {
    fill(ls,0); fill(col,0); edCnt=1;
    int n=read(),m=read(),k=read();
    rep(i,1,m) add_edge(read(),read());
    rep(i,1,n) if (!col[i]) dfs(i);
    bool flag=false;
    rep(i,1,n) if (col[i]>k) flag=true;
    if (!flag) {
        printf("color");
        rep(i,1,n) printf(" %d", col[i]);
    } else { k++; int st;
        drp(i,n,1) if (col[i]==k) st=i;
        printf("path");
        for (int x=st;k--;) {
            printf(" %d", x);
            for (int i=ls[x];i;i=e[i].next) if (col[e[i].y]==col[x]-1) {
                x=e[i].y;
                break;
            }
        }
    } puts("");
}

int main(void) {
    for (int T=read();T--;solve());
    return 0;
}