使用 C/C++ 进行阵列操作和求和

这里我们将了解一个问题，假设给定一个阵列。有 n 个元素。还给出一个值 S。我们必须在阵列中找到一个元素 K，使得如果将所有大于 K 的元素都改成 K，则最终阵列的所有元素的总和将变成 S。如果不可能，则返回 -1。

假设元素为 {12, 6, 3, 7, 8}，求和值为 15，输出为 3。最终阵列为 {3, 3, 3, 3, 3}，阵列元素的总和为 S = 15

算法

getVal(arr, n, S) −

Begin
   sort arr as increasing order
   sum := 0
   for i in range 0 to n-1, do
      if sum + (arr[i] * (n - i)) is same as S, then
         return arr[i]
      end if
      sum := sum + arr[i]
   done
   return -1
End

示例

#include <iostream>
#include <algorithm>
using namespace std;
int getVal(int arr[], int n, int S) {
   sort(arr, arr + n);
   int sum = 0;
   for (int i = 0; i < n; i++) {
      if (sum + (arr[i] * (n - i)) == S) //if current value is satisfying, then return arr[i]
         return arr[i];
      sum += arr[i];
   }
   return -1;
}
int main() {
   int S = 15;
   int arr[] = { 12, 3, 6, 7, 8 };
   int n = sizeof(arr) / sizeof(arr[0]);
   cout << getVal(arr, n, S);
}

输出

3

更新于：20-Aug-2019

2K+ 浏览量

开启您的 职业生涯

完成课程认证

开始
广告