3780. 네트워크 연결


문제

종빈이는 아주 큰 그룹의 총수다. 이 그룹은 1부터 N번까지의 번호로 구분할 수 있는 N개의 기업을 운영하고 있다. 현재 각 기업은 서로 독립적인 자체 컴퓨팅 및 통신센터를 가지고 있다.

어느 날 종빈이는 계열사의 CTO인 서현이에게 서비스 개선을 위해 각 기업의 서버를 네트워크로 연결하여 단일 통신센터에서 관리 가능한 클러스터 형태로 구성할 것을 제안했다. 종빈이의 제안을 들은 서현이는 다음과 같은 병합 과정을 고안해냈다.

  1. 클러스터 A를 제공하는 기존에 존재하는 센터 I를 고른다.
  2. 클러스터 B를 제공하는 기업 J를 고른다. B는 A가 아닌 임의의 클러스터이며, J는 센터가 아닐 수 있다.
  3. I와 J를 통신 라인으로 연결한다. 이때 기업 I와 J를 잇는 라인의 길이는 |I – J|(mod 1000)이다.
  4. 위 방식을 통해 클러스터 A와 B는 새로운 클러스터로 합쳐지며, 이 클러스터는 B의 센터에 의해 제공된다.

이러한 병합 과정을 거치던 중에, 각 기업에서 현재 센터까지 연결되는 라인의 길이가 총 얼마나 되는지에 관한 문의가 들어왔다. 서현이를 위해 병합하는 과정과 그 과정에서 통신센터와 각 기업을 잇는 라인의 길이를 구하는 프로그램을 작성해보자.

입력

입력은 여러 개의 테스트케이스로 주어진다. 입력의 첫 번째 줄에는 테스트 케이스의 개수 T가 주어진다. 각 테스트 케이스에는 기업의 수를 나타내는 N(5 ≤ N ≤ 20,000)이 주어진다. 다음은 몇 개의 줄에 걸쳐 아래 두 가지 종류의 명령어가 들어온다.

  • E I – 기업 I와 현재 I의 센터까지의 거리를 출력한다. 
  • I I J – 센터 I를 기업 J에 연결한다.

각 테스트케이스의 끝에는 단어 O가 주어진다. 각 테스트케이스에서 명령어의 총 개수는 200,000개를 넘지 않으며, 그중 I 명령어의 개수는 N개보다 작다.

출력

E 명령어가 들어올 때마다 한 줄에 해당 거리를 출력한다.

예제 입력 

1
4
E 3
I 3 1
E 3
I 1 2
E 3
I 2 4
E 3
O

예제 출력 

0
2
3
5


1. 접근


각 기업별로 서버를 갖고 있다. 서버는 네트워크로 연결될 수 있고, 네트워크에는 센터 서버가 있다.

따라서 현재 기업마다의 센터는 단일 네트워크이자, 센터 서버이다.


이제 네트워크들을 I 명령어로 연결하고자 한다.

I 1 2 라면, 센터 서버인 1서버와, 2서버를 연결한다. 주의할 점은 첫번째 서버는 반드시 센터 서버이고,

두번째 서버는 그 서버가 속한 네터워크의 센터 서버가 아닐 수도 있다.


또한 연결된 이후에는, 두 네트워크는 하나의 네트워크가 되며, 센터 서버는 두번 째 서버가 담당한다.


우리는 서버들을 노드로 표현할 수 있으며, 서버들의 종속관계를 간선으로 표현할 수 있다.

간선의 크기는 |i - j|(mod 1000)으로 지정된다.


이때 한 기업이 주어지면, 기업의 서버에서, 그 서버의 센터 서버까지의 경로의 길이 합을 구해야 한다.

그래프적인 방법으로 접근해보자.



2. 풀이


네트워크는 집합으로, 센터는 루트 노드로 생각한다면, 바로 디스조인트-셋을 사용할 생각이 든다.

센터노드는 하나이므로, 노드에서 나가는 간선은 최대 한개이기 때문이다.


하지만 문제는 길이 합을 구해야 하기 때문에, 매번 경로를 탐색한다면,

명령어의 수(200,000) * 최악의 경로(20,000)으로 시간초과가 날 것이다.


따라서 디스조인트의 연산을 활용하여, 연결 시 마다 네트워크의 종속관계를 갱신해주고, 길이도 갱신해보도록 하자.


연결을 담당하는 union()시 i는 기존의 센터이고, j는 새로운 센터가 될 서버이므로 매우 쉽게 연결과 길이 정의가 가능하다.


이제 find()가 길이 합을 계산해주는 역할을 해야 한다. find는 부모를 재귀적으로 따라가서 최종적으로 루트노드를 반환해주는 연산이다.

우리가 원하는 연산은 센터를 제외한 나머지 노드의 부모를 센터로 지정하고, 길이도 갱신해줘야 한다.

또한 x서버에서 센터까지 가는 경로는 유일하므로, find를 조금 바꿔주면 가능하다.


재귀적으로 루트노드 갱신을 해주는 원래 기능에, 길이 갱신만 추가하면 된다.

len[x] += len[parent[x]]


다음 예시로 어떤 동작을 해야하는지 알아보자.


I 1, 2

I 2, 3

E 1


위와 같은 경우에, union()을 두번 수행한 결과 1번 노드는 2번 노드를, 2번 노도는 3번 노드를 가리키고 있을 것이다.

이제 find(1)을 수행하면, 1번 노드는 3번 노드를 가리켜야 하고, 길이는 2번 노드의 것을 추가하여 갱신되어야 한다.



3. 코드


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
#include <stdio.h>
#include <algorithm>
#include <math.h>
using namespace std;
 
typedef pair<intint> p;
int t, n, a, b, c, i;
int parent[20001];
int len[20001];
 
int find(int x) {
    if (parent[x] == x)
        return x;
    else {
        int tmp = find(parent[x]);
        len[x] += len[parent[x]];
        parent[x] = tmp; 
 
        return tmp;
    }
}
 
void unite(int x, int y) {
    len[x] = abs(x - y) % 1000;
    parent[x] = y;
}
 
int main() {
    scanf("%d"&t);
    while (t--) {
        scanf("%d"&n);
        for (i = 0; i <= n; ++i) {
            parent[i] = i;
            len[i] = 0;
        }
        while (1) {
            scanf(" %c"&c);
            if (c == 'E') {
                scanf("%d"&a);
                find(a);
                printf("%d\n", len[a]);
            }
            else if (c == 'I') {
                scanf("%d %d"&a, &b);
                unite(a, b);
            }
            else
                break;
        }
    }
    return 0;
}
cs


재귀를 알맞게 구성해내는 경험이 더러 필요하다.

'알고리즘 > Disjoint-set 디스조인트-셋' 카테고리의 다른 글

백준) 10775 공항  (0) 2017.07.29
백준) 9938 방 청소  (0) 2017.07.29
백준) 4195 친구 네트워크  (0) 2017.07.28
백준) 1976 여행 가자  (0) 2017.07.28
백준) 1717 집합의 표현  (0) 2017.07.28

10775 공항


문제

오늘은 신승원의 생일이다.

박승원은 생일을 맞아 신승원에게 인천국제공항을 선물로 줬다.

공항에는 G개의 게이트가 있으며 각각은 1에서 G까지의 번호를 가지고 있다.

공항에는 P개의 비행기가 순서대로 도착할 예정이며, 당신은 i번째 비행기를 1번부터 gi (1 ≤ gi ≤ G) 번째 게이트중 하나에 영구적으로 도킹하려 한다. 비행기가 도킹된 게이트에는 다른 비행기가 도착할 수 없다.

이렇게 공항을 운영할 경우 간혹 비행기가 어떤 곳에도 도킹하지 못하는 사고가 발생한다. 이러한 사고가 일어나면 공항의 평판이 급속히 나빠져, 이후 어떤 비행기도 도착하지 않으려 할 것이다.

신승원은 가장 많은 비행기를 공항에 도킹시켜서 박승원을 행복하게 하고 싶어한다. 승원이는 비행기를 최대 몇 대 도킹시킬 수 있는가?

입력

첫번째 줄에는 게이트의 수 G (1 ≤ G ≤ 105)가 주어진다.

두번째 줄에는 비행기의 수 P (1 ≤ P ≤ 105)가 주어진다.

이후 P개의 줄에 gi (1 ≤ gi ≤ G) 가 주어진다.

출력

승원이가 도킹시킬 수 있는 최대의 비행기 수를 출력한다.

예제 입력 

4
3
4
1
1

예제 출력 

2

예제 입력 2 

4
6
2
2
3
3
4
4

예제 출력 2 

3



1. 접근


비행기가 순차적으로 게이트에 도킹하며, 한 게이트에는 한 대만 도킹 할 수 있다.


비행기의 도착 게이트 번호(g)가 지정되면, 1번 부터 g번 까지 빈 게이트에 도킹할 수 있을 때, 최대한 많이 도킹시키고자 한다.


게이트를 노드로 생각하고, 간선은 이미 도킹된 게이트에 도착한 비행기의 차선책으로 생각해보면, 그래프적인 문제로 이해할 수 있다.


방 청소 문제 https://www.acmicpc.net/problem/9938 와 많이 비슷한 문제로, 디스조인트 셋을 활용해보자.



2. 풀이


비행기가 g번 보다 큰 게이트에는 도킹할 수 없다는 점을 주목하자.


직관적으로 최대한 많이 도킹시키려면 바로 g번 게이트에 도킹시키는게 효율적일 것이다.


비행기(g=2)과 비행기(g=1)가 주어진다면, 2번 게이트에 넣고, 1번 게이트에 도킹시켜야 둘다 도킹 시킬 수 있다.



방 청소 문제를 복기해보면, 간선을 비행기의 차선책을 나타내는 방법으로 사용했었다.


그러면 빈 노드를 find() 해보면 자신이 나올 것이고, 이미 도킹되어 있었으면 g보다 작은 게이트가 나올 것이다.


전략을 g가 비어있지 않으면 g-1에 도킹시키기기로 한다면, 여러 비행기들을 도킹시킨 후 새로운 비행기를 도킹시킬 때,


도킹 시킬 수 없음을 확인할 수 있는 방법은 무엇일까?


find(g)를 했을 때 0이 나오는지 확인하면 될 것이다. 현재 루트 노드는 차선책을 나타내고 있으므로, 0이 나온다면 도킹 불가능이다. 



3. 코드


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
#include <stdio.h>
using namespace std;
 
int g, p, a, i, cnt, empty_gate;
int parent[100001];
 
int find(int x) {
    if (x == parent[x])
        return x;
    else
        return parent[x] = find(parent[x]);
}
 
void unite(int x, int y) {
    x = find(x);
    y = find(y);
    parent[x] = y;
}
 
int main() {
    scanf("%d %d"&g, &p);
    for (i = 1; i <= g; ++i)
        parent[i] = i;
    for (i = 0; i < p; ++i) {
        scanf("%d"&a);
        empty_gate = find(a);
        if (empty_gate == 0)
            break;
        else {
            cnt++;
            unite(empty_gate, empty_gate - 1);
        }
    }
    printf("%d\n", cnt);
    return 0;
}
cs


'알고리즘 > Disjoint-set 디스조인트-셋' 카테고리의 다른 글

백준) 3780 네트워크 연결  (0) 2017.07.30
백준) 9938 방 청소  (0) 2017.07.29
백준) 4195 친구 네트워크  (0) 2017.07.28
백준) 1976 여행 가자  (0) 2017.07.28
백준) 1717 집합의 표현  (0) 2017.07.28

9938. 방 청소


문제

은기는 술병 N개(1부터 N까지 번호가 매겨져 있다)와 서랍 L개(1부터 L까지 번호가 매겨져 있다)를 가지고 있다. 술병은 은기의 방 바닥에 흩어져 있고, 어린이날을 맞이해 방 청소를 하려고 한다.  서랍에는 술병이 하나 들어갈 수 있다. 나중에 원하는 술을 빠르게 찾을 수 있게 하기 위해 은기는 각각의 술병이 들어갈 수 있는 서랍의 번호 Ai와 Bi를 공책에 적어 놓았다.

은기는 술병을 1번부터 N번까지 순서대로 정리할 것이고, 각각의 술병에 대해서 다음과 같은 과정을 거친다.

  1. 서랍 Ai가 비어있다면, i번 술을 그 서랍에 보관한다.
  2. 서랍 Bi가 비어있다면, i번 술을 그 서랍에 보관한다.
  3. Ai에 들어있는 술을 다른 서랍으로 이동시킨다.(다른 서랍은 Ai에 들어있는 술이 들어갈 수 있는 서랍 중 하나이다) 만약, 그 서랍에도 이미 술이 들어있다면, 그 술을 다른 서랍으로 이동시킨다. 이런 과정을 거쳐서 빈 서랍을 하나 찾아 술을 모두 이동할 수 있는 경우에는, 술을 이동시키고 i번 술을 Ai에 보관한다. 불가능한 경우에는 다음 규칙으로 넘어간다.
  4. Bi에 들어있는 술을 다른 서랍으로 이동시킨다. 만약, 그 서랍에도 이미 술이 들어있다면, 그 술을 다른 서랍으로 이동시킨다. 이런 과정을 거쳐서 빈 서랍을 하나 찾아 술을 모두 이동할 수 있는 경우에는, 술을 이동시키고 i번 술을 Bi에 보관한다. 불가능한 경우에는 다음 규칙으로 넘어간다.
  5. 위의 과정이 모두 불가능한 경우에는 i번 술을 그 자리에서 마셔버린다. (은기는 전혀 취하지 않는다)

각각의 술에 대해서, 서랍에 보관할 수 있는지, 그 자리에서 마셔버리는지를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N과 L이 주어진다. (1 ≤ N, L ≤ 300,000)

다음 N개 줄에는 Ai와 Bi가 주어진다. (1 ≤ Ai, Bi ≤ L, Ai ≠ Bi)

출력

1번 술부터 N번 술까지 순서대로 보관할 수 있는지, 그 자리에서 먹어야 하는지를 출력한다.

보관할 수 있는 경우에는 "LADICA"를, 먹어버려야 하는 경우에는 "SMECE"를 출력한다.

예제 입력 

9 10
1 2
3 4
5 6
7 8
9 10
2 3
1 5
8 2
7 9

예제 출력 

LADICA
LADICA
LADICA
LADICA
LADICA
LADICA
LADICA
LADICA
LADICA


1. 접근


L개의 빈 서랍이 있다. 서랍은 술병을 단 하나만 보관할 수 있으며, 1~N번이 적힌 술병은 보관될 수 있는 서랍이 A/B로 지정되어 있다.


술병들의 보관장소가 1번 술병부터 주어지고 규칙에 따라 바로 보관시킬 때, 보관할 수 있는지 없는지 확인해야 한다.


규칙은 다음과 같다.


1) A 서랍이 비었으면 바로 A에 보관

2) B 서랍이 비었으면 바로 B에 보관

3) A 서랍이 비지 않았으면 A서랍이 보관 중인 술병이 보관될 수 있는 다른 서랍으로 연쇄적으로 옮긴다.

즉, 최종적으로 빈 서랍을 찾을 때 까지 계속 술병을 규칙에 따라 옮겨야 한다.

4) B 서랍이 비지 않았다면 3)과 동일 하게 옮긴다.

5) 옮길 수 없다.


서랍을 노드로 생각하고, A/B를 간선이라 생각해보면

예제의 첫째 줄 " 1 2 " 는 1번 노드와 2번 노드를 잇는 간선으로 이해할 수 있다.

1번 노드가 불가능하면(이미 보관 중인 서랍이면) 1)규칙이 불가능하므로 2)규칙을 시도한다. (유향 간선을 타고 다음 노드로 간다)

2번 노드도 불가능하면 2번 노드에서 출발하는 다음 간선으로 가는 과정으로 이해할 수 있다.



2. 풀이


그래프적 분석을 완료했으니 실제로 그래프를 구현하고 매번 탐색을 시도하는 방법을 생각해 볼 수도 있다.


다음 예시 입력을 통해 어떤 변화가 생기는지 알아보자.


4 3

1 2

1 3

1 2

1 3

1번 술병은 곧바로 1번 서랍에 보관되고(주황색으로 표시), 다음 차선책으로 2번 서랍을 가리키고 있다.

1번 노드의 루트노드가 2다.

이제 2번 술병의 보관 장소가 주어지는데, 하필 1번 술병이 보관되고 있는 1번 서랍을 1순위로 두고 있다.

따라서 1번 술병을 차선책인(간선을 타고 이동해서) 2번 서랍에 보관하고, 2번 술병은 1번 서랍에 보관하자.

또한 2번 술병의 차선책인 3번 서랍을 간선으로 이어준다.

1번 노드의 루트 노드가 3으로 변경되었다.

3번 술병도 1순위로 1번 서랍을 원한다. 다행히 현재 1번 서랍이 보관하는 2번 술병은 3번 서랍에 보관될 수 있으므로 옮겨준다.

이제 1번 서랍은 3번 술병을 보관할 수 있다. 차선책인 2번 서랍을 간선으로 이어준다. 1번 노드의 루트 노드가 변경되었다.

당연히 이제 1번 서랍을 1순위로 원하는 술병이 주어지면 어디에도 보관될 수가 없다.("SMECE")


착안점은 한 노드에서 나가는 간선은 최대 1개라는 점이다. 이 점에서 우리는 디스조인트 셋을 쓸 수 있고, 속도의 이점을 얻는다.

대신 간선들의 모양은 위의 그림들과는 다르게 적용될 것이다.

( (1/2) 이후 (1/3)을 입력받으면 2에서 3으로 간선이 생긴다 )

루트노드가 중요한 역할을 한다. 연쇄적으로 수행되는 3), 4) 규칙의 최종적인 빈서랍 역할을 하는 것이다.


노드가 보관 중인지 아닌지는 간단히 배열로 나타낸다.


3. 코드


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
#include <stdio.h>
using namespace std;
 
int n, l, a, b, i;
bool taken[300001];
int parent[300001];
 
int find(int x) {
    if (x == parent[x])
        return x;
    else
        return parent[x] = find(parent[x]);
}
 
void unite(int x, int y) {
    x = find(x);
    y = find(y);
    parent[x] = y;
    puts("LADICA");
}
 
int main() {
    scanf("%d %d"&n, &l);
    for (i = 1; i <= l; ++i) {
        parent[i] = i;
    }
    for (i = 0; i < n; ++i) {
        scanf("%d %d"&a, &b);
        if (!taken[a]) {
            taken[a] = true;
            unite(a, b);
        }
        else if (!taken[b]) {
            taken[b] = true;
            unite(b, a);
        }
        else if (!taken[find(a)]) {
            taken[find(a)] = true;
            unite(a, b);
        }
        else if (!taken[find(b)]) {
            taken[find(b)] = true;
            unite(b, a);
        }
        else
            puts("SMECE");
    }
    return 0;
}
cs


'알고리즘 > Disjoint-set 디스조인트-셋' 카테고리의 다른 글

백준) 3780 네트워크 연결  (0) 2017.07.30
백준) 10775 공항  (0) 2017.07.29
백준) 4195 친구 네트워크  (0) 2017.07.28
백준) 1976 여행 가자  (0) 2017.07.28
백준) 1717 집합의 표현  (0) 2017.07.28

+ Recent posts