简单染色问题:不邻接植花

简单染色问题:不邻接植花

问题:

在这里插入图片描述
在这里插入图片描述

思路:

对图中的节点建立邻接表,设解向量的初值均为0,即均未染色,使用容器set存储4种颜色

按顺序遍历图中的所有节点,在set中去除所有该节点的邻接表中节点已用过的颜色( erase(key_value) 删除键值为key_value的值 ),将set中的第一种颜色赋给该节点

处理完所有节点后,返回解向量

代码:

class Solution {
public:
    vector<int> gardenNoAdj(int N, vector<vector<int>>& paths) {
        vector<vector<int>> table;
        int i,j;
        for( i=0;i<N;i++ )
        {
            table.push_back( vector<int>() );
        }
        for( i=0;i<paths.size();i++ )
        {
            table[ paths[i][0]-1 ].push_back( paths[i][1]-1 );
            table[ paths[i][1]-1 ].push_back( paths[i][0]-1 );
        }
        vector<int> ans(N,0);
        for( i=0;i<N;i++ )
        {
            set<int> color={1,2,3,4};
            for( j=0;j<table[i].size();j++ )
            {
                color.erase( ans[ table[i][j] ] );
            }
            ans[i]=*color.begin();
        }
        return ans;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_44709990/article/details/106216890
今日推荐