[파이썬 알고리즘 문제풀이 입문] 코드 구현력 기르기 - 점수계산

*문제 본문은 강의 내용과 관련되어 있어 자세하게 적지 않았습니다.

 

 

 

문제

OX 문제는 맞거나 틀린 두 경우의 답을 가지는 문제를 말한다. 여러 개의 OX 문제로 만들어진 시험에서 연속적으로 답을 맞히는 경우에는 가산점을 주기 위해서 다음과 같이 점수 계산을 하기로 했다. 1번 문제가 맞는 경우에는 1점으로 계산한다. 앞의 문제에 대해서는 답을 틀리다가 답이 맞는 처음 문제는 1점으로 계산한다. 또한, 연속으로 문제의 답이 맞는 경우에서 두 번째 문제는 2점, 세 번째 문제는 3점, ..., K 번째 문제는 K점으로 계산한다. 틀린 문제는 0점으로 계산한다.
시험 문제의 채점 결과가 주어졌을 때, 총 점수를 계산하는 프로그램을 작성하시오.

 

 

 

 

풀이

• 내 풀이 (답안이랑 동일했음)

import sys
input = sys.stdin.readline

N = int(input())
grades = list(map(int, input().split()))
count = 0
result = 0

for i in grades:
    if i == 1:
        count += 1
        result += count
    else:
        count = 0

print(result)

 

• 답안 : Kotlin (내가 바꿔본 것 / 정확X)

import java.io.*
import java.util.*

fun main(args: Array<String>) = with(BufferedReader(InputStreamReader(System.`in`))){
    var N = readLine().toInt()
    var grades = readLine().split(" ").map{ it.toInt() }
    var count = 0
    var result = 0
    
    for(i in grades) {
        if(i == 1) {
            count += 1
            result += count
        } else {
            count = 0
        }
    }

    println(result)
}