rajeshinternshala

Untitled

Sep 9th, 2023
63
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C 0.83 KB | None | 0 0
  1. long long getm(long long k,vector<long long>r1,vector<long long>r2){
  2.     long long rem = r1.size() - k;
  3.     long long n = r1.size();
  4.     vector<pair<long long,long long>>vp;
  5.     for(long long i=0;i<n;i++){
  6.         vp.push_back({abs(r1[i]-r2[i]),i});
  7.     }
  8.     sort(vp.begin(),vp.end());
  9.     long long tt = 0;
  10.     for(long long i=n-1;i>=0;i--){
  11.         long long in = vp[i].second;
  12.         if(r1[in]>r2[in]){
  13.             if(k!=0){
  14.                 tt += r1[in];
  15.                 k--;
  16.             }
  17.             else {
  18.                 tt += r2[in];
  19.                 rem--;
  20.             }
  21.         }
  22.         else{
  23.              if(rem!=0){
  24.                 tt += r2[in];
  25.                 rem--;
  26.             }
  27.             else {
  28.                 tt += r1[in];
  29.                 k--;
  30.             }
  31.         }
  32.     }
  33.     return tt;
  34. }
Add Comment
Please, Sign In to add comment