[리뷰]그리디 - 큰수의 법칙

github.com/ndb796/python-for-coding-test

ndb796/python-for-coding-test

책을 받아서 공부용으로 코드를 정리합니다.

 

파이썬

import sys

n,m,k =  map(int, sys.stdin.readline().rstrip().split())

narr = list(map(int,sys.stdin.readline().rstrip().split()))

narr.sort()

first = narr[-1]
second = narr[-2]

result = 0

p = m // (k+1)
q = m % (k+1)

result = p * (first * k + second) + q*first
print(result)

c++

  #include <bits/stdc++.h>

using namespace std;

int main() {
  int n, m, k;
  
  vector<int> v;

  cin >>n >>m>>k;
  for(int i = 0;i<n;i++){
    int tmp;
    cin>>tmp;
    v.push_back(tmp);
  }


 sort(v.begin(),v.end());

  int first = v[v.size()-1];
  int second = v[v.size()-2];
  int p = m /(k+1);
  int q = m %(k+1);
  int result = p*(first*k+second)+q*first;
  cout<<result<<'\n';

  return 0;
}