k가 0일 경우에

자바(Java) 알고리즘 문제풀이 입문: 코딩테스트 대비

6. 최대 길이 연속부분수열(복합적 문제)

작성

안녕하세요 선생님. 수업 정말 잘 듣고 있습니다.

이번 문제 조건에 첫번째 줄에 수열의 길이에대한 조건만 있고 k에대한 조건이 없어서 문제의 예시로 나온 수열에서 k에 0을 넣고 코드를 실행 시켜봤는데 정답 코드에서 잘못된 답을 주고 있어서 문의 드립니다. 혹시 k에대한 조건이 1 이상인가요?

아래 입력대로라면 정답은 2가 나와야할 것 같아서 문의 드립니다.

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

입력

14 0

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

public int solution(int[] input, int length, int k) {
    int lt = 0;
    int max = Integer.MIN_VALUE;
    int cnt = 0;

    for (int rt = 0; rt < length; rt++) {
        if (input[rt] == 0) cnt ++;
        while (cnt > k) {
            if (input[k] == 0) cnt--;
            lt ++;
        }
        max = Math.max(max, rt - lt + 1);
    }

    return max;
}

java 코딩-테스트

답변 2

김태원

지식공유자

안녕하세요^^

오타입니다. 아래와 같이 하세요.

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

임인혁

질문자

감사합니다 ㅠㅠ

인프런 AI 인턴

안녕하세요, 인프런 AI 인턴입니다.

임인혁님, 질문 남겨주셔서 감사합니다.

현재 주어진 코드에서 k가 0일 경우를 처리하는 로직에 문제가 있어 보입니다. k가 0일 경우, 연속된 1의 최대 길이를 찾기 위해 0을 포함하지 않은 부분을 찾아야 합니다. 하지만 코드에서는 cnt > k를 조건으로 반복문이 실행되는 곳에서 input[k]를 참조하고 있어서 잘못된 결과가 나옵니다.

예를 들어, 아래와 같이 수정해보면 좋을 것 같습니다:

public int solution(int[] input, int length, int k) {
    int lt = 0;
    int max = Integer.MIN_VALUE;
    int cnt = 0;

    for (int rt = 0; rt < length; rt++) {
        if (input[rt] == 0) cnt++;
        while (cnt > k) {
            if (input[lt] == 0) cnt--;
            lt++;
        }
        max = Math.max(max, rt - lt + 1);
    }

    return max;
}

위 코드에서 cnt > k 조건에서는 적절하게 input[lt]를 사용하는 것이 올바른 계산이 됩니다.

관련 내용으로, 이와 유사한 문제 해결 방법은 아래 링크들의 답변을 참고해 보시면 도움이 될 것입니다:
- k-1이 아니라 k-2로 해야하는거 아닌가요?
- while문 하나만 사용해서 문제를 풀었는데 길이 계산 식이 왜 이렇게 되는지 궁금합니다.

저는 질문자님의 이해를 돕고자 지식공유자님의 기존 답변을 참고하여 유사한 답변 링크를 추천해드리고 있습니다. 현재 베타 기능으로 답변이 만족스럽지 않을 수 있는 점 양해 부탁드립니다. 🙏 추가적으로 궁금한 점이 있으시면, 이어서 질문해 주세요. 곧 지식공유자께서 답변해 주실 것입니다.

인프런 커뮤니티 질문&답변

k가 0일 경우에