자바 알고리즘 문제 풀이 입문. 섹션3. Two Pointers, Sliding Window [효율성 : O(n^2) --> O(n)] : 최대 길이 연속부분수열

 

https://hyejin.tistory.com/1225

자바 알고리즘 문제 풀이 입문. 섹션3. Two Pointers, Sliding Window [효율성 : O(n^2) --> O(n)] : 연속된 자연

-> 이전 문제 풀이 

 

 

6. 최대 길이 연속부분수열

설명

0과 1로 구성된 길이가 N인 수열이 주어집니다. 여러분은 이 수열에서 최대 k번을 0을 1로 변경할 수 있습니다. 여러분이 최대 k번의 변경을 통해 이 수열에서 1로만 구성된 최대 길이의 연속부분수열을 찾는 프로그램을 작성하세요.

만약 길이가 길이가 14인 다음과 같은 수열이 주어지고 k=2라면

1 1 0 0 1 1 0 1 1 0 1 1 0 1

여러분이 만들 수 있는 1이 연속된 연속부분수열은

이며 그 길이는 8입니다.

 

입력

첫 번째 줄에 수열의 길이인 자연수 N(5<=N<100,000)이 주어집니다.

두 번째 줄에 N길이의 0과 1로 구성된 수열이 주어집니다.

 

출력

첫 줄에 최대 길이를 출력하세요.

 

예시 입력 1 

14 2
1 1 0 0 1 1 0 1 1 0 1 1 0 1

 

예시 출력 1

8

 

 

문제 풀이 1 

public int solution(int n, int k, int[] arr) {
    int answer = 0, length = 0, ch = k;

    for (int i = 0; i < n; i++) {
        if (arr[i] == 1) length = 1;
        for (int j = 1; j <= n - i-1; j++) {
            if (arr[i + j] == 1) {
                length++;
            } else {
                if (ch != 0) {
                    length++;
                    ch--;
                }else {
                    break;
                }
            }
        }
        answer = Math.max(answer, length);
        length = 0; ch = k;
    }

    return answer;
}

👩🏻‍💻 : 한결같은 나의 문제 풀이 방법이다. 이번에도 이중 for문을 사용하면서 먼저 arr[i] 가 1이면 length를 1로 설정해주고, 그 다음 

arr[i] 기준 옆으로 넘어가면서 그 값이 1 이면 length에 +1 해주고, 만약 0인데 ch의 값이 아직 0이 아니면 (0으로 바꾸는 기회를 다 쓰지 않았다면) length 를 +1 해주고 ch 의 값을 -1 해준다. 

만약 0으로 바꿀 수 있는 기회를 다 썼다면 ch == 0 이면 break 하고 arr[i] 기준 길이를 최대 값으로 비교하고 

그 다음 길이 구하기를 위해 length, ch 값을 초기화해줬다. 

 

 

 

문제 풀이 2

public int solution2(int n, int k, int[] arr) {
    int answer = 0, cnt = 0, lt = 0;

    for (int rt = 0; rt < n; rt++) {
        if (arr[rt] == 0) cnt++;

        while (cnt > k) {
            if (arr[lt] == 0) cnt--;
            lt++;
        }

        answer = Math.max(answer, rt - lt + 1);
    }

    return answer;
}

👾 : 강사님은 투 포인터 알고리즘 방식으로 이 문제를 해결했다. 

먼저 cnt 는 0으로 교체 횟수이고, lt, rt 포인터를 뒀다. 

 

먼저 rt는 주어진 배열대로 앞으로 움직이며 0을 1로 바꾸며 움직인다. arr[rt] == 0 이면 cnt +1 해준다. (0을 1로 바꿨다는 의미) 

그리고 while문을 도는데 이때 cnt 가 k 넘어가면 (0을 바꿀 수 있는 횟수를 초과하면) lt를 움직여서 이제 lt 가 다시 1로 바꾼 것을 0으로 바꿔준다. 그 다음에야 rt - lt + 1 을 해서 길이를 구할 수 있고, 최대값을 answer에 지정해주면 된다.