220105_Code in the morning

Code in the morning, 매일 아침 알고리즘 문제를 풉니다. 

💬 문제 설명: 예산

S사에서는 각 부서에 필요한 물품을 지원해 주기 위해 부서별로 물품을 구매하는데 필요한 금액을 조사했습니다. 그러나, 전체 예산이 정해져 있기 때문에 모든 부서의 물품을 구매해 줄 수는 없습니다. 그래서 최대한 많은 부서의 물품을 구매해 줄 수 있도록 하려고 합니다.

물품을 구매해 줄 때는 각 부서가 신청한 금액만큼을 모두 지원해 줘야 합니다. 예를 들어 1,000원을 신청한 부서에는 정확히 1,000원을 지원해야 하며, 1,000원보다 적은 금액을 지원해 줄 수는 없습니다.

부서별로 신청한 금액이 들어있는 배열 d와 예산 budget이 매개변수로 주어질 때, 최대 몇 개의 부서에 물품을 지원할 수 있는지 return 하도록 solution 함수를 완성해주세요.

 

 

📍 제한 사항

• d는 부서별로 신청한 금액이 들어있는 배열이며, 길이(전체 부서의 개수)는 1 이상 100 이하입니다.
• d의 각 원소는 부서별로 신청한 금액을 나타내며, 부서별 신청 금액은 1 이상 100,000 이하의 자연수입니다.
• budget은 예산을 나타내며, 1 이상 10,000,000 이하의 자연수입니다.

 

 

💡 나의 풀이 

def solution(d, budget):

	#예산이 모든 부서를 지원할 수 있는 경우 	
    num_of_dept = 0
    if sum(d) <= budget:
        return(len(d))
    
    #예산이 모든 부서를 지원할 수 없는 경우   
    if sum(d) > budget: 
        d.sort()  #필요한 예산 오름차순으로 정렬
        bud_dept = [] #지원받을 수 있는 부서 
        sum_i = 0 #지원한 금액
        for i in d: 
            if i <= budget-sum_i:  #앞 부서를 지원하고 남은 금액이 그 다음부서가 필요한 금액보다 크거나 같다면
                bud_dept.append(i) #그 부서도 지원 가능
                sum_i = sum_i + i
        return(len(bud_dept))

 

 

💡 다른 해설

def solution(d, budget):
    d.sort()
    while budget < sum(d):
        d.pop()
    return len(d)

while문으로 엄청 쉽게 돌리네.. 

👉🏻 문제 바로가기 

 

 

 

---

 

 

Code in the morning, 매일 아침 알고리즘 문제를 풉니다. 

💬 문제 설명: 체육복

점심시간에 도둑이 들어, 일부 학생이 체육복을 도난당했습니다. 다행히 여벌 체육복이 있는 학생이 이들에게 체육복을 빌려주려 합니다. 학생들의 번호는 체격 순으로 매겨져 있어, 바로 앞번호의 학생이나 바로 뒷번호의 학생에게만 체육복을 빌려줄 수 있습니다. 예를 들어, 4번 학생은 3번 학생이나 5번 학생에게만 체육복을 빌려줄 수 있습니다. 체육복이 없으면 수업을 들을 수 없기 때문에 체육복을 적절히 빌려 최대한 많은 학생이 체육수업을 들어야 합니다.

전체 학생의 수 n, 체육복을 도난당한 학생들의 번호가 담긴 배열 lost, 여벌의 체육복을 가져온 학생들의 번호가 담긴 배열 reserve가 매개변수로 주어질 때, 체육수업을 들을 수 있는 학생의 최댓값을 return 하도록 solution 함수를 작성해주세요.

 

📍 제한 사항

• 전체 학생의 수는 2명 이상 30명 이하입니다.
• 체육복을 도난당한 학생의 수는 1명 이상 n명 이하이고 중복되는 번호는 없습니다.
• 여벌의 체육복을 가져온 학생의 수는 1명 이상 n명 이하이고 중복되는 번호는 없습니다.
• 여벌 체육복이 있는 학생만 다른 학생에게 체육복을 빌려줄 수 있습니다.
• 여벌 체육복을 가져온 학생이 체육복을 도난당했을 수 있습니다. 이때 이 학생은 체육복을 하나만 도난당했다고 가정하며, 남은 체육복이 하나이기에 다른 학생에게는 체육복을 빌려줄 수 없습니다.

 

 

💡 나의 풀이 

def solution(n, lost, reserve):
    #temp는 n개의 인덱스를 가지며, 모두 체육복을 한 개씩 가져왔다고 가정. 
    temp = []
    i = 0
    while i < n :
        temp.append(1)
        i += 1
        
    #체육복을 잃어버린 학생들은 -1
    for l in lost: 
        temp[l-1] = temp[l-1] -1
    

    #체육복을 여벌로 가져온 학생들 +1
    for r in reserve: 
        temp[r-1] = temp[r-1] + 1
    
    cnt = 0 #체육 수업에 참여하는 학생 수
   
    for i in range (0,n-1):
        
        if (temp[i] == 0) and (temp[i+1]==2):
            temp[i] += 1
            temp[i+1] -=1
    
        elif (temp[i] == 2) and (temp[i+1]==0):
            temp[i] -= 1
            temp[i+1] +=1
        
    for i in range(0,n):
        if temp[i] >= 1 or temp[i] >=2 :
            cnt+=1
            
            
    return cnt

 

다 풀고나니 엄청 뿌듯하네

test case12에서 걸렸는데, 

파이썬 배열의 음수 인덱스는 끝의 숫자들을 가져오는 것을 깜빡했기 때문. 

 

👉🏻 문제 바로가기