[SWEA] 1289 : 원재의 메모리 복구하기 - Python

문제

원재가 컴퓨터를 만지다가 실수를 저지르고 말았다. 메모리가 초기화된 것이다.

다행히 원래 메모리가 무슨 값이었는지 알고 있었던 원재는 바로 원래 값으로 되돌리려고 했으나 메모리 값을 바꿀 때 또 문제가 생겼다.

메모리 bit중 하나를 골라 0인지 1인지 결정하면 해당 값이 메모리의 끝까지 덮어씌우는 것이다.

예를 들어 지금 메모리 값이 0100이고, 3번째 bit를 골라 1로 설정하면 0111이 된다.

원래 상태가 주어질 때 초기화 상태 (모든 bit가 0) 에서 원래 상태로 돌아가는데 최소 몇 번이나 고쳐야 하는지 계산해보자.

[입력]

첫 번째 줄에 테스트 케이스의 수 T가 주어진다.

각 테스트 케이스는 한 줄로 이루어져 있으며, 메모리의 원래 값이 주어진다.

메모리의 길이는 1이상 50이하이다.

[출력]

각 테스트 케이스마다 ‘#x’(x는 테스트케이스 번호를 의미하며 1부터 시작한다)를 출력하고,

초기값(모든bit가 0)에서 원래 값으로 복구하기 위한 최소 수정 횟수를 출력한다.

T = int(input())
for t in range(T):
    data = list(map(int, input()))
    # 오른쪽에서부터 카운트
    cnt = 0
    n = len(data)
    idx = n-1
    while True:
        x = data[idx]
        for i in range(idx, -1, -1):
            if data[idx] != data[i]:
                idx = i
                break
        else:
            if x == 1:
                cnt += 1
                break
            else:
                break
        cnt += 1
        if x == 0:
            for i in range(idx+1, n):
                data[i] = 1
        else:
            for i in range(idx+1, n):
                data[i] = 0
    print(f"#{t+1} {cnt}")

코멘트

다른 사람들 코드를 보니 굳이 뒤에서부터 비교를 할 필요가 없었다.
앞에서부터 읽으면서 최소 몇번 바꾸면 입력받은 메모리가 되는지를 구하는 문제이다.
따라서 변경할 배열을 만들어 입력받은 메모리를 읽으면서 값이 다르다면 해당 위치를 바꾸고 해당 위치부터 끝까지 같은 값으로 변경하면 돼

새로운 코드

import sys
sys.stdin = open("input.txt", "rt")

T = int(input())
for t in range(1,T+1):
    data = list(input())
    cnt = 0
    tmp = ['0'] * len(data)

    for i in range(len(data)):
        if data[i] != tmp[i]:
            for j in range(i,len(data)):
                tmp[i] = data[i]
            cnt += 1
    print(f"#{t} {cnt}")