20200307 알고리즘

1. 후위식 연산

후위연산식이 주어지면 연산한 결과를 출력하는 프로그램을 작성하세요. 만약 3(5+2)-9 을 후위연산식으로 표현하면 352+9- 로 표현되며 그 결과는 21입니다.

▣ 입력설명 첫 줄에 후위연산식이 주어집니다. 연산식의 길이는 50을 넘지 않습니다. 식은 1~9의 숫자와 +, -, *, /, (, ) 연산자로만 이루어진다.

▣ 출력설명 연산한 결과를 출력합니다.

▣ 입력예제 1

352+*9-

▣ 출력예제 1

12

Solution

import sys
sys.stdin = open("input35.txt", "rt")

expression = input()
stack = list()
for a in expression:
    if a.isdecimal():
        stack.append(int(a))
    else:
        first = stack.pop()
        second = stack.pop()
        if a == '+':
            stack.append(second + first)
        elif a == '-':
            stack.append(second - first)
        elif a == '*':
            stack.append(second * first)
        else:
            stack.append(second / first)

print(stack[0])

2. 공주 구하기

정보 왕국의 이웃 나라 외동딸 공주가 숲속의 괴물에게 잡혀갔습니다. 정보 왕국에는 왕자가 N명이 있는데 서로 공주를 구하러 가겠다고 합니다. 정보왕국의 왕은 다음과 같은 방법으로 공주를 구하러 갈 왕자를 결정하기로 했습니다. 왕은 왕자들을 나이 순으로 1번부터 N번까지 차례로 번호를 매긴다. 그리고 1번 왕자부터 N 번 왕자까지 순서대로 시계 방향으로 돌아가며 동그랗게 앉게 한다. 그리고 1번 왕자부터 시 계방향으로 돌아가며 1부터 시작하여 번호를 외치게 한다. 한 왕자가 K(특정숫자)를 외치면 그 왕자는 공주를 구하러 가는데서 제외되고 원 밖으로 나오게 된다. 그리고 다음 왕자부터 다시 1부터 시작하여 번호를 외친다. 이렇게 해서 마지막까지 남은 왕자가 공주를 구하러 갈 수 있다.

image

예를 들어 총 8명의 왕자가 있고, 3을 외친 왕자가 제외된다고 하자. 처음에는 3번 왕자가 3 을 외쳐 제외된다. 이어 6, 1, 5, 2, 8, 4번 왕자가 차례대로 제외되고 마지막까지 남게 된 7 번 왕자에게 공주를 구하러갑니다. N과 K가 주어질 때 공주를 구하러 갈 왕자의 번호를 출력하는 프로그램을 작성하시오.

▣ 입력설명 첫 줄에 자연수 N(5<=N<=1,000)과 K(2<=K<=9)가 주어진다.

▣ 출력설명 첫 줄에 마지막 남은 왕자의 번호를 출력합니다.

▣ 입력예제 1

8 3

▣ 출력예제 1

7

Solution

import sys
from collections import deque
sys.stdin = open("input36.txt", "rt")

n, m = map(int, input().split())
people = deque(list(range(1,n+1)))

for _ in range(n-1):
    for _ in range(m-1):
        people.append(people.popleft())
    people.popleft()

print(people[0])

3. 응급실

메디컬 병원 응급실에는 의사가 한 명밖에 없습니다. 응급실은 환자가 도착한 순서대로 진료를 합니다. 하지만 위험도가 높은 환자는 빨리 응급조 치를 의사가 해야 합니다. 이런 문제를 보완하기 위해 응급실은 다음과 같은 방법으로 환자의 진료순서를 정합니다. • 환자가 접수한 순서대로의 목록에서 제일 앞에 있는 환자목록을 꺼냅니다. • 나머지 대기 목록에서 꺼낸 환자 보다 위험도가 높은 환자가 존재하면 대기목록 제일 뒤로

다시 넣습니다. 그렇지 않으면 진료를 받습니다.

현재 N명의 환자가 대기목록에 있습니다. N명의 대기목록 순서의 환자 위험도가 주어지면, 대기목록상의 M번째 환자는 몇 번째로 진료 를 받는지 출력하는 프로그램을 작성하세요. 대기목록상의 M번째는 대기목록의 제일 처음 환자를 0번째로 간주하여 표현한 것입니다.

▣ 입력설명 첫 줄에 자연수 N(5<=N<=100)과 M(0<=M<N) 주어집니다. 두 번째 줄에 접수한 순서대로 환자의 위험도(50<=위험도<=100)가 주어집니다. 위험도는 값이 높을 수록 더 위험하다는 뜻입니다. 같은 값의 위험도가 존재할 수 있습니다.

▣ 출력설명 M번째 환자의 몇 번째로 진료받는지 출력하세요.

▣ 입력예제 1

5 2
60 50 70 80 90

▣ 출력예제 1

3

정렬 왜 사용하면 안되는지 생각해보기

Solution1

import sys
from collections import deque
sys.stdin = open("input37.txt", "rt")

n, m = map(int, input().split())
dq = deque([(idx, val) for idx, val in enumerate(list(map(int, input().split())))])

cnt = 0
while True:
    i, v = dq.popleft()
    if any(v < x[1] for x in dq):
        dq.append((i, v))
    else:
        cnt += 1
        if i == m:
            break
print(cnt)
  • any() : 함수안 인자 하나라도 True > True

Solution2

import sys
from collections import deque
sys.stdin = open("input37.txt", "rt")

m, n = map(int, input().split())
dq = deque([(idx, val) for idx, val in enumerate(list(map(int, input().split())))])

def findMax(dq):
    largest = -2147000000
    for i,v in dq:
        if v > largest:
            largest = v
    return largest

largest = findMax(dq)

cnt = 0
while True:
    i, v = dq.popleft()
    if v >= largest:
        cnt += 1
        if i == n:
            break
        largest = findMax(dq)
    else:
        dq.append((i, v))

print(cnt)

4. 교육과정 설계

현수는 1년 과정의 수업계획을 짜야 합니다. 수업중에는 필수과목이 있습니다. 이 필수과목은 반드시 이수해야 하며, 그 순서도 정해져 있 습니다. 만약 총 과목이 A, B, C, D, E, F, G가 있고, 여기서 필수과목이 CBA로 주어지면 필수과목은 C, B, A과목이며 이 순서대로 꼭 수업계획을 짜야 합니다. 여기서 순서란 B과목은 C과목을 이수한 후에 들어야 하고, A과목은 C와 B를 이수한 후에 들 어야 한다는 것입니다. 현수가 C, B, D, A, G, E로 수업계획을 짜면 제대로 된 설계이지만 C, G, E, A, D, B 순서로 짰다면 잘 못 설계된 수업계획이 됩니다. 수업계획은 그 순서대로 앞에 수업이 이수되면 다음 수업을 시작하다는 것으로 해석합니다. 수업계획서상의 각 과목은 무조건 이수된다고 가정합니다. 필수과목순서가 주어지면 현수가 짠 N개의 수업설계가 잘된 것이면 “YES”, 잘못된 것이면 ”NO“를 출력하는 프로그램을 작성하세요.

▣ 입력설명 첫 줄에 한 줄에 필수과목의 순서가 주어집니다. 모든 과목은 영문 대문자입니다. 두 번째 줄에 N(1<=N<=10)이 주어집니다. 세 번 째 줄부터 현수가 짠 N개의 수업설계가 주어집니다.(수업설계의 길이는 30이하이다) 수업설계는 같은 과목을 여러 번 이수하도록 설계해도 됩니다.

▣ 출력설명 수업설계가 잘된 것이면 “YES”, 잘못된 것이면 ”NO“를 출력합니다.

▣ 입력예제 1

CBA
3
CBDAGE 
FGCDAB
CTSBDEA

▣ 출력예제 1

#1 YES
#2 NO
#3 YES

Solution1

import sys
from collections import deque
sys.stdin = open("input38.txt", "rt")

must = input()
n = int(input())

for i in range(n):
    plan = input()
    mustDq = deque(must)
    for s in plan:
        if s in mustDq:
            if s != mustDq.popleft():
                print('#%d NO' %(i+1))
                break
    else:
        if len(mustDq) == 0:
            print('#%d YES' %(i+1))
        else:
            print('#%d NO' %(i+1))

Solution2

import sys
from collections import deque
sys.stdin = open("input38.txt", "rt")

must = input()
n = int(input())

for _ in range(n):
    plan = input()
    mustDq = deque(must)
    planDq = deque(plan)
    while planDq:
        subject = planDq.popleft()
        if any(subject == x for x in mustDq):
            mustSubject = mustDq.popleft()
            if subject != mustSubject:
                print('#NO')
                break
    else:
        if len(mustDq) != 0:
            print('#NO')
        else:
            print('#YES')
  • 포함 여부를 체크 할 때 에는 in 을 쓰자..

5. 단어 찾기

현수는 영어로 시는 쓰는 것을 좋아합니다. 현수는 시를 쓰기 전에 시에 쓰일 단어를 미리 노트에 적어둡니다. 이번에는 N개의 단어를 노트에 적었는데 시에 쓰지 않은 단어가 하나 있다고 합니다. 여러분이 찾아 주세요.

▣ 입력설명 첫 번째 줄에 자연수 N(3<=N<=100)이 주어진다. 두 번째 줄부터 노트에 미리 적어놓은 N개의 단어가 주어지고, 이어 바로 다음 줄부터 시에 쓰인 N-1개의 단어가 주어진다.

▣ 출력설명 첫 번째 줄에 시에 쓰지 않은 한 개의 단어를 출력한다.

▣ 입력예제 1

5
big
good
sky
blue
mouse
sky
good 
mouse 
big

▣ 출력예제 1

blue

Solution1

import sys
sys.stdin = open("input39.txt", "rt")

n = int(input())
p = dict()
for _ in range(n):
    word = input()
    p[word] = 1

for _ in range(n-1):
    word = input()
    p[word] = 0

for key, val in p.items():
    if val == 1:
        print(key)
        break

Solution2

import sys
from collections import Counter
sys.stdin = open("input39.txt", "rt")

n = int(input())
a1 = list()
for _ in range(n):
    a1.append(input())
a2 = list()
for _ in range(n-1):
    a2.append(input())
answer = Counter(a1) - Counter(a2)
print(list(answer.keys())[0])


Written by@[HongDongUk]
공부한 것을 소소하게 적는 블로그.

GitHubFacebook