개발일지

[백준] LCA2(11438) 본문

알고리즘

[백준] LCA2(11438)

devbh 2020. 3. 3. 03:37



import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;
import java.util.StringTokenizer;

public class Main{
    static int N, M;
    static List<Integer>[] map;
    static boolean[] visited;
    static int[] depth;
    static int[][] parent;
    public static void main(String[] args) throws NumberFormatException, IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;

        N = Integer.parseInt(br.readLine().trim());

        map = new ArrayList[N+1];
        visited = new boolean[N+1];
        depth = new int[N+1];
        parent = new int[N+1][21];

        for(int i=1;i<N;i++) {
            st = new StringTokenizer(br.readLine());
            int a = Integer.parseInt(st.nextToken());
            int b = Integer.parseInt(st.nextToken());

            if(map[a]==null) map[a] = new ArrayList<>();
            if(map[b]==null) map[b] = new ArrayList<>();

            map[a].add(b);
            map[b].add(a);
        }

        bfs();
        fillParent();

        M = Integer.parseInt(br.readLine());
        for(int i = 0;i<M;i++) {
            st = new StringTokenizer(br.readLine());
            int a = Integer.parseInt(st.nextToken());
            int b = Integer.parseInt(st.nextToken());

            System.out.println(lca(a,b));
        }
    }
    static int lca(int a, int b) {
        int low, high;

        if(depth[a] < depth[b]) {low=b;high=a;}
        else {low=a;high=b;}

        for(int i=20;0<=i;i--) {
            int diff = depth[low] - depth[high];
            if(Math.pow(2, i) <= diff) {
                low = parent[low][i];
            }
        }

        if(low == high) return high;

        for(int i=20;0<=i;i--) {
            if(parent[low][i] != parent[high][i]) {
                low = parent[low][i];
                high = parent[high][i];
            }
        }

        high = parent[high][0];

        return high;
    }

    static void fillParent() {
        for(int b=1;b<=20;b++) {
            for(int a=1;a<=N;a++) {
                parent[a][b] = parent[parent[a][b-1]][b-1];
            }
        }
    }

    static void bfs() {
        Queue<Integer> q = new LinkedList<Integer>();
        q.add(1);
        visited[1] = true;
        int d = 0;

        while(!q.isEmpty()) {
            d++;

            int qSize = q.size();
            for(int i =0;i<qSize;i++) {
                int now = q.poll();

                if(map[now]==null)continue;

                int mSize = map[now].size();
                for(int j=0;j<mSize;j++) {
                    int next = map[now].get(j);
                    if(visited[next]==false) {
                        parent[next][0] = now;
                        q.add(next);
                        visited[next] = true;
                        depth[next] = d;
                    }
                }
            }
        }
    }

}

'알고리즘' 카테고리의 다른 글

[백준] 게임 개발(1516)  (0) 2020.03.19
[백준] 키 순서(2458)  (0) 2020.03.19
[백준] 네트워크 연결 (1922)  (0) 2020.03.03
[백준] 줄 세우기(2252)  (0) 2020.03.03
[백준] 집합의 표현 (1717)  (0) 2020.02.21
Comments