쵼쥬
쵼쥬의 개발공부 TIL
쵼쥬
전체 방문자
오늘
어제
  • 분류 전체보기 (276)
    • 코딩테스트 (192)
      • [알고리즘] 알고리즘 정리 (7)
      • [백준] 코딩테스트 연습 (126)
      • [프로그래머스] 코딩테스트 연습 (59)
    • Spring (71)
      • [인프런] 스프링 핵심 원리- 기본편 (9)
      • [인프런] 스프링 MVC 1 (6)
      • [인프런] 스프링 MVC 2 (4)
      • [인프런] 실전! 스프링 부트와 JPA 활용1 (7)
      • [인프런] 실전! 스프링 부트와 JPA 활용2 (5)
      • [인프런] 실전! 스프링 데이터 JPA (7)
      • [인프런] 실전! Querydsl (7)
      • JWT (5)
      • [인프런] Spring Cloud (17)
      • [인프런] Spring Batch (4)
    • Java (6)
      • [Java8] 모던인자바액션 (4)
      • [부스트코스] 웹 백엔드 (2)
      • [패스트캠퍼스] JAVA STREAM (0)
    • CS (6)
      • 디자인 패턴과 프로그래밍 패터다임 (2)
      • 네트워크 (4)

블로그 메뉴

  • 홈

공지사항

인기 글

태그

  • 코딩테스트
  • BFS
  • 스프링
  • querydsl
  • 부스트코스
  • 프로그래머스
  • MVC
  • 자바
  • 타임리프
  • 위클리 챌린지
  • 비트마스킹
  • jpa
  • 알고리즘
  • 백분
  • 구현
  • Spring Data JPA
  • 백준
  • 인프런
  • 누적합
  • spring

최근 댓글

최근 글

티스토리

hELLO · Designed By 정상우.
쵼쥬
코딩테스트/[백준] 코딩테스트 연습

게리맨더링 - 17471번

게리맨더링 - 17471번
코딩테스트/[백준] 코딩테스트 연습

게리맨더링 - 17471번

2022. 4. 6. 12:33


내 코드

package com.company;

import java.io.*;
import java.util.*;

public class Main {
    static int N, answer;
    static int[][] arr;
    static int[] p;
    static boolean[] visited;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = null;
        N = Integer.parseInt(br.readLine());
        p = new int[N + 1];
        arr = new int[N + 1][N + 1];
        visited = new boolean[N + 1];
        answer = -1;

        st = new StringTokenizer(br.readLine());
        for (int i = 1; i <= N; i++) {
            p[i] = Integer.parseInt(st.nextToken());
        }

        for (int i = 1; i <= N; i++) {
            st = new StringTokenizer(br.readLine());
            int x = Integer.parseInt(st.nextToken());
            for (int j = 0; j < x; j++) {
                int y = Integer.parseInt(st.nextToken());
                arr[i][y] = 1;
                arr[y][i] = 1;
            }
        }
        dfs(1, 0);
        System.out.println(answer);
    }

    static void dfs(int index, int x) {
        int a = bfs(x);
        int b = bfs((int) Math.pow(2, N) - 1 - x);
        if (a != -1 && b != -1 && (answer == -1 || answer > Math.abs(a - b))) {
            answer = Math.abs(a - b);
        }

        for (int i = index; i <= N; i++) {
            dfs(i + 1, x + (1 << i - 1));
            dfs(i + 1, x);
        }

    }

    static int bfs(int x) {
        int index = 0;
        for (int i = 1; i <= N; i++) {
            if ((1 << (i - 1) & x) > 0) {
                index = i;
                break;
            }
        }
        Queue<Integer> q = new LinkedList<>();
        q.add(index);
        x -= 1 << (index - 1);
        int total = 0;

        while (!q.isEmpty()) {
            int a = q.poll();
            total += p[a];

            for (int i = 1; i <= N; i++) {
                if (arr[a][i] == 1 && (x & 1 << (i - 1)) > 0) {
                    x -= 1 << (i - 1);
                    q.add(i);
                }
            }
        }
        return x == 0 ? total : -1;
    }
}

'코딩테스트 > [백준] 코딩테스트 연습' 카테고리의 다른 글

드래곤 커브 - 15685번  (0) 2022.04.07
미네랄 - 2933번  (0) 2022.04.06
스도쿠 - 2239번  (0) 2022.04.06
사다리 조작 - 15684번  (0) 2022.04.05
톱니바퀴 - 14891번  (0) 2022.04.04
    '코딩테스트/[백준] 코딩테스트 연습' 카테고리의 다른 글
    • 드래곤 커브 - 15685번
    • 미네랄 - 2933번
    • 스도쿠 - 2239번
    • 사다리 조작 - 15684번
    쵼쥬
    쵼쥬

    티스토리툴바

    단축키

    내 블로그

    내 블로그 - 관리자 홈 전환
    Q
    Q
    새 글 쓰기
    W
    W

    블로그 게시글

    글 수정 (권한 있는 경우)
    E
    E
    댓글 영역으로 이동
    C
    C

    모든 영역

    이 페이지의 URL 복사
    S
    S
    맨 위로 이동
    T
    T
    티스토리 홈 이동
    H
    H
    단축키 안내
    Shift + /
    ⇧ + /

    * 단축키는 한글/영문 대소문자로 이용 가능하며, 티스토리 기본 도메인에서만 동작합니다.