inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

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

6. 자릿수의 합

자릿수의 합

159

pati

작성한 질문수 3

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)

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

 

 

python 코테 준비 같이 해요!

답변 1

0

김태원

안녕하세요^^

영상의 코드는 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:

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

기존에 윈도우 10으로 잘 써왔는데 윈도우 11로 바꾸고 나서 채점이 안됩니다.

1

104

2

스택에서 ')'을 만나는 경우

0

108

3

문제가 어디있나요?

0

86

2

변수 or 함수명

0

76

1

침몰하는 타이타닉 문제 질문입니다

0

71

1

AA.py 책점 에러

0

63

1

오늘 구매했는데 파이썬 자료구조 궁금한거 있으면 답변이 잘 될까요.

0

115

2

5.동전분배하기 문제 밑에코드도 정답이될까요?

0

115

1

아나그램 비교 코드

0

123

2

AA.PY파일 복사 후 채점 진행할때 오류 발생합니다.

0

163

2

문제 링크가있나여?

0

153

2

채점기 Time Limit Exceeded 오류 문의

1

177

2

동적계획법은 사용하는 문제

0

132

2

제 코드 좀 봐주세요

0

154

1

예외가 존재할 가능성?

0

100

1

3번이 안풀립니다

0

98

0

5번 틀림

0

124

0

오류원인?

0

104

0

리스트 선언

0

115

1

침몰하는 타이타닉(그리디) 문제 질문

0

114

1

알고리즘

0

72

1

코딩테스트

0

98

1

DFS 순서 질문드립니다.

0

134

2

left, right를 사용한 풀이법에 대한 질문입니다

0

94

1