15925(욱제는 정치쟁이야!!) - 해결

욱제는 정치쟁이야!!

 

문제

1학년 1반의 회장인 욱제는 오늘도 위기에 직면했다. 

아이들이 컴퓨터실 사용후 컴퓨터를 중구난방으로 켜놓고 퇴실 해버린 것이다! 

전교회장 선거에 출마할 생각인 욱제는 여론 관리를 위해 다음 컴퓨터실 시간표를 

고려해서 컴퓨터를 모두 켜거나 꺼두려고 한다. 

다음 시간에 컴퓨터실을 사용하는 반이 있다면 컴퓨터를 모두 켜두고, 그렇지 않다면 컴퓨터를 모두 꺼둘 것이다.

 

컴퓨터실은 N*N개의 학생용 컴퓨터가 정사각형 모양으로 격자에 맞춰 배열되어 있다. N은 항상 홀수이다.

 

욱제는 N*N 격자 밖의 교사용 컴퓨터로 전원을 제어하려고 한다. 

교사용 컴퓨터에 설치된 제어 프로그램은 조금 독특하다(왜인지는 교육청에 문의하는 것이 좋겠다). 

이 프로그램은 격자 상에서 사용자가 고른 어떤 한 줄(가로, 세로 중 하나)의 컴퓨터 상태를 모두 체크한다. 

이때, 그 줄에 켜진 컴퓨터가 꺼진 컴퓨터보다 더 많다면 그 줄의 컴퓨터를 모두 켜고, 반대라면 그 줄의 컴퓨터를 모두 끈다. 

이렇게 뒤집는 작업을 횟수 제한 없이 반복할 수 있다.

 

현재 컴퓨터실의 컴퓨터 전원 상태와, 다음 시간에 컴퓨터실을 사용하는 반이 있는지 여부가 주어진다! 

욱제는 컴퓨터들을 다음 시간표에 맞게 모두 켜거나 끌 수 있을까?

 

입력

첫째 줄에 각 줄의 컴퓨터 개수 N과 이후의 컴퓨터실 사용 여부가 하나의 공백을 사이에 두고 주어진다. 

사용 여부는 사용시 1, 미사용시 0으로 주어진다. (1 ≤ N ≤ 31, N % 2 == 1)

이후 둘째 줄부터 N*N의 컴퓨터실의 전원 상태가 주어진다. (i, j)의 컴퓨터가 켜져있으면 1, 

꺼져있으면 0으로 표현되며 한 줄의 숫자들은 하나의 공백으로 구분된다.

 

출력

조건에 맞게 전원을 조작할 수 있다면 1을, 그렇지 않다면 0을 출력한다.

---

처음에는 문제이해하는게 조금 어려웠는데 천천히 다시 읽어보니 그렇게 어렵지 않았다.

행과 열을 각각 확인하면서 바꿔주기만 하면 될거 같았다.

그리고 횟수는 2번 정도 행과 열을 돌면 못하는 경우 빼고는 다 할 수 있다고 생각을 하였다.

그래서 총 2번을 돌면서 바꿀수 있는 부분은 바꿔주는 식의 방법을 활용하였다.

import sys

N, OO = map(int,sys.stdin.readline().split())

total = [list(map(int,sys.stdin.readline().split())) for _ in range(N)]

#print(N,OO,total)

# 총 2번을 확인하며 바꾸기
for _ in range(2):
    # 행 확인
    for i in range(N):
        if total[i].count(OO) >= N//2 +1:
            total[i] = [OO] * N

    # 열 확인
    for j in range(N):
        cnt = 0
        # 갯수 확인
        for i in range(N):
            if total[i][j] == OO:
                cnt +=1
        #print(cnt)
        # 바꿔주기
        if cnt >= N//2 +1:
            for i in range(N):
                total[i][j] = OO

ck = 1
# 모든 원소가 OO로 바뀌지 않으면 못하는 것이기 때문
for i in range(N):
    if (1 - OO) in total[i]:
        ck = 0
        break

print(ck)