• 카테고리

    질문 & 답변

  • 세부 분야

    알고리즘 · 자료구조

  • 해결 여부

    미해결

자릿수의 합

21.09.19 16:57 작성 조회수 94

0

안녕하세요. 10.점수계산 질문입니다.
강의 잘 듣고 있습니다. 감사합니다.
듣던 중 잘 이해가 가지 않아서 질문 드립니다.
 
하단이 선생님의 코드인데,
import sys
sys.stdin=open("input.txt", "r")
n=int(input())
a=list(map(int, input().split()))
cnt=0
sum=0
for i in range(n):
    if a[i]==1:
        cnt=cnt+1
        sum=sum+cnt
    else:
        cnt=0

print(sum)
a[i] == 1일때,
cnt+=1
sum+=cnt를 주면
예를 들어, n=5이고, a = [1 0 1 1 1]라고 가정했을 때

a[0] == 1,
cnt = 1, sum =1
a[0] == 0,
cnt = 0, sum =1
a[0] == 1,
cnt = 2, sum =3
a[0] == 1,
cnt = 3, sum =6
a[0] == 1,
cnt = 4, sum =10
으로 계산 되지 않는건가요?
제가 어느 부분을 잘못짚고 있는지 잘 모르겠습니다.
제 코드는

n = int(input())
a = list(map(int, input().split()))
count = 0
sum = 0
for i in a:
    if i == 1:
        count += 1
        if a[i+1] == 1:
            sum += count
    else:
        count = 0
print(sum)

이건데 어느 부분이 문제인지 모르겠어요!
답변 부탁드립니다!

 

 

course-thumbnail

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

6. 자릿수의 합

강의실 바로가기

답변 1

답변을 작성해보세요.

0

김태원님의 프로필

김태원

지식공유자

2021.09.21

안녕하세요^^

영상의 코드는 a리스트의 값이 1연속이면 cnt값이 1씩 증가하며 누적되지만 a리스트값이 0이 되면 cnt 값을 0으로 초기화합니다. 

for i in range(n):
    if a[i]==1:
        cnt=cnt+1
        sum=sum+cnt
    else:
        cnt=0 //a리스트의 값이 0일때 cnt값을 다시 0으로 초기화

그리고 님의 코드에서 

if a[i+1] == 1:

이 조건을 할 이유가 없습니다.

이 글과 비슷한 Q&A