Algorithm/Sort

[Python] 두 배열의 원소 교체

코딩쪼앙 2023. 1. 20. 18:05

문제

  • 동빈이는 두 개의 배열 A와 B를 가지고 있다. 두 배열은 N개의 원소로 구성되어 있으며 배열의 원소는 모두 자연수이다
  • 동빈이는 최대 K번의 바꿔치기 연산을 수행할 수 있는데, 바꿔치기란 배열 A에 있는 원소 하나와 배열 B에 있는 원소 하나를 골라서 두 원소를 서로 바꾸는 것을 말한다
  • 동빈이의 최종 목표는 배열 A의 모든 원소의 합이 최대가 되도록 하는 것이며, 여러분은 동빈이를 도와야한다

입력

5 3
1 2 5 4 3
5 5 6 6 5

출력

26

문제 풀이

  • A 배열 오름차순으로 정렬
  • B 배열 내림차순으로 정렬
  • A 배열의 원소 값이 B 배열의 원소 값보다 작다면 두 원소를 바꾼다
  • A 배열의 모든 원소 값을 더한 값 출력

코드

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]:
        a[i],b[i] = b[i],a[i]
print(sum(a))

'Algorithm > Sort' 카테고리의 다른 글

[Python] 실패율  (0) 2023.01.25
[Python] 안테나  (0) 2023.01.22
[Python] 국영수  (0) 2023.01.22
[Python] 성적이 낮은 순서로 학생 출력하기  (2) 2023.01.20
[Python] 위에서 아래로  (0) 2023.01.20