Gym 215177D 母親節的禮物

Gym 215177D 母親節的禮物

Problem :
給n個點m條邊的無向圖,每一個點的度數小於等於7,要求用4種不一樣的顏色給每一個點染色,使得每一個點相鄰的點中最多隻有一個相同顏色的點。(n<=25000)
Solution :
首先給每一個點進行隨機染色,調整不合法的點的顏色。對於某個不合法的點,必然能夠找到一種顏色只出現過1次或0次,調整爲這種顏色就好了。
考慮不合法的邊的數量,假設一開始隨機全部點的顏色都相同,那麼一共有7n條不合法的邊,而每次調整顏色必然能夠減小至少一條不合法的邊的數量。ios

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <string>
#include <map>
#include <vector>
#include <queue>
#include <ctime>
using namespace std;
#define f(i, x, y) for (int i = x; i <= y; ++i)
#define fd(i, x, y) for (int i = x; i >= y; --i)
#define rep(i, x, y) for (int i = x; i <= y; ++i)
#define repd(i, x, y) for (int i = x; i >= y; --i)

queue <int> Q;

const int INF = 1e9 + 7;
const int N = 200008;
int n, m, q;

vector <int> vec[N];

int dp[N];
int cnt[N], vis[N];

void init()
{
    srand(time(NULL));
    cin >> n >> m;
    for (int i = 1; i <= n; ++i)
        vec[i].clear();
    for (int i = 1; i <= m; ++i)
    {
        int u, v;
        cin >> u >> v;
        vec[u].push_back(v);
        vec[v].push_back(u);
    }   
}
void solve()
{   
    for (int i = 1; i <= n; ++i)
        dp[i] = rand() % 4;
    for (int i = 1; i <= n; ++i) Q.push(i), vis[i] = 1;
    while (!Q.empty())
    {
        int u = Q.front(); Q.pop(); vis[u] = 0;
        for (int i = 0; i < 4; ++i) cnt[i] = 0;
        for (auto v : vec[u]) cnt[dp[v]]++;
        if (cnt[dp[u]] <= 1) continue;
        int qmin = INF, cl = 0;
        for (int i = 0; i < 4; ++i)
            if (cnt[i] < qmin)
            {
                qmin = cnt[i];
                cl = i;
            }   
        dp[u] = cl;
        for (auto v : vec[u])
            if (dp[v] == cl && !vis[v])
                Q.push(v);
    }
    for (int i = 1; i <= n; ++i)
        printf("%c",'a' + dp[i]);
    printf("\n");
}
int main()
{   
    cin.sync_with_stdio(0);
    int T; cin >> T;
    for (int cas = 1; cas <= T; ++cas)
    {
        init();
        solve();
    }
}
相關文章
相關標籤/搜索