백준 11376 - 열혈강호 2(C++)

문제 링크입니다. https://www.acmicpc.net/problem/11376

11376번: 열혈강호 2

이분 매칭을 이용하는 문제입니다.

해당 문제에서 응용된 문제입니다. https://junseok.tistory.com/339

백준 11375 - 열혈강호(C++)

 

이 문제에서 추가된 점은 각 직원이 최대 2개의 일을 할 수 있다는 점입니다.

복잡하게 보이지만 매우 간단한데  X번 직원일 때의 DFS를 2번씩 돌려주면 됩니다.

그렇다면 check 배열을 초기화되기 때문에 DFS를 정상적으로 실행할 수 있으며, 두 번째 연결된 정점을 매칭할 수 있게 됩니다.

 

따라서 이분 매칭을 두 번씩 해주면 되는 문제였습니다.

 

자세한 것은 코드를 참고해주세요

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <cstring>
#include <vector>
#include <algorithm>
#include <queue>
#include <cmath>
#include <cstdio>
#include <string>
#include <deque>
#include <stack>
#include <map>
#define LL long long
#define pp pair<int, int>
#define F first
#define S second

using namespace std;

int N, M;
vector<int> Match[1011];
vector<int> a_match, b_match;
bool check[1011];

bool dfs(int start){
    if(check[start]) return false;
    check[start] = true;
    for(int i = 0; i < Match[start].size(); i++){
        int x = Match[start][i];
        if(b_match[x] == -1 || dfs(b_match[x])){
            a_match[start] = x;
            b_match[x] = start;
            return true;
        }
    }
    return false;
}

void solve(){
    a_match = vector<int>(N+1, -1);
    b_match = vector<int>(M+1, -1);
    int size = 0;
    for(int i = 1; i <= N; i++){
        for(int j = 1; j <= 2; j++){
            memset(check, false, sizeof(check));
            if(dfs(i)) size++;
        }
    }
    cout << size;
}

int main() {
    cin.tie(0);
	cout.tie(0);
	cin >> N >> M;
	for(int i = 1; i <= N; i++){
	    int n;
	    cin >> n;
	    for(int j = 1; j <= n; j++){
	        int x;
	        cin >> x;
	        Match[i].push_back(x);
	    }
	}
	solve();
	return 0;
}

 

 

질문 및 조언은 댓글을 남겨주세요