say repository

import sys
n,k = map(int,sys.stdin.readline().split())
q=[]
result = []
for i in range(1,n+1):
    q.append(i)

index = 0
while q:
    index += k-1
    if index >= len(q):
        index = index % len(q)

    result.append(q.pop(index))

print("<", ', '.join(str(i) for i in result),">",sep="")

풀이

index 는 삭제할 인덱스이고 계속 k만큼 더해주는데 배열의 인덱스이니까 k-1을 더해준다.

삭제할 인덱스가 큐 한바퀴를 돌면, 원 모양이니까 다시 인덱스가 돌아가야한다.

index = index % len(q)

이렇게 하면 돌아갈 수 있다.

마지막 출력은 join()함수를 써서 깔끔하게 작성한다.

', '.join(i for i in result)로 하면 result에 있는 i는 정수여서 str로 바꿔서 join해야한다.

이 때, "<", ">"를 쉼표로 같이 이어붙이면 공백이 발생한다.

정답은 공백을 제거해야하니까 sep = ""로 제거해준다.

import sys
n = int(sys.stdin.readline())
stack = []
op = []
now = 1
possible = 1
for i in range(n):
    num = int(sys.stdin.readline())
    while now <= num:
        stack.append(now)
        op.append('+')
        now+=1

    if stack[-1] == num:
        stack.pop()
        op.append('-')
    else:
        possible = 0
if possible ==0:
    print("NO")
else:
    for i in op:
        print(i)

풀이

push()를 1부터 n까지 오름차순으로 해야한다.

그럼 입력값이 pop() 이 가능할 때까지 1부터 스택에 append() 해준다.

입력값이 pop()가능하면 pop()하고

불가능하면 NO 출력한다.

코드1

import math
import sys
a,b = map(int,sys.stdin.readline().split())
print(math.gcd(a,b))
print(math.lcm(a,b))

풀이

python의 math라이브러리는

최대공약수 math.gcd(a,b) 와

최소공배수 math.lcm(a,b)를 제공한다.

코드2

import math
import sys
a,b = map(int,sys.stdin.readline().split())

def GCD(a,b):
    while b>0:
        a,b = b, a%b
    return a
    
def LCM(a,b):
    return int(a*b / GCD(a,b))
    
print(GCD(a,b))
print(LCM(a,b))

풀이

유클리드 호제법

최대공약수

a,b의 최대공약수는 a를 b로 나눈 나머지와 b의 최대공약수와 같다.

최소공배수는

a,b를 곱한 값을 a,b의 최대공약수로 나눈 값이 최소공배수이다.

외워잉

import sys
import math
n, m = map(int,sys.stdin.readline().split())
arr = [True for i in range(m+1)]
for i in range(2,int(math.sqrt(m))+1):
    if arr[i]==True:
        j=2
        while i*j <= m:
            arr[i*j] = False
            j+=1
for i in range(n,m+1):
    if i>1 and arr[i]:
        print(i)

데이터가 1,000,000 개 이하에서 여러 소수를 구하는 문제는 에라토스테네스의 체 알고리즘을 사용한다.

구해야하는 크기만큼의 True 배열을 만든다.

2부터 구해야하는 크기의 제곱근까지의 자기자신을 제외한 배수를 모두 False을 만든다.

2부터 2를 제외하고 4, 6, 8, 10... False

3부터 3을 제외하고 3, 6, 9, 12 ... False

...

이렇게 하고 True 인 것만 출력하면 소수만 나온다.

마지막에 출력 시, i는 2부터임을 주의하자.

코드

import sys
s = list(sys.stdin.readline().rstrip())
n = int(sys.stdin.readline())
cursor = len(s)
for i in range(n):
    orders = sys.stdin.readline().split()
    if orders[0] == "L":
        if cursor == 0:
            continue
        cursor -= 1
    elif orders[0] == "D":
        if cursor == len(s):
            continue
        cursor += 1
    elif orders[0] == "B":
        if cursor == 0:
            continue
        s.remove(s[cursor-1])
        cursor-=1
    elif orders[0] == "P":
        s.insert(cursor,orders[1])
        cursor+=1
print(''.join(s))

풀이

시간초과나서 다른 사람 풀이를 참고했다....

WOW 

나는 슈퍼 멍충이...공부하자

stack을 2개 사용했다.

커서 기준으로 커서 왼쪽 관련이면 입력받은 s1에서 해결, 오른쪽 관련이면 s2에서 해결한다.

1. L이면 s1에 맨 위에 있는 것을 s2에 삽입
2. D이면 s2에 맨 위에 있는 것을 s1에 삽입
3. B이면 s1에 맨 위에 삭제
4. P이면 s1에 입력 원소 삽입

이렇게 조건에 만족하면 s1+s2을 출력 시에 s2는 뒤집어서 출력해야한다.

list(reversed(s2)

다른 사람 코드

import sys
s1 = list(sys.stdin.readline().rstrip())
n = int(sys.stdin.readline())
s2 = []
for i in range(n):
    orders = sys.stdin.readline().split()
    if orders[0] == "L" and len(s1)!=0:
        s2.append(s1.pop())
    elif orders[0] == "D" and len(s2)!=0:
        s1.append(s2.pop())
    elif orders[0] == "B" and len(s1)!=0:
        s1.pop()
    elif orders[0] == "P":
        s1.append(orders[1])
print(''.join(s1+list(reversed(s2))))

코드

import sys
from collections import deque
input = sys.stdin.readline
n = int(input())
q = deque()
for i in range(n):
    orders = input().split()

    if orders[0] == 'push_front':
        q.appendleft(orders[1])
    elif orders[0] == 'push_back':
        q.append(orders[1])
    elif orders[0] == 'pop_front':
        if len(q) == 0:
            print(-1)
        else:
            print(q.popleft())
    elif orders[0] == 'pop_back':
        if len(q) == 0:
            print(-1)
        else:
            print(q.pop())
    elif orders[0] == 'size':
        print(len(q))
    elif orders[0] == 'empty':
        if len(q) == 0:
            print(1)
        else:
            print(0)
    elif orders[0] == 'front':
        if len(q) == 0:
            print(-1)
        else:
            print(q[0])
    elif orders[0] == 'back':
        if len(q) == 0:
            print(-1)
        else:
            print(q[-1])

풀이

collections 의 deque를 사용했다.

deque는 첫 번째 원소를 제거할 때 popleft(), 마지막 원소를 제거할 때 pop()

첫 번째 인덱스에 원소 a 추가 할때 appendleft(a) , 마지막 인덱스에 원소를 삽입할 때 append(a)

코드

import sys
from collections import deque
n = int(input())
q = deque()
for i in range(n):
    orders = sys.stdin.readline().split()

    if orders[0] == 'push':
        q.append(orders[1])
    elif orders[0] == 'pop':
        if len(q) == 0:
            print(-1)
        else:
            print(q.popleft())
    elif orders[0] == 'size':
        print(len(q))
    elif orders[0] == 'empty':
        if len(q) == 0:
            print(1)
        else:
            print(0)
    elif orders[0] == 'front':
        if len(q) == 0:
            print(-1)
        else:
            print(q[0])
    elif orders[0] == 'back':
        if len(q) == 0:
            print(-1)
        else:
            print(q[-1])

풀이

sys 라이브러리로 입력받고 명령대로 수행한다.

python에서 큐는 리스트로도 가능하지만 collections에서 deque를 사용했다.