C++ 中的最大权重差

c++server side programmingprogramming

在这个问题中,给定一个数组 arr[] 和一个数字 M。我们的任务是编写一个程序来计算 C++ 中的最大权重差。

问题描述

我们将从数组中找出 M 个元素,使得它们的和与其余元素和的绝对差最大。

让我们举个例子来理解这个问题:

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

输出: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

相关文章