tenid's dev

import sys
input = sys.stdin.readline
commands = int(input())


class Stack:
    def __init__(self):
        self.items = []

    def push(self, item):
        self.items.append(item)

    def pop(self):
        if not self.is_empty():
            return self.items.pop()
        return -1

    def peek(self):
        if not self.is_empty():
            return self.items[-1]
        return -1

    def is_empty(self):
        return len(self.items) == 0

    def size(self):
        return len(self.items)


s = Stack()

for _ in range(commands):
    command = input()
    if command[0] == "1":
        num = int(command[2:])
        s.push(num)
    elif command[0] == "2":
        print(s.pop())
    elif command[0] == "3":
        print(s.size())
    elif command[0] == "4":
        print(int(s.is_empty()))
    elif command[0] == "5":
        print(s.peek())

정리

가독성을 위해 class를 사용하여 스택을 구현한다.

전체 시간복잡도는 command(명령어) 만큼 실행하므로 O(N)

출처 : https://www.acmicpc.net/problem/28278

1. 문제 설명

휴대폰의 자판은 컴퓨터 키보드 자판과는 다르게 하나의 키에 여러 개의 문자가 할당될 수 있습니다. 키 하나에 여러 문자가 할당된 경우, 동일한 키를 연속해서 빠르게 누르면 할당된 순서대로 문자가 바뀝니다.

예를 들어, 1번 키에 "A", "B", "C" 순서대로 문자가 할당되어 있다면 1번 키를 한 번 누르면 "A", 두 번 누르면 "B", 세 번 누르면 "C"가 되는 식입니다.

같은 규칙을 적용해 아무렇게나 만든 휴대폰 자판이 있습니다. 이 휴대폰 자판은 키의 개수가 1개부터 최대 100개까지 있을 수 있으며, 특정 키를 눌렀을 때 입력되는 문자들도 무작위로 배열되어 있습니다. 또, 같은 문자가 자판 전체에 여러 번 할당된 경우도 있고, 키 하나에 같은 문자가 여러 번 할당된 경우도 있습니다. 심지어 아예 할당되지 않은 경우도 있습니다. 따라서 몇몇 문자열은 작성할 수 없을 수도 있습니다.

이 휴대폰 자판을 이용해 특정 문자열을 작성할 때, 키를 최소 몇 번 눌러야 그 문자열을 작성할 수 있는지 알아보고자 합니다.

1번 키부터 차례대로 할당된 문자들이 순서대로 담긴 문자열배열 keymap과 입력하려는 문자열들이 담긴 문자열 배열 targets가 주어질 때, 각 문자열을 작성하기 위해 키를 최소 몇 번씩 눌러야 하는지 순서대로 배열에 담아 return 하는 solution 함수를 완성해 주세요.

단, 목표 문자열을 작성할 수 없을 때는 -1을 저장합니다.

- 제한 사항

• 1 ≤ keymap의 길이 ≤ 100
  • 1 ≤ keymap의 원소의 길이 ≤ 100
  • keymap[i]는 i + 1번 키를 눌렀을 때 순서대로 바뀌는 문자를 의미합니다.
    • 예를 들어 keymap[0] = "ABACD" 인 경우 1번 키를 한 번 누르면 A, 두 번 누르면 B, 세 번 누르면 A 가 됩니다.
  • keymap의 원소의 길이는 서로 다를 수 있습니다.
  • keymap의 원소는 알파벳 대문자로만 이루어져 있습니다.
• 1 ≤ targets의 길이 ≤ 100
  • 1 ≤ targets의 원소의 길이 ≤ 100
  • targets의 원소는 알파벳 대문자로만 이루어져 있습니다.

2. 풀이 코드

#include <string>
#include <vector>
#include <map>
using namespace std;

vector<int> solution(vector<string> keymap, vector<string> targets) {
    vector<int> answer;

    map<char, int> keyNumber;


    for (string keyInfo : keymap)
    {
        for (int i = 0; i < keyInfo.length(); i++)
        {
            if (keyNumber.find(keyInfo[i]) == keyNumber.end())
                keyNumber[keyInfo[i]] = i + 1;
            else if(keyNumber[keyInfo[i]] > i)
                keyNumber[keyInfo[i]] = i+1;

        }

    }


    for (string target : targets)
    {
        int count = 0;
        for (int i = 0; i < target.length(); i++)
        {
            if (keyNumber.find(target[i]) == keyNumber.end())
            {
                count = -1;
                break;
            }
            else       
                count += keyNumber[target[i]];


        }
        answer.push_back(count);
    }



    return answer;
}

3. 정리

K:  keymap 문자열 수
L₁ : 각 keymap  문자열의 길이
T  : target 문자열 수
L₂ : 각 target 문자열의 길이

1. map에 각 문자별로 가장 작은 횟수로 도달할 수 잇는 횟수는 저장한다.

2. targets의 배열의 문자열을 순회하면서 1번에서 저장한 도달할 수 있는 최소 횟수를 더한다. 

전체 시간 복잡도는 문자열 길이만큼 순회하기 때문에 O(K × L₁ + T × L₂)이다.

출처 :https://school.programmers.co.kr/learn/courses/30/lessons/160586