문제 설명

1. N 개의 정수를 가진 리스트 A와 정수 K가 입력으로 주어진다.
2. 문제의 목표는 A리스트를 K번 회전한 결과 리스트를 리턴하는 것.
3. 회전이란 리스트의 모든 요소가 하나씩 오른쪽으로 이동하는 것을 말한다(마지막 요소는 첫번째로 이동)

풀면서 주의할 점은 예외케이스를 놓치지 않는 것이다. 

입력으로 빈 리스트가 주어지거나 

리스트는 입력 됬는데 회전 횟수가 0으로 입력되는 경우를 처음에 생각못해서 헤맸다...  

 you can write to stdout for debugging purposes, e.g.
# print("this is a debug message")

def solution(A, K):
    # write your code in Python 3.6
    if len(A) == 0:	## 빈 리스트가 입력으로 들어올 수 있다.
        return []
    if K == 0:		## 리스트만 입력되고 회전 수가 0으로 주어질 수 있다.
        return A
        
    while K > 0:
        
        result = []
        for i in range(len(A)):
            if i == 0:
                result.append(A[-1])
            else:
                result.append(A[i-1])
        A = result
        K -= 1
    return result
    pass

Codility 사이트에서 문제를 풀어보았다.

영어로 된 문제에 처음부터 당황..

하지만 수많은 예시들 덕분에 이해하는데 어렵진 않았다.

파이썬 알고리즘 공부를 하면서 여러가지 유용하고, 헷갈리는 문법들 정리

#1 시간, 공간 복잡도 관련 팁

알고리즘 수행시간을 알아보기 위한 코드

import time
start_time = time.time()

# 프로그램 작성

end_time = time.time()
print("time :", end_time - start_time)

- pypy3는 파이썬3의 문법을 그대로 지원하며, 대부분 파이썬3보다 실행 속도가 더 빠르다. Pypy3를 지원하는 환경이라면 속도를 줄이기 위해 이를 이용하도록 하자

- 기업코딩 테스트 환경에서는 파이썬으로 제출한 코드가 1초에 2000만번 연산을 수행한다고 가정하면 크게 무리가 없다.

- 시간제한이 1초인 문제를 만났을 경우 (참고만 하세요)

- 일반적으로 탐색해야할 전체 데이터의 개수가 100만개 이하일 때 완전탐색을 사용하면 적절하다.

- 정수형 자료형 데이터의 개수에 따른 메모리 사용량

#2 입출력

공백을 기준으로 입력받기

# 3 3 3     <-- 변수로 저장
n, m, k = map(int, input().split())

# 2 4 5 2 3 <-- 배열로 저장
data = list(map(int, input().split()))

2차원 배열 입력받기

data = []
for i in range(n):
    # 데이터가 공백없는 형태로 입력될 때
    # 12345 --> [1, 2, 3, 4, 5]
    data.append(list(map(int, input())))
    
    # 데이터가 공백을 두고 입력될 떄
    # 1 2 3 4 5 --> [1, 2, 3, 4, 5]
    data.append(list(map(int, input().split())))

빠르게 입력받기 (일반적인 입력 방식보다 더 효율적)

import sys

data = sys.stdin.readline().rstrip()

f-string (더 간결하게 여러 자료형을 한 줄로 출력할 수 있음)

num = 7
print(f"정답은 {num}입니다.")

#3 정렬

#0 내림차순 정렬
B.sort(reverse=True) # 내림차순정렬

#1 튜플리스트, 2차원리스트일 경우 여러 가지 기준으로 정렬하기.
s_list.sort(key = lambda x : (-x[1], x[2], -x[3], x[0]))

#2 튜플에 숫자형과 문자형이 섞여있을 경우 숫자형에 int()를 씌워주면
# bad operand type for unary -: 'str' 이슈를 해결할 수 있다
s_list.sort(key = lambda x : (-int(x[1]), int(x[2]), -int(x[3]), x[0] ))

#3 리스트에서 median 값 구하는 식
data[(n-1) // 2] # n은 리스트 길이

#4 sorted with key
array = [('홍길동', 35), ('이순신', 75), ('아무개', 50)]
result = sorted(array, key = lambda x: x[1])
# 성적을 기준으로 오름차순정렬


#5 문자열, 배열, 튜플 등을 역순으로 출력(간결하게)
s = 'abcde'
print(s[::-1])  # 'edcba'
print(s[3:0:-1])  # dcba, 0~3인덱스를 역순으로

l = ['a', 'b', 'c', 'd', 'e']
print(l[::-1])  # ['e', 'd', 'c', 'b', 'a']
t = ('a', 'b', 'c', 'd', 'e')
print(t[::-1])  # ('e', 'd', 'c', 'b', 'a')

#4 문자열, 배열 다루기

파이썬 문자열 아스키변환

주요 아스키 코드값 0 ~ 9 : 48 ~ 57 , A ~ Z : 65 ~ 90, a ~ z : 97 ~ 122

ord('a') ## 아스키코드 숫자 반환
chr(ord('a')) ## 아스키코드에 해당하는 문자 반환

크기가 N이고 모든 값이 0인 리스트 초기화

N = 10
arr = [0] * N

1부터 10까지 증가하는 리스트 초기화

array = [i for i in range(1, 11)]

2차원 배열 좌표 이동

## 풀이 - 좀더 간단한 코드로

n = int(input())
x, y = 1,1
plans = input().split()

## L, R, U, D
dx = [0, 0, -1, 1]
dy = [-1,1,  0, 0]
move = ['L', 'R', 'U', 'D']

for plan in plans:
    for i in range(len(move)):
        if plan == move[i]:
            nx = x + dx[i]
            ny = y + dy[i]
    if nx < 1 or ny < 1 or nx > n or ny > n:
        continue
    x, y = nx, ny
print(x, y)

2차원 배열 90도 회전

n = len(a) # 행 길이 계산
m = len(a[0]) # 열 길이 계산
result = [[0] * n for _ in range(m)]
after = [] # 90도 회전된 배열 저장

# 시계방향으로 회전
for i in range(n):
	tmp = []
	for j in range(m):
    	tmp.append(before[m-1-j][i])
    after.append(tmp)
    
# 반시계방향으로 회전
for i in range(m):
    tmp  = []
    for j in range(n):
    	tmp.append(before[j][m-1-i]) #반시계

    after.append(tmp)

#5 자료형

- 1e9, 1E9 는 10의 9제곱과 같다.

- 임의의 큰 수를 표현하기 위해 자주 표현되는 형식

# 실수형의 계산은 정확하게 떨어지지 않는다.
# 반올림해서 사용하는 것을 습관화하자

if 0.3 + 0.6 == 0.9:
	print(True)
else:
	print(False)

# --> False

- 튜플은 한 번 선언되면 변경할 수 없다.

- 튜플은 리스트에 비해 공간 효율적이다.

#6 내장함수 및 라이브러리

eval() : 문자열로 표현된 수식의 계산 결과값을 반환

result = eval("(3+5)*7")
# --> 56

순열과 조합

# 순열 : 주어진 데이터를 가지고 서로 다른 n 개를 선택하여 일렬로 나열하는 것 
from itetools import permutations

data = [1, 4, 5, 6, 7]
result = list(permutations(data,3))

# 조합 : 주어진 데이터를 가지고 서로 다른 n 개를 선택하는 것 
from itetools import combinations

data = [1, 4, 5, 6, 7]
result = list(combinations(data,3))

Counter : 원소의 등장 횟수를 간단히 셀 수 있다, 원소와 등장 횟수를 사전 형태로 저장할 수 있다.

from collections import Counter

counter = Counter(['red', 'blue', 'green'])
print(dict(counter))

# --> {'red' : 1, 'blue' : 1, 'green' : 1}

7 3

<3, 6, 2, 7, 5, 1, 4>

<풀이과정>

큐를 이용해서 1부터 n까지 만들고 

원을 이루면서 돌아가는 것을  

큐에서 첫 번째를 뽑아서 다시 맨 뒤에 넣는 방식으로 설계 

항상 첫번 째를 가리키고 한 칸씩 움직이는 것을 m-1번씩 반복하면서 구현 

정답~!

Hello
Baekjoon
Online Judge

Hello
Baekjoon
Online Judge

<풀이과정> 

1. C++ 언어를 이용한 풀이 

문자열을 통채로 저장하는 string 자료형을 사용한다. 

한 줄을 입력받아 저장할 수 있는 getline 함수를 사용 

whille(true)로 설정하여 무한 루프를 만들고 입력값이 공백일 때 루프를 빠져나가도록 한다. 

2. C언어를 이용한 풀이

한 줄에 받을 수 있는 최대 100글자의 배열을 char 자료형의 배열로 만들어 저장한다. 

줄바꿈을 제외하고 한 줄로 입력받고 한 줄을 문자형 배열로 입력받는다. 

입력이 제대로 되었으면 계속 반복 여기서는 루프를 빠져나가는 탈출 조건을 따로 주지 않았다. 

둘다 정답!!

코드의 효율은 C언어를 사용한 것이 더 좋은 모습

yellow
violet
red

4700

<풀이과정> 

아래 처음 코드는 

자료형을 제대로 사용하지 않아 틀렸다..

아래 코드는 최대값을 입력했을 때 오류를 일으킨다. 

문제에서 나올 수 있는 최대 저항의 값은 

가장 큰 곱셉값인 10억 * 처음색 두개 두자리수 -> 백억단위 이상

문제의 조건을 보면 음수 값은 나올 수 없다.

--> unsigned long long 자료형 사용 

<c++ 자료형 범위와 크기 비교>

<최종코드>

정답!

14
push 1
push 2
top
size
empty
pop
pop
pop
size
empty
pop
push 3
empty
top

2
2
0
2
1
-1
0
1
-1
0
3

<풀이과정>

명령어의 수 만큼 반복되는 루프 만들고 

명령어를 입력 받아 if문을 통해 명령어를 확인

명령어가 확인되면 해당하는 명령어의 기능을 수행한다. 

6
(())())
(((()())()
(()())((()))
((()()(()))(((())))()
()()()()(()()())()
(()((())()(

NO
NO
YES
NO
YES
NO

<풀이과정>

첫번째 틀린 풀이 

문자열을 입력받고 '(' 가 나오면 스택에 추가하고 ')'가 나오면 스택에서 삭제하는 방식으로 설계했다. 

그래서 스택이 최종적으로 비어있으면 YES, 스택에 무언가 남아있으면 NO를 출력하게 하였다. 

그러나 이 방식의 문제점은 ')'가 먼저 나오는 경우에 대해서 오류를 잡아내지 못했다. 

5
3 2 1 -3 -1

1 4 5 3 2

<풀이과정> 

풍선에 적힌 쪽지가 음수인지 양수인지에 따라서 앞뒤로 push, pop 을 해줘야 하기 때문에 

기본적으로 deque자료형이 편리하지만 나는 그냥 vector자료형을 사용해서 풀었다. 

풍선의 번호와 숫자쪽지를 pair 배열로 생성. 

풍선이 터지면 배열에서 삭제하고 배열에 남아있는게 없을 때 까지 반복

종이에 적힌 숫자가 양수면 맨 앞의 요소를 맨뒤로 이동, 음수면 맨 뒤의 요소를 맨 앞으로 이동하는 방식으로 풍선의 이동을 구현.

이동의 횟수는 양수일 때는 종이에 적힌 숫자-1, 음수일 경우는 종이에적인 숫자의 절대값만큼 반복해야한다.  

abcd
3
P x
L
P y

abcdyx

abc
9
L
L
L
L
L
P x
L
B
P y

yxabc

dmih
11
B
B
P x
L
B
B
B
P y
D
D
P z

yxz

<풀이과정> 

- 리스트를 이용한 풀이

초기 문자열을 입력받고 문자열을 리스트로 만들어서 풀이한다. 

문자열을 그대로 배열처럼 사용하게되면 삽입, 삭제 할 때 마다 모든 요소에 변화가 생겨 비효율적인 코드가 된다.

주의할 점은 'B' 삭제 명령어를 사용할 때 왼쪽 문자를 삭제하기 위해 커서를 왼쪽으로 이동하여 삭제한다. 

그러나 커서 오른쪽에 오는 문자는 그대로라고 했기 때문에 커서의 위치는 삭제 후에도 변함이 없어야 하기 때문에 

다시 오른쪽으로 이동하여 커서의 위치를 유지해준다. 

- 스택을 이용한 풀이 

커서를 기준으로 왼쪽과 오른쪽의 스택을 만든다. 

문제에서 필요한 작업은 왼쪽스택에서의 삽입과 삭제 - push, pop 명령으로 구현한다. 

커서의 오른쪽 이동은 왼쪽스택에 있는 요소를 오른쪽 스택에 옮기는 것으로,

왼쪽 이동은 오른쪽 스택에  있는 것을 왼쪽 스택에 옮기는 것으로 구현한다. 

마지막 출력을 위해서 오른쪽스택에 다 이동한 다음에 스택의 요소들을 하나씩 순서대로 출력한다.