글 읽기 - (정답주의) 어떠부분에서 속도가 느린지 모르겠습니다 [파이썬]

audxo112 1년 전 0

맞추신 분들 보니 속도가 빠른 분들은 148 까지 있으시더라구요
근데 전 1692 가 나오는데
코드를 비교를 해봐도 어느 부분 때문에 느린 건지 모르겠습니다

혹시 제가 잘못 사용하는 부분이 있을까요

import sys


def rotate(sticker):
    for _ in range(4):
        yield sticker
        sticker = list(zip(*sticker[::-1]))


def attach(board, sticker, sx, sy):
    SN, SM = len(sticker), len(sticker[0])
    pos = [(x, y) for y in range(SN) for x in range(SM) if sticker[y][x] == 1]
    for x, y in pos:
        if board[y + sy][x + sx] == 1:
            return False
    for x, y in pos:
        board[y + sy][x + sx] = 1
    return True


def putting(N, M, board, sticker):
    SN, SM = len(sticker), len(sticker[0])
    for y in range(N - SN + 1):
        for x in range(M - SM + 1):
            if attach(board, sticker, x, y):
                return True
    return False


def solution(N, M, stickers):
    board = [[0] * M for _ in range(N)]

    for stk in stickers:
        for sticker in rotate(stk):
            if putting(N, M, board, sticker):
                break
    return sum(sum(board, []))


IN, IM, IK = map(int, sys.stdin.readline().split())
IStickers = []
for _ in range(IK):
    ISN, ISM = map(int, sys.stdin.readline().split())
    ISticker = [list(map(int, sys.stdin.readline().split())) for _ in range(ISN)]
    IStickers.append(ISticker)
print(solution(IN, IM, IStickers))

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

(정답주의) 어떠부분에서 속도가 느린지 모르겠습니다 [파이썬]

audxo112 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말