jm_p_op

알고리즘 .py 본문

수학/알고리즘

알고리즘 .py

jm_p_op 2023. 5. 4. 22:42

1.패킹을 활용하여 input값 분리하기

2.split과 len을 활용하여 0의 길이 구하기

# 인풋이 OX니까  .split("X")한 뒤,
# .count("O")로 O의 연속된 갯수만큼 뽑아 등차수열의 합
input_value = '''5
OOXXOXXOOO
OOXXOOXXOO
OXOXOXOXOXOXOX
OOOOOOOOOO
OOOOXOOOOXOOOOX'''
num, *input_list = input_value.split('\n')
for ox in input_list:
    ox_split_list = ox.split('X')
    sum_o = 0
    for ox_split in ox_split_list:
        len_o = len(ox_split)
        sum_o += len_o*(len_o+1)/2
    print(int(sum_o))
a = []
a.append(1)