Category/빅데이터&파이썬

프로그래머스 알고리즘(예산, 내적, 나머지가 1이 되는 수 찾기, 문자열 내 p와 y의 개수, 정수 제곱근 판별)

sumin 2023. 4. 10. 22:22
728x90
반응형
  • 1. 예산

작성 코드

1
2
3
4
5
6
7
8
9
10
11
12
def solution(d, budget):
    answer = 0
    d.sort()
    while True:
        for i in d:
            if budget >= i:
                budget -= i
                answer += 1
            else:
                break
        break
    return answer
cs

 

  • 2. 내적

 

작성 코드

1
2
3
4
5
def solution(a, b):
    answer = 0
    for i in range(len(a)):
        answer += a[i]*b[i]
    return answer
cs

 

  • 3. 나머지가 1이 되는 수 찾기

 

작성 코드

1
2
3
4
5
6
7
8
9
10
11
12
def solution(n):
    answer = 0
    a = 0
    for i in range(n):
        if n%(i+1== 1:
            a = i+1
        if answer == 0:
            answer = a
        if answer >= a:
            answer = a
            
    return answer
cs

 

  • 4. 문자열 내 p와 y의 개수

 

작성 코드 1

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
def solution(s):
    answer = True
    a=0
    b=0
    
    for i in range(len(s)):
        if s[i] == 'p':
            a += 1
        if s[i] == 'P':
            a += 1
        if s[i] == 'y':
            b += 1
        if s[i] == 'Y':
            b += 1
    if a==b:
        return True
    else:
        return False
cs

작성 코드 2

1
return s.lower().count(‘p') == s.lower().count(‘y’)
cs

 

  • 5. 정수 제곱근 판별 

작성 코드

1
2
3
4
5
6
7
8
9
10
def solution(n):
    answer = 0
    num = n ** 0.5
 
    if num == int(num):
        answer = (num+1) ** 2
    else:
        answer = -1
 
    return answer
cs

 

728x90
반응형