-
boj)14889 - 스타트와 링크PS/boj 2020. 12. 3. 11:26123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263import java.io.*;import java.util.*;public class boj_14889 {static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));static StringTokenizer st;static int N;static int min = Integer.MAX_VALUE;static int[][] map;static ArrayList<Integer> team1 = new ArrayList<>();static ArrayList<Integer> team2 = new ArrayList<>();public static void main(String[] args) throws IOException {N = Integer.parseInt(br.readLine());map = new int[N][N];for (int i = 0; i < N; i++) {st = new StringTokenizer(br.readLine());for (int j = 0; j < N; j++) {map[i][j] = Integer.parseInt(st.nextToken());}}func(0);System.out.println(min);}static void func(int index) {if (index == N) {if (team1.size() != N/2) return;if (team2.size() != N/2) return;int diff = calculate(team1, team2);min = Math.min(min, diff);}if (team1.size() > N/2) return;if (team2.size() > N/2) return;team1.add(index);func(index+1);team1.remove(team1.size()-1);team2.add(index);func(index+1);team2.remove(team2.size()-1);}private static int calculate(ArrayList<Integer> team1, ArrayList<Integer> team2) {int t1 = 0;int t2 = 0;for (int i = 0; i < N/2; i++) {for (int j = 0; j < N/2; j++) {if (i == j) continue;t1 += map[team1.get(i)][team1.get(j)];t2 += map[team2.get(i)][team2.get(j)];}}return Math.abs(t1 - t2);}}
cs - 재귀, 브루트포스
- 현재의 선수를 team1에 넣을지 team2에 넣을지 라고 생각
- N 제한이 20 까지니까 2^20이라 시간제한 충분
- 백트래킹하듯이 team에 넣고 다음 선수 찾기, team에서 빼기 이런식으로 func 계속 호출
- 모든 선수를 팀에 다 배정 완료되었고 인원수도 맞으면 calculate 이후에 min 갱신
1234567891011121314151617181920212223242526272829303132333435public static void main(String[] args) throws IOException {N = Integer.parseInt(br.readLine());map = new int[N][N];for (int i = 0; i < N; i++) {st = new StringTokenizer(br.readLine());for (int j = 0; j < N; j++) {map[i][j] = Integer.parseInt(st.nextToken());}}for (int i = 0; i < (1 << N); i++) {int cnt = 0;for (int j = 0; j < N; j++) {if ((i & (1 << j)) == 0) {cnt += 1;}}if (cnt != N / 2) continue;ArrayList<Integer> team1 = new ArrayList<>();ArrayList<Integer> team2 = new ArrayList<>();for (int j = 0; j < N; j++) {if ((i & (1 << j)) == 0) {team1.add(j);} else {team2.add(j);}}int calculate = calculate(team1, team2);min = Math.min(min, calculate);}System.out.println(min);}cs - 20.12.09 비트마스크 풀이 추가
'PS > boj' 카테고리의 다른 글
boj)2529 - 부등호 (0) 2020.12.08 boj)15661 - 링크와 스타트 (0) 2020.12.06 boj)14501 - 퇴사 (0) 2020.12.02 boj)1759 - 암호 만들기 (0) 2020.12.01 boj)6603 - 로또 (0) 2020.12.01