[C/Python] 백준 10845번 - 큐

문제

정수를 저장하는 큐를 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.
명령은 총 여섯 가지이다.

• push X: 정수 X를 큐에 넣는 연산이다.
• pop: 큐에서 가장 앞에 있는 정수를 빼고, 그 수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• size: 큐에 들어있는 정수의 개수를 출력한다.
• empty: 큐가 비어있으면 1, 아니면 0을 출력한다.
• front: 큐의 가장 앞에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• back: 큐의 가장 뒤에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.

입력

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.

출력

출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.

발상

이전에 풀었던 10828번 - 스택과 비슷하다!
C언어에서는 단일 연결 리스트(singly Linkedlist)를 써야지 하하

Clang

#include <stdio.h>
#include <string.h>
#include <stdlib.h>

struct queue {
    int data;
    struct queue* next;
};

struct queue* tail = NULL;
struct queue* head = NULL;
void push(int data) {
    struct queue* newnode = (struct queue*)malloc(sizeof(struct queue));
    newnode->data=data;
    newnode->next=NULL;

    if (tail == NULL) {
        tail = newnode;
        head = newnode;
    }
    else {
        tail->next = newnode;
        tail = newnode;
    }
}
void pop() {
    if (head != NULL) {
        struct queue* tmp = head;
        printf("%d\n", head->data);
        head = head->next;
        if (head == NULL) {
            tail = NULL;
        }
        free(tmp);
    }
    else {
        printf("-1\n");
    }
}
void size() {
    int size=0;
    struct queue* tmp = head;
    while (tmp != NULL) {
        size++;
        tmp = tmp->next;
    }
    printf("%d\n", size);
}
void empty() {
    if (tail!=NULL)
        printf("0\n");
    else
        printf("1\n");
}
void front() {
    if (tail!=NULL)
        printf("%d\n", head->data);
    else
        printf("-1\n");
}
void back() {
    if (tail!=NULL)
        printf("%d\n", tail->data);
    else
        printf("-1\n");
}
int input() {
    char input[6];
    int n;
    scanf("%s", input);

    if (strcmp(input,"push")==0) {
        scanf("%d", &n);
        push(n);
    }
    if (strcmp(input,"pop")==0)
        pop();
    if (strcmp(input,"size")==0)
        size();
    if (strcmp(input,"empty")==0)
        empty();
    if (strcmp(input,"front")==0)
        front();
    if (strcmp(input,"back")==0)
        back();
}
int main() {
    int n;
    scanf("%d", &n);

    for (int i=0; i<n; i++)
        input();

    struct queue* current = head;
    while (current != NULL) {
        struct queue* tmp = current;
        current = current->next;
        free(tmp);
    }
    return 0;
}

Python

import sys

n = int(sys.stdin.readline())
stack = []
for i in range(n):
    Sinput = sys.stdin.readline().split()
    if Sinput[0]== 'push':
        stack.append(Sinput[1])
    elif Sinput[0]== 'pop':
        try:
            print(stack[-1])
            stack.pop(-1)
        except:
            print(-1)
    elif Sinput[0]== 'size':
        print(len(stack))
    elif Sinput[0]== 'empty':
        if len(stack)==0:
            print(1)
        else:
            print(0)
    elif Sinput[0]=='top':
        try:
            print(stack[-1])
        except:
            print(-1)