C++程序：移除操作后数组的减小尺寸

假设我们有一个包含n个元素的数组A。考虑一个包含n个正整数的密码。我们对数组执行以下操作：移除两个不相邻的元素，然后将它们的和放在该位置。此操作将使数组的大小减少1。我们必须找到执行这些操作后数组的最短可能长度。

因此，如果输入类似于A = [2, 1, 3, 1]，则输出将为1，因为如果我们选择(1, 3)，数组将为[2, 4, 1]，然后选择(2, 4)使数组变为[6, 1]，然后选择最后两个得到[7]。

步骤

为了解决这个问题，我们将遵循以下步骤：

n := size of A
Define one set se
for initialize i := 0, when i < n, update (increase i by 1), do:
   insert A[i] into se
if size of se is same as 1, then:
   return n
Otherwise
   return 1

示例

让我们看下面的实现以更好地理解：

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

int solve(vector<int> A) {
   int n = A.size();
   set<int> a;
   for (int i = 0; i < n; i++) {
      a.insert(A[i]);
   }
   if (a.size() == 1)
      return n;
   else
      return 1;
}
int main() {
   vector<int> A = { 2, 1, 3, 1 };
   cout << solve(A) << endl;
}

输入

{ 2, 1, 3, 1 }

输出

1

Arnab Chakraborty

更新于：2022年3月3日

122 次浏览

启动您的职业生涯

完成课程后获得认证

开始