uva193 - Graph Coloring

不算难的一个题。我却wa了好几次。

最后没有回车。。。

看了好多遍都没看出来。

另外。我们剪枝要剪的及时，能在本层递归结束的话，就立即结束。最好不要拖到下一层。

耗时代码：

#include <cstdio>
#include <cstring>
int max, n, e, ve[120][120], color[120], ans[120];
void dfs(int cur, int c)
{
    for(int i = 1; i <= n; i++) if(ve[cur][i]&&color[i]&&color[cur]) return;
    if(cur==n)
    {
        if(max<c) { for(int i = 1; i <= n; i++) ans[i] = color[i]; max = c; }
        return ;
    }
    color[cur+1] = 1; dfs(cur+1,c+1);
    color[cur+1] = 0; dfs(cur+1,c);
}
int main ()
{
    int t, a, b;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&e);
        memset(ve,0,sizeof(ve));
        memset(color,0,sizeof(color));
        for(int i = 0; i < e; i++)
        {
            scanf("%d%d",&a,&b);
            ve[a][b] = ve[b][a] = 1;
        }
        max = 0;
        color[1] = 1; dfs(1,1);
        color[1] = 0; dfs(1,0);
        printf("%d\n",max);
        for(int i = 1, f = 0; i <= n; i++) if(ans[i]==1)
        { if(f) printf(" "); printf("%d", i); f++; }
        puts("");
    }
    return 0;
}

优化后代码：

#include <cstdio>
#include <cstring>
int max, n, e, ve[120][120], color[120], ans[120];
void dfs(int cur, int c)
{
    if(cur==n)
    {
        if(max<c) { for(int i = 1; i <= n; i++) ans[i] = color[i]; max = c; }
        return ;
    }
    color[cur+1] = 0; dfs(cur+1,c);
    color[cur+1] = 1;
    for(int i = 1; i <= n; i++) if(ve[cur+1][i]&&color[i]) { color[cur+1] = 0; return;}
    dfs(cur+1,c+1);
    color[cur+1] = 0;
}
int main ()
{
    int t, a, b;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&e);
        memset(ve,0,sizeof(ve));
        memset(color,0,sizeof(color));
        for(int i = 0; i < e; i++)
        {
            scanf("%d%d",&a,&b);
            ve[a][b] = ve[b][a] = 1;
        }
        max = 0;
        color[1] = 0; dfs(1,0);
        color[1] = 1; dfs(1,1);
        color[1] = 0;
        printf("%d\n",max);
        for(int i = 1, f = 0; i <= n; i++) if(ans[i]==1)
        { if(f) printf(" "); printf("%d", i); f++; }
        puts("");
    }
    return 0;
}