C++ 陣列的平衡索引


在這個問題中,我們給定一個包含 n 個整數值的陣列 arr[]。我們的任務是建立一個程式來查詢陣列的平衡索引。

平衡索引是指陣列中索引,該索引之前所有元素的和等於該索引之後所有元素的和。

對於大小為 n 的陣列 arr[],平衡索引 e 滿足以下條件:

sum (arr[0… e-1] ) = sum (arr[e… n-1])

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

輸入:arr[] = {5, 1, 2, 8, 3, 4, 1}

輸出:3

解釋:

arr[0] +  arr[1] + arr[2] = arr[4] + arr[5] + arr[6]

=> 5 + 1 + 2 = 3 + 4 + 1

=> 8 = 8

解決方案方法

一個簡單的方法是透過不斷檢查陣列中的任何元素是否可以成為平衡數來找到該元素。

為此,我們將使用巢狀迴圈。外部迴圈用於迭代陣列的元素。內部迴圈將檢查陣列中的任何元素是否可以成為平衡數。

程式說明了我們解決方案的工作原理:

示例

線上演示

#include <bits/stdc++.h>
using namespace std;

int findEquilibriumIndex(int arr[], int n)
{
   int prevSum, nextSum;

   for (int i = 0; i < n; ++i) {  

      prevSum = 0;
      for (int j = 0; j < i; j++)
         prevSum += arr[j];
      nextSum = 0;
      for (int j = i + 1; j < n; j++)
         nextSum += arr[j];
         
      if (prevSum == nextSum)
         return i;
   }
   return -1;
}

int main() {
   
   int arr[] = {5, 1, 2, 8, 3, 4, 1};
   int n = sizeof(arr) / sizeof(arr[0]);
   cout<<"The equilibrium index is "<<findEquilibriumIndex(arr, n);
   return 0;
}

輸出 -

The equilibrium index is 3

更新於: 2021年1月22日

527 次瀏覽

開啟您的 職業生涯

透過完成課程獲得認證

立即開始
廣告

© . All rights reserved.