C++ 中的最大重量差


在這個問題中,我們給定一個數組 arr[] 和一個數 M。我們的任務是建立一個程式,用 C++ 計算最大重量差。

問題表述

我們將從陣列中找 M 個元素,使它們之和與其餘元素之和的絕對差最大。

讓我們舉個例子來理解這個問題:

輸入:arr[] = {3, 1, 6, 9, 4} M = 3

輸出:15:15

說明:

我們考慮 4、6、9。它們的和為 19。它們與其餘數字之和的絕對差為

|19 - 4| = 15

解決方案途徑

這個問題的解決方案基於這樣一個事實,即只有在 M 個最大數或 M 個最小數的情況下才可能有最大差值。我們需要檢查這兩個條件並比較絕對差,以找到最終結果。為了簡化查詢 M 個最大數/最小數的過程,我們對陣列進行排序,然後對 K 個最大數和最小數進行計算。

示例

 線上演示

#include <bits/stdc++.h>
using namespace std;
int maxWeightDifference(int arr[], int N, int M){
   int maxabsDiff = -1000;
   sort(arr, arr + N);
   int sumMin = 0, sumMax = 0, arrSum = 0;
   for(int i = 0; i < N; i++){
      arrSum += arr[i];
      if(i < M)
         sumMin += arr[i];
      if(i >= (N-M))
         sumMax += arr[i];
   }
   maxabsDiff = max(abs(sumMax - (arrSum - sumMax)), abs(sumMin -(arrSum - sumMin)));
   return maxabsDiff;
}
int main(){
   int arr[] = {3, 1, 6, 9, 4} ;
   int M = 3;
   int N = sizeof(arr)/sizeof(arr[0]);
   cout<<"The maximum weight difference is "<<maxWeightDifference(arr,N, M);
   return 0;
}

輸出

The maximum weight difference is 15

更新於:2020-10-09

171 次瀏覽

開啟你的 職業 生涯

完成課程即可獲得認證

開始學習
廣告