[이코테] 두 배열의 원소 교체

출처: 이것이 취업을 위한 코딩 테스트다  with 파이썬 (나동빈)

https://www.youtube.com/watch?v=94RC-DsGMLo&list=PLRx0vPvlEmdAghTr5mXQxGpHjWqSz0dgC&index=5

 

문제

• 돈빈이는 두 개의 배열 A와 B를 가지고 있습니다. 두 배열은 N개의 원소로 구성되어 있으며, 배열의 원소는 모두 자연수입니다.
• 동빈이는 최대 K번의 바꿔치기 연산을 수행할 수 있는데, 바꿔치기 연산이란 배열 A에 있는 원소 하나와 배열 B에 있는 원소 하나를 골라서 두 원소를 서로 바꾸는 것을 말합니다.
• 동빈이의 최종 목표는 배열 A의 모든 원소의 합이 최대가 되도록 하는 것이며, 여러분은 동빈이를 도와야 합니다.
• N,K 그리고 배열 A와 B의 정보가 주어졌을 때, 최대 K번의 바꿔치기 연산을 수행하여 만들 수 있는 배열 A의 모든 원소의 합의 최댓값을 출력하는 프로그램을 작성하세요.

 

문제 조건

• 풀이 시간: 15분
• 제한 시간: 2초
• 메모리 제한: 128MB

---

아이디어

 -A는 오름차순, B는 내림차순으로 정렬한다

 -첫번째 ~ k번째 원소들에 대하여 A,B의 원소들을 하나씩 교체해주는데,

 -교체하려는 A의 원소가 B의 원소보다 크다면 그때부터는 더이상 교체를 할 필요가 없으므로 교체작업을 종료해준다

 

작성 코드

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

A = list(map(int,input().split()))
B = list(map(int,input().split()))

A.sort()
B.sort(reverse=True)

for i in range(k):
  if A[i] > B[i]:
    break
  else: 
    A[i],B[i] = B[i],A[i]

sum=0
for i in range(n):
  sum += A[i]

print(sum)

 sum() 함수가 있는줄 몰라서 반복문 돌리면서 합을 구했다 헣

 

 

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

A = list(map(int,input().split()))
B = list(map(int,input().split()))

A.sort()
B.sort(reverse=True)

for i in range(k):
  if A[i] > B[i]:
    break
  else: 
    A[i],B[i] = B[i],A[i]

print(sum(A))

코드 마지막 부분만 수정해주면 완벽!