강의

멘토링

커뮤니티

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

메론사탕님의 프로필 이미지
메론사탕

작성한 질문수

파이썬 알고리즘 문제풀이 입문(코딩테스트 대비)

5. 수의 합

혹시 제가 올린 풀이도 답이될까요?

작성

·

249

0

맥북이라 채점이안되서 이렇게 질문글올립니다

문제 테스트케이스의 경우 답이 맞긴한데 이게 확실히 답이되는지 잘모르겠습니다.

http://colorscripter.com/s/w6VKL4t
만약안된다면 혹시 어떤케이스에서 안되고 어떻게 해결해야할까요?

a,k=map(int, input().split())

li = list(map(int, input().split()))


m=0
n=0
cnt=0
# n == a 일때 까지 반복
while n<=a:
    if sum(li[m:n+1]) == k:
        if li[m] == li[n]:
            cnt+=1
            n+=1
        else:
            cnt+=1
            m+=1
        if m > n:
            break
    elif sum(li[m:n+1]) < k:
        n+=1
    elif sum(li[m:n+1]) > k:
        m+=1
print(cnt)

답변 1

0

김태원님의 프로필 이미지
김태원
지식공유자

안녕하세요^^

5 3

1 1 1 1 1

넣으면 index out of range 에러가 납니다. 위 입력으로 디버그해보세요.

메론사탕님의 프로필 이미지
메론사탕

작성한 질문수

질문하기