BOJ/백준-2506-python

문제

풀이

• 연속되면 가중치를 주면 해결되는 문제이다.
• if 조건문으로 이전 인덱스와 현재 인덱스가 같을 때, 가중치를 더해주면 된다.

코드

# https://www.acmicpc.net/problem/2506
# boj, 2506: 점수계산, python3
import sys

input = sys.stdin.readline

def solve(scores: list) -> int:
    sequence, result = 0, scores[0]

    for x in range(1, len(scores)):
        if scores[x] == 1 and scores[x-1]:
            sequence += 1
        if scores[x] == 1:
            result += scores[x] + sequence
        else:
            sequence = 0

    return result

if __name__ == '__main__':
    n = int(input())
    scores = list(map(int, input().split()))

    print(solve(scores))

결과

출처 & 깃허브

boj 2506
github