[Algorithm] 기초 문제 / 점수 계산

▶ 문제

OX 문제는 맞거나 틀린 두 경우의 답을 가지는 문제를 말한다. 여러 개의 OX 문제로 만들어진 시험에서 연속적으로 답을 맞히는 경우에는 가산점을 주기 위해서 다음과 같이 점수 계산을 하기 로 하였다. 1번 문제가 맞는 경우에는 1점으로 계산한다. 앞의 문제에 대해서는 답을 틀리다가 답이 맞는 처음 문제는 1점으로 계산한다. 또한, 연속으로 문제의 답이 맞는 경우에서 두 번째 문제는 2점, 세 번째 문제는 3점, ..., K번째 문제는 K점으로 계산한다. 틀린 문제는 0점으로 계 산한다. 예를 들어, 아래와 같이 10 개의 OX 문제에서 답이 맞은 문제의 경우에는 1로 표시하고, 틀린 경 우에는 0으로 표시하였을 때, 점수 계산은 아래 표와 같이 계산되어, 총 점수는 1+1+2+3+1+2=10 점이다.

 

1 0 1 1 1 0 0 1 1 0

채점	1	0	1	1	1	0	0	1	1	0
점수	1	0	1	2	3	0	0	1	2	0

 

시험문제의 채점 결과가 주어졌을 때, 총 점수를 계산하는 프로그램을 작성하시오.

 

▶ 입력 설명

첫째 줄에 문제의 개수 N (1 ≤ N ≤ 100)이 주어진다. 둘째 줄에는 N개 문제의 채점 결과를 나 타내는 0 혹은 1이 빈 칸을 사이에 두고 주어진다. 0은 문제의 답이 틀린 경우이고, 1은 문제의 답이 맞는 경우이다.

 

▶ 출력 설명

첫째 줄에 입력에서 주어진 채점 결과에 대하여 가산점을 고려한 총 점수를 출력한다.

 

▶ 입력

10

1 0 1 1 1 0 0 1 1 0

 

▶ 출력

10

 

---

▷ 내가 짠 코드

import sys
sys.stdin=open("input.txt", "r")
n=int(input())
a = list(map(int, input().split()))

total= 0
cnt = 0
for i in a:
    if i == 1:
        cnt += 1
    else:
        cnt = 0
    total += cnt
print(total)

내가 짜놓고도  '오? 이게 되네?!!'  했던 코드

 

입력값으로 for 문을 돌면서 1이 연속적으로 등장하는 만큼 카운트 수를 올리고, 0을 만나면 카운트를 리셋한다.

이 카운트를 점수에다가  더하면 결과값이 나온다. 

 

 

▷ 예시 코드

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)

예시 코드도 내가 짠 것과 같은 코드이다.

 

다만 나는 입력값 하나하나를 for 문으로 돌렸다면, 여기서는 입력값 list의 index를 사용해 for 문을 돌렸다.