回溯法-地图着色问题

一、问题描述

地图上某些区域挨着,挨着的区域不能是相同的颜色,如果使用m中材料对地图着色,共有多少种着色方式

二、解决思路

地图可以看做是图,地图上的位置相当于图上的点,如果挨着代表点与点之间有边,否则没有,使用回溯发,确定限界条件个剪枝函数,不过当前问题是求解所有的方案,所以没有必要需要剪枝函数。当问题到达子节点则代表是一个可行解。到达之后进行回溯,求解其他的方案。


clipboard.png

三、代码

public class MapColoring {

   public static void main(String[] args) {
       //邻接矩阵存储图 0代表相连否则不相连
       int[][] adjacencyMatrix = new int[][]{
               {0, 1, 1, 1, 0, 0, 0},
               {1, 0, 1, 0, 1, 0, 0},
               {1, 1, 0, 1, 1, 0, 0},
               {1, 0, 1, 0, 1, 0, 1},
               {0, 1, 1, 1, 0, 1, 1},
               {0, 0, 0, 0, 1, 0, 1},
               {0, 0, 0, 1, 1, 1, 0},
       };
       //记录所有的结果
       int[][] allResult = new int[adjacencyMatrix.length][adjacencyMatrix.length];
       int[] currentValue = new int[adjacencyMatrix.length];
       coloring(adjacencyMatrix, allResult, 0, currentValue, 3, 0);
       for (int i = 0; i < allResult.length; i++) {
           System.out.println(Arrays.toString(allResult[i]));
       }


   }

   /**
    * 地图着色
    *
    * @param adjacencyMatrix 邻接矩阵
    * @param allResult       所有的结果集
    * @param i               当前点
    */
   public static int coloring(int[][] adjacencyMatrix, int[][] allResult, int i, int[] currentValue, int color, int resultIndex) {

       /**
        * 判断当前节点的着色
        * 循环已经着色的点 ,如果相连则判断能否着色当前颜色
        * 循环颜色
        */

       if (i > adjacencyMatrix.length - 1) {
           //将当前结果保存到全部结果中
           for (int j = 0; j < currentValue.length; j++) {
               allResult[resultIndex][j] = currentValue[j];
           }
           resultIndex += 1;
           return resultIndex;
       }

       for (int j = 1; j <= color; j++) {
           // 记录当前节点的颜色尝试
           if (isOk(adjacencyMatrix, j, currentValue, i)) {
               // 当前节点染色j
               currentValue[i] = j;
               //向下递归 如果这条路最终能走到叶子节点,则会加入到allResult
               resultIndex = coloring(adjacencyMatrix, allResult, i + 1, currentValue, color, resultIndex);
               //由于这里是for循环尝试其他的道路,所以不用手动回溯,下次currentValue[i] = j;  会将本次循环值覆盖,直到达到叶子节点保存结果
           }
       }
       return resultIndex;
   }


   /**
    * 判断当前节点是否可以进行找色
    * 循环已经作色的点 当相邻再判断颜色是否相等
    *
    * @param adjacencyMatrix 邻接矩阵
    * @param colorIndex      判断的颜色
    * @param currentValue    当前值列表
    * @param i               当前节点
    * @return
    */
   public static boolean isOk(int[][] adjacencyMatrix, int colorIndex, int[] currentValue, int i) {
       for (int j = 0; j < i; j++) {
           if (adjacencyMatrix[j][i] == 1) {
               if (currentValue[j] == colorIndex) {
                   return false;
               }
           }
       }
       return true;
   }

四、优化空间

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容