C++ program to find reduced size of the array after removal operations

Suppose we have an array A with n elements. Consider there is a password with n positive integers. We apply the following operations on the array. The operations is to remove two adjacent elements that are not same as each other, then put their sum at that location. So this operation will reduce the size of array by 1. We have to find the shortest possible length of the array after performing these operations.

So, if the input is like A = [2, 1, 3, 1], then the output will be 1, because if we select (1, 3), the array will be [2, 4, 1], then pick (2, 4) to make the array [6, 1], then select the last two to get [7].

Steps

To solve this, we will follow these steps −

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

Example

Let us see the following implementation to get better understanding −

#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;
}

Input

{ 2, 1, 3, 1 }

Output

1

Arnab Chakraborty

Updated on 03-Mar-2022 10:13:13

• Related Questions & Answers
• C++ Program to find array after removal from maximum
• C++ program to find array after removal of left occurrences of duplicate elements
• Program to find winner of array removal game in Python
• Program to find length of longest contiguously strictly increasing sublist after removal in Python
• Program to find maximize score after n operations in Python
• C++ code to find corrected text after double vowel removal
• C++ program to count expected number of operations needed for all node removal
• Program to find lexicographically smallest string after applying operations in Python
• C++ program to find winner of ball removal game
• C++ code to find final number after min max removal game
• Print modified array after multiple array range increment operations in C Program.
• Program to find minimum possible maximum value after k operations in python
• Binary array after M range toggle operations?
• C++ Program to find maximum score of bit removal game
• Program to find minimum operations to make the array increasing using Python