[BaekJoon] 삼성 sw 역량 테스트(스타트와 링크)

• 이 문제는 c++로 풀이를 작성하였습니다.

문제

 

예시

 

풀이

•  DFS 이용한 풀이 

#include <iostream>
#include <algorithm>
#include <cmath>
using namespace std;

int n;
int arr[20][20];
int result=987987987;

// dfs로 탐색하여 cnt가 n/2가 되면 각각의 능력치 계산
void dfs(int s,int cnt,bool check[]){
    if(cnt==n/2){
        int sum1=0;
        int sum2=0;
        
        for(int i=0;i<n-1;i++){
            for(int j=i+1;j<n;j++){
                if(check[i] && check[j]){
                    sum1+=arr[i][j];
                    sum1+=arr[j][i];
                }
                if(!check[i] && !check[j]){
                    sum2+=arr[i][j];
                    sum2+=arr[j][i];
                }
            }
        }
        
        result=min(abs(sum1-sum2),result);
        
    }
    else{
        for(int i=s;i<n;i++){
            if(!check[i]){
                check[i]=1;
                dfs(i,cnt+1,check);
                check[i]=0;
            }
        }
    }
}

int main(){
    cin>>n;
    
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            cin>>arr[i][j];
        }
    }
    
    bool check[20]={0};
    dfs(0,0,check);
    cout<<result;
}

 

• Next_permuation 이용한 풀이 

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;


int n;
int arr[20][20];
int result=987987987;

int main(){
    
    cin>>n;
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            cin>>arr[i][j];
        }
    }
    
    vector<int> temp;
    for(int i=0;i<n/2;i++){
        temp.push_back(0);
        temp.push_back(1);
    }
    
    sort(temp.begin(),temp.end());
    
    do{
        int sum1=0;
        int sum2=0;
        for(int i=0;i<n-1;i++){
            for(int j=i;j<n;j++){
                if(temp[i] && temp[j]){
                    sum1+=arr[i][j];
                    sum1+=arr[j][i];
                }
                if(!temp[i] && !temp[j]){
                    sum2+=arr[i][j];
                    sum2+=arr[j][i];
                }
            }
        }
        
        result=min(abs(sum1-sum2),result);
    }while(next_permutation(temp.begin(),temp.end()));
    
    cout<<result;
}

• 추가로 궁금한 점이나 수정할 부분 있으면 댓글로 남겨주세요.