GIVEN AN INTEGER ARRAY A OF SIZE N, CHECK IF THE INPUT ARRAY CAN BE SPLITTED IN TWO PARTS SUCH THAT –
TITLE:-Split Array
PROBLEM:-
Given an integer array A of size N, check if the input array can be splitted in two parts such that –
– Sum of both parts is equal
– All elements in the input, which are divisible by 5 should be in the same group.
– All elements in the input, which are divisible by 3 (but not divisible by 5) should be in another group.
– Elements that are neither divisible by 5 nor by 3, can be put in any group.
Groups can be made with any set of elements, i.e. elements need not to be continuous. And you need to consider each and every element of input array in some group.
Return true, if array can be split according to the above rules, else return false.
Note : You will get marks only if all the test cases are passed.
Input Format :
Line 1 : Integer N (size of array)
Line 2 : Array A elements (separated by space)
Output Format :
true or false
Constraints :
1 <= N <= 50
Sample Input 1 :
2
1 2
Sample Output 1 :
false
Sample Input 2 :
3
1 4 3
Sample Output 2 :
true
SOLUTION:-
#include <iostream>
using namespace std;
bool helper(int* arr, int n, int start, int lsum, int rsum)
{
// If reached the end
if (start == n)
return lsum == rsum;
// If divisible by 5 then add to the left sum
if (arr[start] % 5 == 0)
lsum += arr[start];
// If divisible by 3 but not by 5
// then add to the right sum
else if (arr[start] % 3 == 0)
rsum += arr[start];
// Else it can be added to any of the sub-arrays
else
// Try adding in both the sub-arrays (one by one)
// and check whether the condition satisfies
return helper(arr, n, start + 1, lsum + arr[start], rsum)
|| helper(arr, n, start + 1, lsum, rsum + arr[start]);
// For cases when element is multiple of 3 or 5.
return helper(arr, n, start + 1, lsum, rsum);
}
// Function to start the recursive calls
bool splitArray(int* arr, int n)
{
// Initially start, lsum and rsum will all be 0
return helper(arr, n, 0, 0, 0);
}
int main() {
int size;
cin>>size;
int *input=new int[1+size];
for(int i=0;i<size;i++)
cin>>input[i];
if(splitArray(input, size)) {
cout << "true" << endl;
}
else {
cout << "false" << endl;
}
return 0;
}