본문 바로가기

연구노트/Python

파이썬 배열, 문자열, 반복문 활용 코드

 

 
 
 파이썬 배열, 문자열, 반복문 활용 코드
1번 문제
파이썬 배열, 문자열, 반복문 활용 코드

 

정수 N개로 이루어진 수열 A와 정수 X가 주어진다. 이때, A에서 X보다 작은 수를 모두 출력하는 프로그램을 작성하시오.

 

입력

첫째 줄에 N과 X가 주어진다. (1 ≤ N, X ≤ 10,000)

둘째 줄에 수열 A를 이루는 정수 N개가 주어진다. 주어지는 정수는 모두 1보다 크거나 같고, 10,000보다 작거나 같은 정수이다.

 

출력

X보다 작은 수를 입력받은 순서대로 공백으로 구분해 출력한다. X보다 작은 수는 적어도 하나 존재한다.

# 예제
10 5
1 10 4 9 2 3 8 5 7 6
→ 1 4 2 3
 

 

풀이

# 풀이
import random

N = int(input("정수N: "))
X = int(input("정수X: "))

while True:
    if 1 <= N <= 10000 and 1 <= X <= 10000:
        A = []
        for i in range(N):
            sec = random.randrange(1,N+1)
            A.append(sec)
        print(A)
        B = []
        for j in range(N):
            if A[j] < X:
                B.append(A[j])
        if B == []:
            print("다시 실행하세요.")
        else:
            print(B)
        break
    else:
        print("숫자를 다시 입력하세요.")
 

설명

A리스트에 1 ~ N까지의 랜덤 숫자를 저장합니다. 그 다음 A리스트의 숫자 중 X보다 작은 수를 B리스트에 저장합니다.

B리스트를 출력하면 원하는 결괏값이 나오게 됩니다.

 

 

 

2번 문제
파이썬 배열, 문자열, 반복문 활용 코드

 

상근이의 할머니는 아래 그림과 같이 오래된 다이얼 전화기를 사용한다.

전화를 걸고 싶은 번호가 있다면, 숫자를 하나를 누른 다음에 금속 핀이 있는 곳 까지 시계방향으로 돌려야 한다. 숫자를 하나 누르면 다이얼이 처음 위치로 돌아가고, 다음 숫자를 누르려면 다이얼을 처음 위치에서 다시 돌려야 한다.

숫자 1을 걸려면 총 2초가 필요하다. 1보다 큰 수를 거는데 걸리는 시간은 이보다 더 걸리며, 한 칸 옆에 있는 숫자를 걸기 위해선 1초씩 더 걸린다.

상근이의 할머니는 전화 번호를 각 숫자에 해당하는 문자로 외운다. 즉, 어떤 단어를 걸 때, 각 알파벳에 해당하는 숫자를 걸면 된다. 예를 들어, UNUCIC는 868242와 같다.

할머니가 외운 단어가 주어졌을 때, 이 전화를 걸기 위해서 필요한 최소 시간을 구하는 프로그램을 작성하시오.

 

입력

첫째 줄에 알파벳 대문자로 이루어진 단어가 주어진다. 단어의 길이는 2보다 크거나 같고, 15보다 작거나 같다.

출력

첫째 줄에 다이얼을 걸기 위해서 필요한 최소 시간을 출력한다.

 

# 예제
WA
→ 13
UNUCIC
→ 36
 

 

풀이

# 풀이
a = {
    'A': 3, 'B': 3, 'C': 3,
    'D': 4, 'E': 4, 'F': 4,
    'G': 5, 'H': 5, 'I': 5,
    'J': 6, 'K': 6, 'L': 6,
    'M': 7, 'N': 7, 'O': 7,
    'p': 8, 'Q': 8, 'R': 8, 'S': 8,
    'T': 9, 'U': 9, 'V': 9,
    'W': 10, 'X': 10, 'Y': 10, 'z': 10,
    '1':2, '0':11}                          #dictionary

def add(*args):
    result = 0
    for i in args:
        result = result + i
    return result                           #function

text = input("문자를 입력하세요.: ")
TEXT = text.upper()
time = []
for j in TEXT:
    sec = a[j]
    time.append(sec)                        #input

total_time = add(*time)
print(total_time)                           #output
 

설명

우선 다이얼 알파벳에 맞게 a 딕셔너리를 선언합니다.

여러 개의 인자를 받기위해 *args를 이용해 덧셈 함수를 만들어 줍니다.

문자를 입력하면 대문자로 만들어 주고 time 리스트 각 문자에 대응하는 숫자를 추가합니다.

함수를 이용해 총 걸린 시간을 계산하여 출력합니다.

 

3번 문제
파이썬 배열, 문자열, 반복문 활용 코드

 

도현이는 바구니를 총 N개 가지고 있고, 각각의 바구니에는 1번부터 N번까지 번호가 매겨져 있다. 또, 1번부터 N번까지 번호가 적혀있는 공을 매우 많이 가지고 있다. 가장 처음 바구니에는 공이 들어있지 않으며, 바구니에는 공을 1개만 넣을 수 있다.

도현이는 앞으로 M번 공을 넣으려고 한다. 도현이는 한 번 공을 넣을 때, 공을 넣을 바구니 범위를 정하고, 정한 바구니에 모두 같은 번호가 적혀있는 공을 넣는다. 만약, 바구니에 공이 이미 있는 경우에는 들어있는 공을 빼고, 새로 공을 넣는다. 공을 넣을 바구니는 연속되어 있어야 한다.

공을 어떻게 넣을지가 주어졌을 때, M번 공을 넣은 이후에 각 바구니에 어떤 공이 들어 있는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N (1 ≤ N ≤ 100)과 M (1 ≤ M ≤ 100)이 주어진다.

둘째 줄부터 M개의 줄에 걸쳐서 공을 넣는 방법이 주어진다. 각 방법은 세 정수 i j k로 이루어져 있으며, i번 바구니부터 j번 바구니까지에 k번 번호가 적혀져 있는 공을 넣는다는 뜻이다. 예를 들어, 2 5 6은 2번 바구니부터 5번 바구니까지에 6번 공을 넣는다는 뜻이다. (1 ≤ i ≤ j ≤ N, 1 ≤ k ≤ N)

도현이는 입력으로 주어진 순서대로 공을 넣는다.

출력

1번 바구니부터 N번 바구니에 들어있는 공의 번호를 공백으로 구분해 출력한다. 공이 들어있지 않은 바구니는 0을 출력한다.

 

# 예제
5 4
1 2 3
3 4 4
1 4 1
2 2 2
→ 1 2 1 1 0
 

 

풀이

# 풀이
N = int(input("N숫자를 입력하세요:"))
M = int(input("M숫자를 입력하세요:"))

basket = [0]*N
if 1<= N <= 100 and 1<= M <= 100:
    for p in range(M):
        i, j, k = map(int,input().split())
        if 1<=i<=j<=N and 1<=k<=100:
            for q in range(i-1,j):
                basket[q] = k
        else:
            print("숫자를 다시 입력하세요.")
            break
print(basket)
 

설명

0이 N개 존재하는 basket 리스트를 만듭니다.

N, M에 대한 조건식을 작성 후 i, j, k에 정수를 M번 입력하도록 반복문을 작성했습니다.

※i,j,k는 띄어 쓰기를 통해 숫자를 구분합니다.

이후 i, j, k에 대한 조건식을 바탕으로 q번째 바구니에 k숫자를 대입합니다.

이를 반복하면 바구니에 마지막 대입 숫자만 저장되게 됩니다.

 

'연구노트 > Python' 카테고리의 다른 글

파이썬 숫자 및 문자열 처리 기초  (0) 2025.09.30
파이썬 반복문 코드  (2) 2025.06.26
Python 정규 표현식  (0) 2025.05.08
파이썬 프로그래밍 기초(3)  (0) 2025.04.30
파이썬 프로그래밍 기초(2)  (0) 2025.04.30