코딩묻은 디자이너

https://www.acmicpc.net/problem/11723

문제

비어있는 공집합 S가 주어졌을 때, 아래 연산을 수행하는 프로그램을 작성하시오.

• add x: S에 x를 추가한다. (1 ≤ x ≤ 20) S에 x가 이미 있는 경우에는 연산을 무시한다.
• remove x: S에서 x를 제거한다. (1 ≤ x ≤ 20) S에 x가 없는 경우에는 연산을 무시한다.
• check x: S에 x가 있으면 1을, 없으면 0을 출력한다. (1 ≤ x ≤ 20)
• toggle x: S에 x가 있으면 x를 제거하고, 없으면 x를 추가한다. (1 ≤ x ≤ 20)
• all: S를 {1, 2, ..., 20} 으로 바꾼다.
• empty: S를 공집합으로 바꾼다. 

입력

첫째 줄에 수행해야 하는 연산의 수 M (1 ≤ M ≤ 3,000,000)이 주어진다.

둘째 줄부터 M개의 줄에 수행해야 하는 연산이 한 줄에 하나씩 주어진다.

출력

check 연산이 주어질때마다, 결과를 출력한다.

풀이

처음에는 배열로 풀었으나 시간초과 + 메모리초과가 나와서 고민했다.

https://www.educative.io/answers/tuples-vs-list-in-python

튜플은 배열보다 메모리를 적게 사용하고, 또 이 문제에서는 내부에 x가 중복으로 속해 있지 않으니 tuple로 풀어보았다!

python 코드

# 11723 집합
import sys

S = set([])
M = int(sys.stdin.readline())

for tc in range(M):
    oper = sys.stdin.readline().split()

    if len(oper) <= 1:
        if oper[0] == 'all':
            S = set(['1', '2', '3', '4', '5', '6', '7', '8', '9', '10', '11', '12', '13', '14', '15', '16', '17', '18', '19', '20'])
        elif oper[0] == 'empty':
            S = set([])
    else:
        if oper[0] == 'add':
            S.add(oper[1])
        elif oper[0] == 'remove':
            if oper[1] in S:
                S.remove(oper[1])
        elif oper[0] == 'check':
            if oper[1] in S:
                print(1)
            else:
                print(0)
        elif oper[0] == 'toggle':
            if oper[1] in S:
                S.remove(oper[1])
            else:
                S.add(oper[1])
    # print(S)

# 11723 집합
import sys

S = set([])
M = int(sys.stdin.readline())

for tc in range(M):
    oper = sys.stdin.readline().split()

    if len(oper) <= 1:
        if oper[0] == 'all':
            S = set(['1', '2', '3', '4', '5', '6', '7', '8', '9', '10', '11', '12', '13', '14', '15', '16', '17', '18', '19', '20'])
        elif oper[0] == 'empty':
            S = set([])
    else:
        if oper[0] == 'add':
            S.add(oper[1])
        elif oper[0] == 'remove':
            if oper[1] in S:
                S.remove(oper[1])
        elif oper[0] == 'check':
            if oper[1] in S:
                print(1)
            else:
                print(0)
        elif oper[0] == 'toggle':
            if oper[1] in S:
                S.remove(oper[1])
            else:
                S.add(oper[1])
    # print(S)

https://www.acmicpc.net/problem/1012

문제

차세대 영농인 한나는 강원도 고랭지에서 유기농 배추를 재배하기로 하였다. 농약을 쓰지 않고 배추를 재배하려면 배추를 해충으로부터 보호하는 것이 중요하기 때문에, 한나는 해충 방지에 효과적인 배추흰지렁이를 구입하기로 결심한다. 이 지렁이는 배추근처에 서식하며 해충을 잡아 먹음으로써 배추를 보호한다. 특히, 어떤 배추에 배추흰지렁이가 한 마리라도 살고 있으면 이 지렁이는 인접한 다른 배추로 이동할 수 있어, 그 배추들 역시 해충으로부터 보호받을 수 있다. 한 배추의 상하좌우 네 방향에 다른 배추가 위치한 경우에 서로 인접해있는 것이다.

한나가 배추를 재배하는 땅은 고르지 못해서 배추를 군데군데 심어 놓았다. 배추들이 모여있는 곳에는 배추흰지렁이가 한 마리만 있으면 되므로 서로 인접해있는 배추들이 몇 군데에 퍼져있는지 조사하면 총 몇 마리의 지렁이가 필요한지 알 수 있다. 예를 들어 배추밭이 아래와 같이 구성되어 있으면 최소 5마리의 배추흰지렁이가 필요하다. 0은 배추가 심어져 있지 않은 땅이고, 1은 배추가 심어져 있는 땅을 나타낸다.

입력

입력의 첫 줄에는 테스트 케이스의 개수 T가 주어진다. 그 다음 줄부터 각각의 테스트 케이스에 대해 첫째 줄에는 배추를 심은 배추밭의 가로길이 M(1 ≤ M ≤ 50)과 세로길이 N(1 ≤ N ≤ 50), 그리고 배추가 심어져 있는 위치의 개수 K(1 ≤ K ≤ 2500)이 주어진다. 그 다음 K줄에는 배추의 위치 X(0 ≤ X ≤ M-1), Y(0 ≤ Y ≤ N-1)가 주어진다. 두 배추의 위치가 같은 경우는 없다.

출력

각 테스트 케이스에 대해 필요한 최소의 배추흰지렁이 마리 수를 출력한다.

풀이

백터를 사용해 풀었다

dfs로 재귀를 통해 배추가 심어진 구간을 통과하면 2로 변환시키고 더이상 배추가 없으면 재귀를 나와 cnt 를 증가시킨다

예제는 풀리는데 36%쯤에서 계속 컴파일 오류가 나서 왜 그럴까 하고 찾아보니 파이썬에서 재귀를 사용할 경우

import sys
sys.setrecursionlimit(10 ** 6)

를 꼭 넣어줘야 한다고 한다

참고 블로그

https://fuzzysound.github.io/sys-setrecursionlimit

재귀 깊이의 제한이 1000으로 한정되어있기 때문이라고...!!

python 코드

# 1012 유기농 배추
# 재귀 깊이 연장
import sys
sys.setrecursionlimit(10 ** 6)

# 상하좌우
dr = [0, 0, 1, -1]
dc = [1, -1, 0, 0]

#재귀 함수
def find(x, y):
    for k in range(4):
# 상하좌우의 배추 찾기
        nr = x + dr[k]
        nc = y + dc[k]

        if 0 <= nr < N and 0 <= nc < M:
            if arr[nr][nc] == 1:
                arr[nr][nc] = 2
                find(nr, nc)

#입력부분
T = int(input())

for tc in range(T):
    M, N, K = map(int, input().split())
    cnt = 0

    arr = [[0] * M for _ in range(N)]
# 문제에서 주어진 밭 만들기
    for kc in range(K):
        y, x = map(int, input().split())
        arr[x][y] = 1

    # print(arr)
    for j in range(M):
        for i in range(N):
            if arr[i][j] == 1:
                find(i, j)
# 카운드 증가
                cnt += 1
    print(cnt)

https://www.acmicpc.net/problem/1620

문제

안녕? 내 이름은 이다솜. 나의 꿈은 포켓몬 마스터야. 일단 포켓몬 마스터가 되기 위해선 포켓몬을 한 마리 잡아야겠지? 근처 숲으로 가야겠어.

(뚜벅 뚜벅)

얏! 꼬렛이다. 꼬렛? 귀여운데, 나의 첫 포켓몬으로 딱 어울린데? 내가 잡고 말겠어. 가라! 몬스터볼~

(펑!) 헐랭... 왜 안 잡히지?ㅜㅜ 몬스터 볼만 던지면 되는 게 아닌가...ㅜㅠ

(터벅터벅)

어? 누구지?

오박사 : 나는 태초마을의 포켓몬 박사 오민식 박사라네. 다솜아, 포켓몬을 잡을 때는, 일단 상대 포켓몬의 체력을 적당히 바닥으로 만들어놓고 몬스터 볼을 던져야 한단다. 자, 내 포켓몬 이상해꽃으로 한번 잡아보렴. 포켓몬의 기술을 쓰는 것을 보고 포켓몬을 줄지 안줄지 결정을 하겠네. 자 한번 해보아라. 다솜아.

이다솜 : 이상해꽃이라...음.. 꽃이니깐 왠지 햇빛을 받아서 공격을 할 것 같은데... 음... 이상해꽃! 햇빛공격!!!

(꼬렛이 이상해꽃에게 공격을 받아 체력이 25 감소했다.)    가라! 몬스터 볼!!!    (꼬렛을 잡았습니다.)    야호! 신난다. 꼬렛을 잡았다.

오박사 : 오우!! 방금 쓴 공격은 솔라빔이라고 하네.. 어떻게 공격을 한 건가? 솔라빔이란 공격에 대해서 공부를 한 건가?

이다솜 : 꽃이니깐 왠지 햇빛을 제대로 받으면 광합성을 해서 음.. 그냥 그럴 것 같아서요 ☞☜

오박사 : 다른 아이들은 넝쿨채찍이나, 나뭇잎 공격을 하는데, 다솜이는 역시 뭔가 다르구나. 그럼 나와 함께 연구소로 가자꾸나. 내가 포켓몬을 한 마리 줄 테니, 너의 꿈을 펼쳐보아라. 꿈은 이루어진단다.

이다솜 : 네! 오박사님, 고마워요.ㅜㅜ

오박사 : 가자. 나의 연구소는 너의 옆집의 아랫집이란다. 같이 가도록하자. 지금 포켓몬을 주마.

이다솜 : 네. 야호!!

'

오영식 : 어? 오박사님 얘는 누구인가요?

오박사 : 얘는 너의 라이벌이 될 친구 이다솜이라고 하네. 자, 포켓몬을 한 마리 골라보도록 해봐라 다솜아. 레이디퍼스트 네가 먼저 골라봐라.

이다솜 : 저는 생각해둔 포켓몬이 있어요. 피카츄 골라도 될까요?

오박사 : 그래 여기 피카츄가 한 마리 있단다. 피카츄를 가져가거라.

오영식 : 그럼 저는 이브이를 가져가겠어요. 그럼 나중에 보자 이다솜.

이다솜 : 그럼 꼬렛을 다시 잡으러 가야겠다. 영식아, 그리고 민식박사님 빠잉!

이다솜 : 피카츄 공격!

가라 몬스터 볼!

이다솜 : 야호! 신난다. 꼬렛을 잡았다!!!!!

이다솜 : 그럼! 일단 사천왕을 이기고 오겠어!

이다솜 : 여기가 사천왕과 대결하려면 가야하는 곳인가..

경비원 : 사천왕과 대결을 하려면, 마을의 체육관 리더를 이겨서 배지를 8개를 모아야 한다네... 배지를 모아서 오도록 하게

이다솜 : 잉ㅠㅜ... 그럼 배지부터 모아야 하는구나ㅠㅜㅠㅜ 나쁘당 그냥 좀 봐주지..

<1 년 후>

그동안의 줄거리 : 이다솜은 일단 상록 숲의 체육관 리더에게 도전을 했다. 하지만 상록숲 체육관의 리더는 실종된 상태. 따라서 회색마을부터 도전하기로 했다. 체육관의 리더를 이기면서, 로켓단을 해체시키기도 하고, 여러 가지 사건도 있었다. 결국 전설의 포켓몬도 잡고, 이제 사천왕을 이기려고 도전하기로 했다. 사천왕은 모두 가볍게 이기고, 이제 마지막 라이벌 오!영!식! 이다.

오영식 : 훗. 1년 전의 그 이다솜이 사천왕을 이기고 현재 포켓몬 마스터인 나에게 덤벼? 어디 한번 덤벼보시지.

이다솜 : 헐랭... 나를 우습게보네.... 한번 두고 보시지! 그럼 대결이닷!

이다솜 : 휴... 이겼다.

오영식 : 내가 지다니 분하다. ㅜㅜ

오박사 : 그럼 다솜아 이제 진정한 포켓몬 마스터가 되기 위해 도감을 완성시키도록 하여라. 일단 네가 현재 가지고 있는 포켓몬 도감에서 포켓몬의 이름을 보면 포켓몬의 번호를 말하거나, 포켓몬의 번호를 보면 포켓몬의 이름을 말하는 연습을 하도록 하여라. 나의 시험을 통과하면, 내가 새로 만든 도감을 주도록 하겠네.

입력

첫째 줄에는 도감에 수록되어 있는 포켓몬의 개수 N이랑 내가 맞춰야 하는 문제의 개수 M이 주어져. N과 M은 1보다 크거나 같고, 100,000보다 작거나 같은 자연수인데, 자연수가 뭔지는 알지? 모르면 물어봐도 괜찮아. 나는 언제든지 질문에 답해줄 준비가 되어있어.

둘째 줄부터 N개의 줄에 포켓몬의 번호가 1번인 포켓몬부터 N번에 해당하는 포켓몬까지 한 줄에 하나씩 입력으로 들어와. 포켓몬의 이름은 모두 영어로만 이루어져있고, 또, 음... 첫 글자만 대문자이고, 나머지 문자는 소문자로만 이루어져 있어. 아참! 일부 포켓몬은 마지막 문자만 대문자일 수도 있어. 포켓몬 이름의 최대 길이는 20, 최소 길이는 2야. 그 다음 줄부터 총 M개의 줄에 내가 맞춰야하는 문제가 입력으로 들어와. 문제가 알파벳으로만 들어오면 포켓몬 번호를 말해야 하고, 숫자로만 들어오면, 포켓몬 번호에 해당하는 문자를 출력해야해. 입력으로 들어오는 숫자는 반드시 1보다 크거나 같고, N보다 작거나 같고, 입력으로 들어오는 문자는 반드시 도감에 있는 포켓몬의 이름만 주어져. 그럼 화이팅!!!

출력

첫째 줄부터 차례대로 M개의 줄에 각각의 문제에 대한 답을 말해줬으면 좋겠어!!!. 입력으로 숫자가 들어왔다면 그 숫자에 해당하는 포켓몬의 이름을, 문자가 들어왔으면 그 포켓몬의 이름에 해당하는 번호를 출력하면 돼. 그럼 땡큐~

이게 오박사님이 나에게 새로 주시려고 하는 도감이야. 너무 가지고 싶다ㅠㅜ. 꼭 만점을 받아줬으면 좋겠어!! 파이팅!!!

풀이

처음에는 배열을 만들어서 배열에 포켓몬 도감을 넣고

내가 찾아야하는 포켓몬이 입력으로 들어오는 경우 isdigit()을 통해 숫자이면 해당 인덱스에 속한 포켓몬 이름을, 문자이면 해당 포켓몬의 인덱스 값을 출력하도록 하였지만, 시간 초과가 발생했다.

찾아본 결과 딕셔너리에 인덱스값과 이름을 둘 다 입력하면 더 빠른 속도로 출력을 할 수 있었다!

그리고 추가로 sys.stdin.readline() 의 경우 입력 받을때 이대로 입력하면 문자열 뒤에 \n이 같이 입력이 된다

그래서 뒤에 rstrip()을 입력해주어야 한다

시간초과 python 코드

# 1620 나는야 포켓몬 마스터 이다솜
import sys

N, M = map(int,sys.stdin.readline().split())
pocketmon = []

for n in range(N):
    pocketmon.append(sys.stdin.readline().rstrip())

#print(pocketmon)

for m in range(M):
    mon = input()
    if mon.isdigit():
        print(pocketmon[int(mon) - 1])
    else:
        print(pocketmon.index(mon) + 1)

맞은 python 코드

# 1620 나는야 포켓몬 마스터 이다솜
import sys

N, M = map(int,sys.stdin.readline().split())
pocketmon = {}

for n in range(1, N+1):
    monster = sys.stdin.readline().rstrip()
    pocketmon[n] = monster
    pocketmon[monster] = n

for m in range(M):
    mon = sys.stdin.readline().rstrip()
    if mon.isdigit():
        print(pocketmon[int(mon)])
    else:
        print(pocketmon[mon])

https://www.acmicpc.net/problem/10039

문제

상현이가 가르치는 아이폰 앱 개발 수업의 수강생은 원섭, 세희, 상근, 숭, 강수이다.

어제 이 수업의 기말고사가 있었고, 상현이는 지금 학생들의 기말고사 시험지를 채점하고 있다. 기말고사 점수가 40점 이상인 학생들은 그 점수 그대로 자신의 성적이 된다. 하지만, 40점 미만인 학생들은 보충학습을 듣는 조건을 수락하면 40점을 받게 된다. 보충학습은 거부할 수 없기 때문에, 40점 미만인 학생들은 항상 40점을 받게 된다.

학생 5명의 점수가 주어졌을 때, 평균 점수를 구하는 프로그램을 작성하시오.

입력

입력은 총 5줄로 이루어져 있고, 원섭이의 점수, 세희의 점수, 상근이의 점수, 숭이의 점수, 강수의 점수가 순서대로 주어진다.

점수는 모두 0점 이상, 100점 이하인 5의 배수이다. 따라서, 평균 점수는 항상 정수이다. 

출력

첫째 줄에 학생 5명의 평균 점수를 출력한다.

풀이

if 40보다 작으면 40이 hap에 들어가도록 했다

브론즈 문제인데 틀린 문제에 있길래 왜지...? 예에에엣날에 풀었던 문제인가? 했는데 16일 전...왜 틀렸던거지

python코드

# 10039
hap = 0
for t in range(5):
    num = int(input())
    if num > 40:
        hap += num
    else:
        hap += 40
print(hap//5)

https://www.acmicpc.net/problem/11650

문제

2차원 평면 위의 점 N개가 주어진다. 좌표를 x좌표가 증가하는 순으로, x좌표가 같으면 y좌표가 증가하는 순서로 정렬한 다음 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 점의 개수 N (1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N개의 줄에는 i번점의 위치 xi와 yi가 주어진다. (-100,000 ≤ xi, yi ≤ 100,000) 좌표는 항상 정수이고, 위치가 같은 두 점은 없다.

출력

첫째 줄부터 N개의 줄에 점을 정렬한 결과를 출력한다.

풀이

최근 프로그래머스에서 javascript 연습하다가 손풀이로 백준 python 풀기!

근데 딱히 풀건 없었다. sort 썼다.

이 문제 꽤 최근까지 문제를 이해못해서 못풀고 있었는데 지금 보니 이걸 왜 이해못했지;;;

python 코드

# 11650 좌표 정렬하기
import sys

n = int(sys.stdin.readline())
arr = []

for i in range(n):
    x, y = map(int,sys.stdin.readline().split())
    arr.append([x,y])

arr.sort()

for i in range(n):
    print(*arr[i])

https://www.acmicpc.net/problem/1676

문제

N!에서 뒤에서부터 처음 0이 아닌 숫자가 나올 때까지 0의 개수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N이 주어진다. (0 ≤ N ≤ 500)

출력

첫째 줄에 구한 0의 개수를 출력한다.

풀이

팩토리얼로 계산한 후 int를 str로 변환하여 뒤에서 부터 0의 개수를 더했다.

python 코드

# 1676 팩토리얼 0의 개수

N = int(input())
num = 1
cnt = 0

for i in range(1, N+1):
    num *= i
# print(num)
for i in range(len(str(num))-1,-1,-1):
    if str(num)[i] == "0":
        cnt += 1
    else:
        break

print(cnt)

https://www.acmicpc.net/problem/10814

문제

온라인 저지에 가입한 사람들의 나이와 이름이 가입한 순서대로 주어진다. 이때, 회원들을 나이가 증가하는 순으로, 나이가 같으면 먼저 가입한 사람이 앞에 오는 순서로 정렬하는 프로그램을 작성하시오.

입력

첫째 줄에 온라인 저지 회원의 수 N이 주어진다. (1 ≤ N ≤ 100,000)

둘째 줄부터 N개의 줄에는 각 회원의 나이와 이름이 공백으로 구분되어 주어진다. 나이는 1보다 크거나 같으며, 200보다 작거나 같은 정수이고, 이름은 알파벳 대소문자로 이루어져 있고, 길이가 100보다 작거나 같은 문자열이다. 입력은 가입한 순서로 주어진다.

출력

첫째 줄부터 총 N개의 줄에 걸쳐 온라인 저지 회원을 나이 순, 나이가 같으면 가입한 순으로 한 줄에 한 명씩 나이와 이름을 공백으로 구분해 출력한다.

풀이

sort를 사용했다.

숫자만을 기준으로 하기위해 lambda를 사용했다

python 코드

# 10814 나이순 정렬

N = int(input())
people = []

for t in range(N):
    age, name = input().split()
    people.append([int(age),name])
# 숫자만 기준으로 정렬
people.sort(key = lambda people:people[0])

for i in range(N):
    print('{} {}'.format(people[i][0],people[i][1]))

https://www.acmicpc.net/problem/9012

문제

괄호 문자열(Parenthesis String, PS)은 두 개의 괄호 기호인 ‘(’ 와 ‘)’ 만으로 구성되어 있는 문자열이다. 그 중에서 괄호의 모양이 바르게 구성된 문자열을 올바른 괄호 문자열(Valid PS, VPS)이라고 부른다. 한 쌍의 괄호 기호로 된 “( )” 문자열은 기본 VPS 이라고 부른다. 만일 x 가 VPS 라면 이것을 하나의 괄호에 넣은 새로운 문자열 “(x)”도 VPS 가 된다. 그리고 두 VPS x 와 y를 접합(concatenation)시킨 새로운 문자열 xy도 VPS 가 된다. 예를 들어 “(())()”와 “((()))” 는 VPS 이지만 “(()(”, “(())()))” , 그리고 “(()” 는 모두 VPS 가 아닌 문자열이다.

여러분은 입력으로 주어진 괄호 문자열이 VPS 인지 아닌지를 판단해서 그 결과를 YES 와 NO 로 나타내어야 한다. 

입력

입력 데이터는 표준 입력을 사용한다. 입력은 T개의 테스트 데이터로 주어진다. 입력의 첫 번째 줄에는 입력 데이터의 수를 나타내는 정수 T가 주어진다. 각 테스트 데이터의 첫째 줄에는 괄호 문자열이 한 줄에 주어진다. 하나의 괄호 문자열의 길이는 2 이상 50 이하이다. 

출력

출력은 표준 출력을 사용한다. 만일 입력 괄호 문자열이 올바른 괄호 문자열(VPS)이면 “YES”, 아니면 “NO”를 한 줄에 하나씩 차례대로 출력해야 한다.

풀이

4949 균형잡힌 세상 코드를 재활용 해서 풀어보았다.

2022.06.22 - [study/백준] - [백준] 4949. 균형잡힌 세상 : python

python코드

# 9012 괄호

T = int(input())

for i in range(T):
    test = input()
    bracket = []
    flag ='YES'
    for t in test:
        if len(bracket) == 0:
            if t == ')':
                flag = 'NO'
                break
            elif t == '(':
                bracket.append(t)
        else:
            if t == '(':
                bracket.append(t)
            if t == ')' and bracket[-1] == '(':
                bracket.pop()
            elif t == ')' and bracket[-1] != '(':
                flag = 'NO'
                break
    if len(bracket) > 0:
        flag = 'NO'
    print(flag)

https://www.acmicpc.net/problem/2960

문제

에라토스테네스의 체는 N보다 작거나 같은 모든 소수를 찾는 유명한 알고리즘이다.

이 알고리즘은 다음과 같다.

1. 2부터 N까지 모든 정수를 적는다.
2. 아직 지우지 않은 수 중 가장 작은 수를 찾는다. 이것을 P라고 하고, 이 수는 소수이다.
3. P를 지우고, 아직 지우지 않은 P의 배수를 크기 순서대로 지운다.
4. 아직 모든 수를 지우지 않았다면, 다시 2번 단계로 간다.

N, K가 주어졌을 때, K번째 지우는 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N과 K가 주어진다. (1 ≤ K < N, max(1, K) < N ≤ 1000)

출력

첫째 줄에 K번째 지워진 수를 출력한다.

풀이

문제에 충실하게 풀었다

모든 정수를 적고, 가장 작은 수의 배수를 지웠다

C++로 먼저 풀려고 했는데 1시간 30분을 쏟았는데 안풀려서 열받아서 python으로 풀기 시작했고 11분만에 풀었다.

화가 나는 것이다.

python 코드

# 2960 에라노스테네스의 체

N, K = map(int,input().split())
arr = []
cnt = 0
ans = 0

for i in range(2, N+1):
    arr.append(i)

min_num = arr[0]

while cnt < K:
    i = 1
    while min_num * i < N + 1:
        if min_num * i in arr:
            cnt += 1
            arr.remove(min_num * i)
            if cnt == K:
                ans = min_num * i
        i += 1
        # print(arr)
    if len(arr) > 0:
        min_num = arr[0]
print(ans)

https://www.acmicpc.net/problem/4949

문제

세계는 균형이 잘 잡혀있어야 한다. 양과 음, 빛과 어둠 그리고 왼쪽 괄호와 오른쪽 괄호처럼 말이다.

정민이의 임무는 어떤 문자열이 주어졌을 때, 괄호들의 균형이 잘 맞춰져 있는지 판단하는 프로그램을 짜는 것이다.

문자열에 포함되는 괄호는 소괄호("()") 와 대괄호("[]")로 2종류이고, 문자열이 균형을 이루는 조건은 아래와 같다.

• 모든 왼쪽 소괄호("(")는 오른쪽 소괄호(")")와만 짝을 이뤄야 한다.
• 모든 왼쪽 대괄호("[")는 오른쪽 대괄호("]")와만 짝을 이뤄야 한다.
• 모든 오른쪽 괄호들은 자신과 짝을 이룰 수 있는 왼쪽 괄호가 존재한다.
• 모든 괄호들의 짝은 1:1 매칭만 가능하다. 즉, 괄호 하나가 둘 이상의 괄호와 짝지어지지 않는다.
• 짝을 이루는 두 괄호가 있을 때, 그 사이에 있는 문자열도 균형이 잡혀야 한다.

정민이를 도와 문자열이 주어졌을 때 균형잡힌 문자열인지 아닌지를 판단해보자.

입력

하나 또는 여러줄에 걸쳐서 문자열이 주어진다. 각 문자열은 영문 알파벳, 공백, 소괄호("( )") 대괄호("[ ]")등으로 이루어져 있으며, 길이는 100글자보다 작거나 같다. 각 줄은 마침표(".")로 끝난다.

출력

각 줄마다 해당 문자열이 균형을 이루고 있으면 "yes"를, 아니면 "no"를 출력한다.

풀이

스택을 사용하여 풀었다.

리스트를 만들어두고 괄호를 넣고 조건에 맞는 괄호가 들어오면 pop 해줬다.

C++로 먼저 풀어 볼 생각이었지만,

엇... 이거 C++로 입력 어떻게 받지... 멍해져서 그냥 python 으로 먼저 풀었다.

마지막에 배열에 남는 괄호가 없는지도 꼭 파악해줘야한다.

코드

# 4949 균형잡힌 세상
while True:
    t = input()
    bracket = []
    flag = 'yes'
    if t == '.':
        break
    for i in range(len(t)):
        if len(bracket) == 0:
            if t[i] == ')' or t[i] ==']':
                flag = 'no'
                break
            elif t[i] == '(' or t[i] == '[':
                bracket.append(t[i])
        else:
            if t[i] == '(':
                bracket.append(t[i])
            if t[i] == '[':
                bracket.append(t[i])
            if t[i] == ')' and bracket[-1] == '(':
                bracket.pop()
            elif t[i] == ')' and bracket[-1] != '(':
                flag = 'no'
                break
            elif t[i] == ']' and bracket[-1] == '[':
                bracket.pop()
            elif t[i] == ']' and bracket[-1] != '[':
                flag= 'no'
                break
    if len(bracket) != 0:
        flag = 'no'
    print(flag)