글 읽기 - 5시간동안 고민해보았지만 반례를 찾지 못했습니다

ai4youej 2년 전 0

진짜 정신 나갈거 같습니다...

코드는 다음과 같이 짰습니다.

1. Album 구조체를 만들어서 하위 폴더 셋, 자기 자신을 포함한 모든 하위 폴더의 개수, 앨범의 이름, 하위 사진 셋, 모든 하위 사진의 개수, 상위 폴더의 Album 구조체를 만들었습니다.

2. 문제를 풀 Program 구조체를 만들었습니다. top은 가장 위의 앨범으로 가기 위한 통로이고, 초기 위치는 top인 앨범 위치입니다.

그리고 각 명령어와 쿼리에 대해서 구현했습니다.

대체 반례나 코드에서 무엇이 잘못되었길래 계속 4%에서 틀리는걸까요? ㅠㅠㅠ

import sys
input = sys.stdin.readline

class Album:
    def __init__(self, name):
        self.folder = set() 
        self.folder_cnt = 1 # Itself
        self.name = name
        self.photo = set()
        self.photo_cnt = 0
        self.parent = None
    
    def __gt__(self, A):
        return self.name > A.name
        
class Program:
    def __init__(self):
        top = Album('album')
        self.locate = top
        self.top = top
    
    def mkalb(self, S):
        now = self.locate
        for folder in now.folder:
            if folder.name == S:
                print('duplicated album name')
                return None
        A = Album(S); A.parent = now
        now.folder.add(A)
        now.folder_cnt += 1
        while now.parent:
            now = now.parent
            now.folder_cnt += 1
    
    def rmalb(self, S):
        now = self.locate
        deletedFolder, deletedPhoto = 0, 0
        if S == '-1':
            if now.folder:
                to_del_album = min(now.folder)
                now.folder.remove(to_del_album)
                deletedFolder = to_del_album.folder_cnt
                deletedPhoto = to_del_album.photo_cnt
                while to_del_album.parent:
                    to_del_album = to_del_album.parent
                    to_del_album.folder_cnt -= deletedFolder
                    to_del_album.photo_cnt -= deletedPhoto
        elif S == '0':
            for folder in now.folder:
                deletedFolder += folder.folder_cnt
                deletedPhoto += folder.photo_cnt
            now.folder = set(); now.folder_cnt = 1
            while now.parent:
                now = now.parent
                now.folder_cnt -= deletedFolder
                now.photo_cnt -= deletedPhoto
        elif S == '1':
            if now.folder:
                to_del_album = max(now.folder)
                now.folder.remove(to_del_album)
                deletedFolder = to_del_album.folder_cnt
                deletedPhoto = to_del_album.photo_cnt
                while to_del_album.parent:
                    to_del_album = to_del_album.parent
                    to_del_album.folder_cnt -= deletedFolder
                    to_del_album.photo_cnt -= deletedPhoto            
        else:
            for folder in now.folder:
                if folder.name == S:
                    to_del_album = folder
                    now.folder.remove(to_del_album)
                    deletedFolder = to_del_album.folder_cnt
                    deletedPhoto = to_del_album.photo_cnt
                    while to_del_album.parent:
                        to_del_album = to_del_album.parent
                        to_del_album.folder_cnt -= deletedFolder
                        to_del_album.photo_cnt -= deletedPhoto                   
                    break        
        print(deletedFolder, deletedPhoto)
    
    def insert(self, S):
        now = self.locate
        if S in now.photo:
            print('duplicated photo name')
        else:
            now.photo.add(S)
            now.photo_cnt += 1
            while now.parent:
                now = now.parent
                now.photo_cnt += 1
    
    def delete(self, S):
        now = self.locate
        deletedPhoto = 0
        if S == '-1':
            if now.photo:
                to_del_photo = min(now.photo)
                now.photo.remove(to_del_photo)
                now.photo_cnt -= 1
                while now.parent:
                    now = now.parent
                    now.photo_cnt -= 1
                deletedPhoto = 1
        elif S == '0':
            deletedPhoto = len(now.photo)
            now.photo = set()
            now.photo_cnt -= deletedPhoto
            while now.parent:
                now = now.parent
                now.photo_cnt -= deletedPhoto
        elif S == '1':
            if now.photo:
                to_del_photo = max(now.photo)
                now.photo.remove(to_del_photo)
                now.photo_cnt -= 1
                while now.parent:
                    now = now.parent
                    now.photo_cnt -= 1
                deletedPhoto = 1
        else:
            if S in now.photo:
                now.photo.remove(S)
                now.photo_cnt -= 1
                while now.parent:
                    now = now.parent
                    now.photo_cnt -= 1
                deletedPhoto = 1
        print(deletedPhoto)
        
    def ca(self, S):
        if S == '..':
            if self.locate.parent:
                self.locate = self.locate.parent
        elif S == '/':
            self.locate = self.top
        else:
            for folder in self.locate.folder:
                if folder.name == S:
                    self.locate = folder
                    break
        print(self.locate.name)

N = int(input())
program = Program()
for _ in range(N):
    query, S = input().split()
    if query == 'mkalb':
        program.mkalb(S)
    elif query == 'rmalb':
        program.rmalb(S)
    elif query == 'insert':
        program.insert(S)
    elif query == 'delete':
        program.delete(S)
    elif query == 'ca':
        program.ca(S)

WeissBlume 2년 전 0

11
mkalb deptha
ca deptha
mkalb depthb
ca depthb
mkalb depthc
ca depthc
insert photoc
ca ..
rmalb 0
ca ..
rmalb 0

ai4youej 2년 전 0

반례 정말정말 감사합니다 ㅠㅠ

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

문제

문제

출처

ICPC

5시간동안 고민해보았지만 반례를 찾지 못했습니다

ai4youej 2년 전 0

WeissBlume 2년 전 0

ai4youej 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말