BOJ / 1802번 / 종이 접기 [Go][Python3]

👩🏻‍💻 문제

1802번: 종이 접기

---

✍🏻 풀이

🎨 Go

// https://www.acmicpc.net/problem/1802
package main

import (
	"bufio"
	"fmt"
	"os"
	"strings"
)

func main() {
	reader := bufio.NewReader(os.Stdin)
	writer := bufio.NewWriter(os.Stdout)
	defer writer.Flush()

	var t int
	fmt.Fscanln(reader, &t)

	for i := 0; i < t; i++ {
		input, _ := reader.ReadString('\n')
		rules := strings.Split(strings.Replace(input, "\n", "", 1), "")
		if check(rules) {
			fmt.Fprintln(writer, "YES")
		} else {
			fmt.Fprintln(writer, "NO")
		}
	}
}

func check(rules []string) bool {
	if len(rules) == 1 {
		return true
	}
	for i := 0; i < len(rules)/2; i++ {
		if rules[i] == rules[len(rules)-1-i] { // 가운데를 기준으로 접히는 부분은 서로 다른 값이어야 함
			return false
		}
	}
	return check(rules[0:len(rules)/2]) && check(rules[len(rules)/2+1:len(rules)]) // 가운데를 기준으로 왼쪽과 오른쪽을 체크
}

🎨 Python3

# https://www.acmicpc.net/problem/1802
import sys

def check(rules):
    if len(rules) == 1:
        return True
    for i in range(len(rules)//2):
        if rules[i] == rules[len(rules)-1-i]:
            return False
    return check(rules[0:len(rules)//2]) and check(rules[len(rules)//2+1:len(rules)])

if __name__ == "__main__":
    t = int(sys.stdin.readline())
    for i in range(t):
        rules = list(sys.stdin.readline().rstrip())
        if check(rules):
            print("YES")
        else:
            print("NO")