13-g0rnn

[g0rnn]

🔗 문제 링크

스타트와 링크

✔️ 소요된 시간

45m

❗ 문제가 어렵다면 혹은 이해가 안되면 혹은 애매하다면

백트레킹에서 N과 M 1번과 2번을 풀어보심이 좋을 듯합니다.

위 문제가 왜 백트레킹에 해당하는지 모르겠다면 해당 문제를 설명해주는 사이트에 들어가서 풀이와 함께 개념을 잡아보는 것을 추천드립니다!

개념을 잘 잡았다면 1번을 풀고 2번 풀 때에 3초 안에 해결할 수 있습니당 ^^

✨ 수도 코드

이번 문제는 조합(combination)을 구현하는 문제 같아 보입니다. 분류는 백트레킹이지만 저는 조합을 구현해내는 게 핵심이라고 생각합니다.

예전에 제가 올린 pr에서 dfs를 활용하여 개수를 세는 코드가 있었습니다. 이것도 그것처럼 dfs를 재귀적으로 호출하고 작업이 끝나면 해당 함수에서 수정한 내역을 복구하면 쉽게 풀립니다.

문제를 풀 수 있는 아이디어와 수도 코드를 소개하며 제 풀이를 보여드리겠습니다.

[idea]
1. 배열 index를 사용하여 사람을 구별하고 각 엔트리에는 방문 여부(혹은 선택 여부)를 저장한다.
2. n명의 사람 중 절반을 선택하면 팀이 나눠진 것이다.

[pseudo code]
1. dfs를 통해 n/2명의 사람을 뽑는다.
  - index를 중복하여 방문하지 않도록 주의
  - 사람을 뽑고 작업을 마치면 원상복귀
  - n/2 명을 뽑은 후엔 포인트 계산
2. 포인트는 visited를 순회하며 true인것과 false인 것을 구분하여 합한다.
3. 최소 점수 차를 업데이트 한다.

코드를 보았을 때 dfs 함수와 sumPoint 함수의 동작 방식이 같다는 것을 눈치해야합니다. dfs를 해체해보면 for문이 아마 n개 중첩되어 있을 겁니다. 그리고 각 for문은 이전 for문의 index 뒤에서 시작합니다.

📚 새롭게 알게된 내용

clion에서는 #include <vector>를 하지 않아도 벡터를 사용할 수 있다..

[kangrae-jo]

유튜브 보면서 해서 1시간이나 넘게 걸렸습니다...
그리고 예전에 시험 문제로 마주했던 조합을 구성하는 문제를 보니 꼭 혼자 힘으로 풀고싶었습니다.
(그리고 성공했습니다 ㅎㅎ)

2개로 분류하는 문제라고 생각했고, 지금까지 자주 써왔던 코드로 구현해보았습니다.

int dfs() {
    if (base case) return ~ ; 

    flag[k] = true;
    int a = dfs(flag, k + 1, count + 1);
    flag[k] = false;
    int b = dfs(flag, k + 1, count);

    return ~ ;
}

이런 식으로 bool배열에 true 와 false를 넣으며 basecase에 도달했을떄 조건을 검사합니다.
그리고 조건에 일치하면 문제에 맞는 답을 구하고 return 합니다. (여기서는 Ability의 Diff)
그 다음 백트래킹하며 최소 diff를 찾는 로직입니다.

그리고
제가 구현한 팀원의 능력치를 구하는 코드는 알아보기는 쉽지만 조금 긴 것 같습니다.
균호님은 visited와 s의 인덱스를 잘 조합해서 한 번에 더하셨네요. 좋은 것 같습니다.

CPP CODE

#include <algorithm>
#include <climits>
#include <iostream>
#include <vector>

using namespace std;

int N;
vector<vector<int>> team;

int getAbilityDiff(vector<bool>& flag) {
    vector<int> myTeam, yourTeam;
    for (int i = 1; i <= N; i++) {
        if (flag[i]) myTeam.push_back(i);
        else yourTeam.push_back(i);
    }

    int my = 0, your = 0;
    for (int y : myTeam) {
        for (int x : myTeam) 
            my += team[y][x];
    }
    for (int y : yourTeam) {
        for (int x : yourTeam) 
            your += team[y][x];
    }
    
    return abs(my - your);
}

int dfs(vector<bool>& flag, int k, int count) {
    if (count > N / 2) return INT_MAX;
    if (k == N) {
        if (count == N / 2) return getAbilityDiff(flag);
        return INT_MAX;
    }
    flag[k] = true;
    int select = dfs(flag, k + 1, count + 1);
    flag[k] = false;
    int notSelect = dfs(flag, k + 1, count);

    return min(select, notSelect);
}

int main() {
    cin >> N;

    team = vector<vector<int>>(N + 1, vector<int>(N + 1, 0));
    for (int y = 1; y <= N; y++)
        for (int x = 1; x <= N; x++) 
            cin >> team[y][x];

    vector<bool> flag(N + 1, false);
    cout << dfs(flag, 1, 0);

    return 0;
}

[kokeunho]

조합 문제는 처음인데
확실히 인원 수를 나눈 조합을 구현하는게 쉽사리 떠오르지 않더라구요.
그래서 균호님의 풀이를 참조했고 덕분에 백트래킹에 대해 공부하게 되었습니다.
코멘트 내용도 늘릴 겸 공부한 내용 짤막하게 남겨봅니다.
백트래킹
가능한 모든 해를 탐색하는 알고리즘으로 일반적으로 재귀를 사용(dfs 기반)
장점: 조건에 맞지 않는 경로에서는 되돌아 가(backTracking) 불필요한 탐색을 줄임
구조: 1) 종료 조건 확인
2) 가능성 판단(가지치기)
3) 재귀 호출
4) 백트래킹 (탐색 완료 후 원래 상태로 돌아감)

이번 PR 도 수고하셨습니다.