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
广告