UVA11080- Place the Guards(二分圖染色)

UVA11080- Place the Guards(二分圖染色)

題目鏈接

題意：放最少的士兵去監視所有的道路, 但士兵不可相鄰，符合的話，就輸出最少的士兵數，否則輸出-1

思路：其實就是二分圖染色，即黑白染色，然後選擇黑白染色最少的那個顏色累加，但要注意可能有多個連通塊，只要有一個連通塊不符合的話，就不符合。

代碼：

#include 
#include 
#include 
#include 
#include 

using namespace std;

const int MAXN = 205;

vector g[MAXN];

int color[MAXN];
int v, e, cur, num;

bool bipartite(int u, int &cur, int &num) {
    num++;
    if (color[u] == 1) cur++;
    for (int i = 0; i < g[u].size(); i++) {
        int v = g[u][i]; 
        if (color[v] == color[u]) return false;
        if (!color[v]) {
            color[v] = 3 - color[u]; 
            if (!bipartite(v, cur, num)) return false;
        }
    }
    return true;
}

int main() {
    int cas;
    scanf("%d", &cas);
    while (cas--) {
        scanf("%d%d", &v, &e);    
        for (int i = 0; i < v; i++)
            g[i].clear();
        int a, b;
        for (int i = 0; i < e; i++) {
            scanf("%d%d", &a, &b); 
            g[a].push_back(b);
            g[b].push_back(a);
        } 

        int flag = 1;
        int ans = 0;
        memset(color, 0, sizeof(color));
        for (int i = 0; i < v; i++) {
            if (!color[i]) {
                num = cur = 0;
                color[i] = 1;
                if (!bipartite(i, cur, num)) { 
                    flag = 0;
                    break;
                }
                if (num == 1) ans++;
                else {
                    ans += min(cur, num - cur); 
                } 
            }
        } 

        if (flag)
            printf("%d\n", ans);
        else
            printf("-1\n");
    }
    return 0;
}