본문 바로가기
Computer Science

프양연 평등주의 / 이진탐색, 구현

by OKOK 2018. 12. 20.

1. 어떻게 하면 되는지

2. 예제를 통해서

3. 정리를 함

4. 그리고 이분 탐색과 구현만 하면  끝 


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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
#include <iostream>
#include <stdio.h>
 
//#define MAX_N (int)1e5
//#define MAX_K (int)1e9
 
#define MAX_N 7
#define MAX_K 7
 
//HACK inline 보다 매크로를 사용하는 것이 더 빠르다
#define MIN(a,b) (a) < (b) ? (a) : (b)
#define MAX(a,b) (a) > (b) ? (a) : (b)
#define ABSDIFF(a,b) ( ( a ) > ( b ) ) ? ( ( a ) - ( b ) ) : ( ( b ) - ( a ) )
 
int answer, N, K;
 
int A[MAX_N];
int validA[MAX_N];
 
//HACK 이분탐색을 통해 최적값을 찾아갈 수 있음. 1강 1번과 같은 방법으로 푸는 문제였음
 
int isValid(int gap)
{
    validA[0= A[0];
 
    for (int i = 1; i < N; i++)
    {
        validA[i] = MIN(A[i], validA[i - 1+ gap);
    }
 
    for (int i = N - 20 <= i; i--)
    {
        validA[i] = MIN(validA[i], validA[i + 1+ gap);
    }
 
    int totalGap = 0;
 
    for (int i = 0; i < N; i++)
    {
        totalGap += A[i] - validA[i];
 
        if (totalGap > K)
            return 0;
    }
 
    return 1;
}
 
int main(int argc, char** argv)
{
    freopen("input.txt""r", stdin);
    int T;
    scanf("%d"&T);
 
    for (int case_num = 1; case_num <= T; case_num++)
    {
        answer = 0;
 
        scanf("%d %d"&N, &K);
 
        for (int i = 0; i < N; i++)
        {
            scanf("%d"&A[i]);
        }
 
        int minGap = 0;
        int maxGap = 0;
 
        //HACK maxGap을 O(N) 알고리즘을 통해서 줄일수 있는 만큼 줄인다
        for (int i = 0; i < N - 1; i++)
        {
            maxGap = MAX(maxGap, ABSDIFF(A[i], A[i + 1]));
        }
 
        while (minGap < maxGap)
        {
            int midGap = (minGap + maxGap) / 2;
 
            if (isValid(midGap))
            {
                maxGap = midGap;
            }
            else
            {
                minGap = midGap + 1;
            }
        }
 
        answer = minGap;
 
        printf("#%d %d\n", case_num, answer);
    }
 
    return 0;
}
 
cs


댓글