回溯法（3着色问题）

3着色问题

考虑3着色问题：给出一个无向图G=(V,E)，需要用三种颜色之一为V中的每个顶点着色，三种颜色分别为1，2，3，使得没有两个邻接的顶点有同样的颜色。我们把这样的着色称为合法的；否则，如果两个邻接的顶点有同一种颜色就是非法的。      

【问题解析】

 一种着色可以用n元组(c1,c2,…,cn)来表示，使ci∈{1,2,3}, 1≤i≤n。一个n个顶点的图共有3n种可能的着色（合法的和非法的），所有可能的着色的集合可以用一棵完全的三叉树来表示，称为搜索树(状态空间树)，从根到叶节点的每一条路径代表一种着色的指派。

【算法思路】

回溯法是一种常用的解决组合优化问题的算法，其中3着色问题是一个经典的图论问题，要求给定一个图，确定每个顶点使用不同颜色，使得相邻的顶点不使用相同颜色。以下是使用回溯法解决3着色问题的算法思路：

1. 从图中选择一个顶点作为起始点，给它一个颜色并标记为已访问。
2. 递归地对其相邻的未访问过的顶点进行着色，确保它们与当前顶点颜色不同。
3. 如果所有顶点都被着色，则返回true，表示求解成功；否则返回false，表示无法求解。
4. 如果无法给当前顶点相邻的顶点着不同颜色，回溯到上一个顶点重新选择颜色尝试着色。
5. 不断重复以上步骤，直到所有顶点被着色或者无法进一步着色。

【代码思路】

伪代码

#define TOTAL_VERTICES 6
#define NUM_COLORS 3

int graph[TOTAL_VERTICES][TOTAL_VERTICES] = {
    {0, 1, 1, 0, 0, 0},
    {1, 0, 1, 1, 0, 0},
    {1, 1, 0, 1, 1, 0},
    {0, 1, 1, 0, 1, 1},
    {0, 0, 1, 1, 0, 1},
    {0, 0, 0, 1, 1, 0}
};
int colors[TOTAL_VERTICES]; // Array to store colors for each vertex

int isSafe(int vertex, int color) {
    for (int i = 0; i < TOTAL_VERTICES; i++) {
        if (graph[vertex][i] && color == colors[i]) {
            return 0; // Color conflict
        }
    }
    return 1; // Color is safe
}

int graphColoring(int vertex) {
    if (vertex == TOTAL_VERTICES) {
        return 1; // All vertices are colored
    }

    for (int color = 1; color <= NUM_COLORS; color++) {
        if (isSafe(vertex, color)) {
            colors[vertex] = color;
            if (graphColoring(vertex + 1)) {
                return 1; // Found a valid coloring
            }
            colors[vertex] = 0; // Backtrack
        }
    }
    return 0; // No valid coloring found
}

int main() {
    if (graphColoring(0)) {
        for (int i = 0; i < TOTAL_VERTICES; i++) {
            printf("Vertex %d is colored with color %d\n", i, colors[i]);
        }
    } else {
        printf("No valid coloring found\n");
    }

    return 0;
}

C++实现

#include <iostream>
#include <vector>

#define TOTAL_VERTICES 6
#define NUM_COLORS 3

std::vector<std::vector<int>> graph = {
    {0, 1, 1, 0, 0, 0},
    {1, 0, 1, 1, 0, 0},
    {1, 1, 0, 1, 1, 0},
    {0, 1, 1, 0, 1, 1},
    {0, 0, 1, 1, 0, 1},
    {0, 0, 0, 1, 1, 0}
};
std::vector<int> colors(TOTAL_VERTICES, 0); // Vector to store colors for each vertex

bool isSafe(int vertex, int color) {
    for (int i = 0; i < TOTAL_VERTICES; i++) {
        if (graph[vertex][i] && color == colors[i]) {
            return false; // Color conflict
        }
    }
    return true; // Color is safe
}

bool graphColoring(int vertex) {
    if (vertex == TOTAL_VERTICES) {
        return true; // All vertices are colored
    }

    for (int color = 1; color <= NUM_COLORS; color++) {
        if (isSafe(vertex, color)) {
            colors[vertex] = color;
            if (graphColoring(vertex + 1)) {
                return true; // Found a valid coloring
            }
            colors[vertex] = 0; // Backtrack
        }
    }
    return false; // No valid coloring found
}

int main() {
    if (graphColoring(0)) {
        for (int i = 0; i < TOTAL_VERTICES; i++) {
            std::cout << "Vertex " << i << " is colored with color " << colors[i] << std::endl;
        }
    } else {
        std::cout << "No valid coloring found" << std::endl;
    }

    return 0;
}