[백준] 정렬/파이썬 11004 K번째 수, 문자열/1110더하기싸이클

📌 11004 K번째 수

www.acmicpc.net/problem/11004

11004번: K번째 수

 

문제보고 읭?

이거 받아서 정렬하고 인덱스로 풀면 되겠는데,,, 햇지만 매우 찜찜해햇다.

시간초과를 걱정하며 

넣었는데 일단 통과.

 

내 풀이

import sys
input = sys.stdin.readline;

n,k = map(int,input().split())

num = list(map(int,input().split()))
num.sort()

print(num[k-1])

< 하지만 살짝 처참한 시간 결과...

 

 

 

랭크정렬로 해볼까 했지만 중복이 없다는 얘기가 없어서 일단 패스.

 

 

해당 부분 축약 가능

num = list(map(int,input().split()))
num.sort()
print(num[k-1])


print(sorted(list(map(int,inpuyt().split()))[k-1])

굳이 num이라는 배열을 만들어 넣을 필요 없이 바로 출력이 가능하며, 이렇게 하면 3900ms 대로 단축할 수 잇다.

---

 

📌 1110_더하기사이클

내 풀이1

import sys
input = sys.stdin.readline


num=input().strip()
newNum=num
count=0


while True:
    if int(newNum)<10:
        newNum='0'+newNum

    fn=int(newNum)%10
    sn=(int(newNum[0])+int(newNum[1]))%10
    ans=str(fn)+str(sn)
    if ans == num:
        count+=1
        print(count)
        break
    else:
        newNum=str(ans);
        count+=1

처음에는 받을때 문자로 받고 

문자열끼리 더해주고 

str <-> int를  왔다갔다하는 코드를 만들었다.(ㅠㅠ)

 

26 과같은 두자리 수는 잘 나오지만

1과 같이 한자리 수는 에러가 나왔다.

 

sn의 연산이 두자리 수에서만 적용되기 때문이다.

그 이후에 엄청난 지옥에 빠졌다.

분명 간단한 건데 저 알고리즘에서 벗어나지 않으려고만 해서 답이 나오지 않았다. 

다 갈아엎고 수학적으로 접근하기로 결정했다.

 

 

 

내 풀이2

import sys
input = sys.stdin.readline

num=int(input())
newNum=100
fn=num//10 
sn=num%10
add=fn+sn
newNum = sn*10+add%10

count=1


while num!=newNum:
    fn=newNum//10 
    sn=newNum%10
    add=fn+sn
    newNum = sn*10+add%10
    count+=1
    
print(count)

1) n을 숫자로 입력받는다 26

2) newNum은 100으로초기화 (num은 두자리수를 넘지 않기 때문에)

3) fn : 2 sn: 6

4) add : 8

5) 새로운숫자 newNum : 68 (sn*10 + add*%10) 

 

계속 연산결과는 newNum에 입력하고 num과 비교해보며 count를 센다.

 

축약 간으

import sys
input = sys.stdin.readline

num=int(input())
newNum=num

count=1

while (True):
    fn=newNum//10 
    sn=newNum%10
    add=fn+sn
    newNum = sn*10+add%10
    if newNum==num:
        break
    count+=1
    
print(count)

나의 풀이에서는 처음 계산하는 부분이 중복되어있지만 

while문 + if문으로도 가능하다.

다만 속도 차이는 없었다.

 

💙오답이 나왔을 때 과감하게 다른 풀이 방법으로 전환하는 태도가 필요하다. 작은 부분에 집착하여 큰 흐름을 잃지 말자