Equal sum array partition excluding a given element

Given an array arr[] and an index in it. Find whether the array arr[] can be partitioned into two disjoint sets such that sum of both the sets is equal and none of the sets includes arr[index]

Examples :

Input : arr[] = {2, 1, 3, 4}, 
        index = 2
Output : No
We need to exclude arr[2] which is 3.
Possible sets are : 
Set 1: (2, 1), Set 2: 4, sum = 3?4
Set 1: 2, Set 2: (4, 1), sum = 2?5
Set 1: 1, Set 2: (4, 2), sum = 1?6
Neither of the sums are equal.

Input : arr[] = {2, 5, 1, 4, 0}, 
         index = 4
Output : Yes
Set 1 : (2, 4), sum = 6
Set 2 : (5, 1), sum = 6

Approach: This problem is a variation of partition problem with an additional constraint that index cannot be included in either of the partitioned sets of array. 
First find sum S of array excluding the index-th element. If the sum is even then array can be partitioned otherwise not. If the sum is even then define two variables set1Sum and set2Sum to store the sum of two sets. 

It can be determined recursively that whether set1Sum is equal to set2Sum. Start from position 0 and traverse the array recursively. At every array position, there are two choices: either include current array element in set 1 or in set 2. Recursively call for both the conditions, by including current element in set 1 first then in set 2. If current position is the index to be excluded then recursively call for next position without updating any sum. When entire array is traversed then check for equality of both sets sum. If the sums are equal then result is found otherwise backtrack and check for other possibilities.

Steps to solve this problem:

1. Define a function isSubsetSumPoss that takes in an array arr, the number of elements in the array n, the sum of the first set set1Sum, the sum of the second set set2Sum, the index of the element that is to be excluded index, and the current position in the array pos.

2. If pos is equal to n, then the function returns whether set1Sum is equal to set2Sum.

3. If pos is equal to index, then the function calls itself with pos incremented by 1 without updating set1Sum or set2Sum.

4. If pos is not equal to index, then the function calls itself twice: once with set1Sum incremented by the value of arr[pos] and once with set2Sum incremented by the value of arr[pos].

5. Define a function canPartition that takes in an array arr, the number of elements in the array n, and the index of the element that is to be excluded index.

6. Calculate the sum of the entire array excluding the element at position index.

7. If the sum is not even, then return false as it is not possible to partition the array into two equal sum sets.

8. If the sum is even, call the isSubsetSumPoss function with set1Sum and set2Sum initialized to 0, index equal to the passed in index, and pos equal to 0. Return the result of the isSubsetSumPoss function.

Implementation: 

C++

// C++ program to determine whether an array can be
// partitioned into two equal sum sets when an index
// is always excluded from both sets.
#include <bits/stdc++.h>
using namespace std;

// Utility function to partition array into two sets
// and check whether sum of both sets are equal or not.
bool isSubsetSumPoss(int arr[], int n, int set1Sum,
                    int set2Sum, int index, int pos)
{

    // If the entire array is traversed, then check
    // whether sum of both the sets are equal or not.
    if (pos == n)
        return (set1Sum == set2Sum);

    // If current position is the index to be excluded
    // then call the function for next position without
    // updating any sum.
    if (pos == index)
        isSubsetSumPoss(arr, n, set1Sum,
                 set2Sum, index, pos + 1);

    // Each element can be included either in
    // set 1 or in set 2. Call function for
    // both the cases.
    return isSubsetSumPoss(arr, n, set1Sum + arr[pos],
                           set2Sum, index, pos + 1)
           || isSubsetSumPoss(arr, n, set1Sum, set2Sum +
                           arr[pos], index, pos + 1);
}

// Function that calls the main utility
// function and returns whether array can
// be partitioned into two sets or not.
bool canPartition(int arr[], int n, int index)
{

    // Calculate sum of entire array
    // excluding position index.
    int sum = 0;

    for (int i = 0; i < n; i++) {
        if (i == index)
            continue;
        sum += arr[i];
    }

    // If sum is not even then array
    // cannot be partitioned into two
    // equal sum sets.
    if (sum % 2 != 0)
        return false;

    // If sum is even call utility function.
    return isSubsetSumPoss(arr, n, 0, 0,
                              index, 0);
}

int main()
{
    int arr[] = { 2, 5, 1, 4, 0 };
    int index = 4;
    int n = sizeof(arr) / sizeof(arr[0]);

    if (canPartition(arr, n, index))
        cout << "Yes";
    else
        cout << "No";
    return 0;
}

Java

// Java program to determine whether an array
// can be partitioned into two equal sum
// sets when an index is always excluded
// from both sets.
import java.io.*;
import java.util.*;

public class GFG {

    // Utility function to partition array
    // into two sets and check whether sum
    // of both sets are equal or not.
    static boolean isSubsetSumPoss(int []arr,
          int n, int set1Sum, int set2Sum,
                       int index, int pos)
    {

        // If the entire array is traversed,
        // then check whether sum of both
        // the sets are equal or not.
        if (pos == n)
            return (set1Sum == set2Sum);

        // If current position is the index
        // to be excluded then call the
        // function for next position without
        // updating any sum.
        if (pos == index)
            isSubsetSumPoss(arr, n, set1Sum,
                    set2Sum, index, pos + 1);

        // Each element can be included
        // either in set 1 or in set 2.
        // Call function for both the cases.
        return isSubsetSumPoss(arr, n, set1Sum
           + arr[pos], set2Sum, index, pos + 1)
            || isSubsetSumPoss(arr, n, set1Sum,
           set2Sum + arr[pos], index, pos + 1);
    }

    // Function that calls the main utility
    // function and returns whether array can
    // be partitioned into two sets or not.
    static boolean canPartition(int []arr, int n,
                                    int index)
    {

        // Calculate sum of entire array
        // excluding position index.
        int sum = 0;

        for (int i = 0; i < n; i++) {
            if (i == index)
                continue;
            sum += arr[i];
        }

        // If sum is not even then array
        // cannot be partitioned into two
        // equal sum sets.
        if (sum % 2 != 0)
            return false;

        // If sum is even call utility function.
        return isSubsetSumPoss(arr, n, 0, 0,
                                index, 0);
    }

    // Driver code
    public static void main(String args[])
    {
        int []arr = { 2, 5, 1, 4, 0 };
        int index = 4;
        int n = arr.length;

        if (canPartition(arr, n, index))
            System.out.print("Yes");
        else
            System.out.print("No");
    }
}

// This code is contributed by Manish Shaw
// (manishshaw1)

Python3

# Python3 program to determine whether an array can be
# partitioned into two equal sum sets when an index
# is always excluded from both sets.

# Utility function to partition array into two sets
# and check whether sum of both sets are equal or not.
def isSubsetSumPoss(arr, n, set1Sum, set2Sum, index, pos) :

    # If the entire array is traversed, then check
    # whether sum of both the sets are equal or not.
    if (pos == n) :
        return (set1Sum == set2Sum)

    # If current position is the index to be excluded
    # then call the function for next position without
    # updating any sum.
    if (pos == index) :
        isSubsetSumPoss(arr, n, set1Sum, set2Sum,
                                     index, pos + 1)

    # Each element can be included either in
    # set 1 or in set 2. Call function for
    # both the cases.
    return (isSubsetSumPoss(arr, n, set1Sum + arr[pos],
                               set2Sum, index, pos + 1)
                    or isSubsetSumPoss(arr, n, set1Sum,
                   set2Sum + arr[pos], index, pos + 1))

# Function that calls the main utility
# function and returns whether array can
# be partitioned into two sets or not.
def canPartition(arr, n, index) :

    # Calculate sum of entire array
    # excluding position index.
    sum = 0

    for i in range (0, n) :
        if (i == index) :
            continue
        sum += arr[i]

    # If sum is not even then array
    # cannot be partitioned into two
    # equal sum sets.
    if (sum % 2 != 0) :
        return false

    # If sum is even call utility function.
    return isSubsetSumPoss(arr, n, 0, 0, index, 0)

# Driver Code
arr = [ 2, 5, 1, 4, 0 ]
index = 4
n = len(arr)

if (canPartition(arr, n, index)) :
    print ("Yes")
else :
    print ("No")

# This code is contributed by Manish Shaw
# (manishshaw1)

C#

// C# program to determine whether an array
// can be partitioned into two equal sum
// sets when an index is always excluded
// from both sets.
using System;
using System.Collections.Generic;
using System.Linq;
using System.Collections;

class GFG {

    // Utility function to partition array
    // into two sets and check whether sum
    // of both sets are equal or not.
    static bool isSubsetSumPoss(int []arr,
          int n, int set1Sum, int set2Sum,
                       int index, int pos)
    {

        // If the entire array is traversed,
        // then check whether sum of both
        // the sets are equal or not.
        if (pos == n)
            return (set1Sum == set2Sum);

        // If current position is the index
        // to be excluded then call the
        // function for next position without
        // updating any sum.
        if (pos == index)
            isSubsetSumPoss(arr, n, set1Sum,
                    set2Sum, index, pos + 1);

        // Each element can be included
        // either in set 1 or in set 2.
        // Call function for both the cases.
        return isSubsetSumPoss(arr, n, set1Sum
           + arr[pos], set2Sum, index, pos + 1)
            || isSubsetSumPoss(arr, n, set1Sum,
           set2Sum + arr[pos], index, pos + 1);
    }

    // Function that calls the main utility
    // function and returns whether array can
    // be partitioned into two sets or not.
    static bool canPartition(int []arr, int n,
                                    int index)
    {

        // Calculate sum of entire array
        // excluding position index.
        int sum = 0;

        for (int i = 0; i < n; i++) {
            if (i == index)
                continue;
            sum += arr[i];
        }

        // If sum is not even then array
        // cannot be partitioned into two
        // equal sum sets.
        if (sum % 2 != 0)
            return false;

        // If sum is even call utility function.
        return isSubsetSumPoss(arr, n, 0, 0,
                                index, 0);
    }

    // Driver code
    public static void Main()
    {
        int []arr = { 2, 5, 1, 4, 0 };
        int index = 4;
        int n = arr.Length;

        if (canPartition(arr, n, index))
            Console.Write("Yes");
        else
            Console.Write("No");
    }
}

// This code is contributed by Manish Shaw
// (manishshaw1)

PHP

<?php
// PHP program to determine whether
// an array can be partitioned into
// two equal sum sets when an index
// is always excluded from both sets.

// Utility function to partition array
// into two sets and check whether sum
// of both sets are equal or not.
function isSubsetSumPoss($arr, $n, $set1Sum,
                         $set2Sum, $index, $pos)
{

    // If the entire array is traversed,
    // then check whether sum of both
    // the sets are equal or not.
    if ($pos == $n)
        return ($set1Sum == $set2Sum);

    // If current position is the index
    // to be excluded then call the
    // function for next position without
    // updating any sum.
    if ($pos == $index)
        isSubsetSumPoss($arr, $n, $set1Sum,
                        $set2Sum, $index,
                        $pos + 1);

    // Each element can be included
    // either in set 1 or in set 2.
    // Call function for both the cases.
    return isSubsetSumPoss($arr, $n, $set1Sum +
                           $arr[$pos], $set2Sum,
                           $index, $pos + 1) ||
           isSubsetSumPoss($arr, $n, $set1Sum,
                           $set2Sum + $arr[$pos],
                           $index, $pos + 1);
}

// Function that calls the main utility
// function and returns whether array can
// be partitioned into two sets or not.
function canPartition($arr, $n, $index)
{

    // Calculate sum of entire array
    // excluding position index.
    $sum = 0;

    for ($i = 0; $i < $n; $i++)
    {
        if ($i == $index)
            continue;
        $sum += $arr[$i];
    }

    // If sum is not even then array
    // cannot be partitioned into two
    // equal sum sets.
    if ($sum % 2 != 0)
        return false;

    // If sum is even call
    // utility function.
    return isSubsetSumPoss($arr, $n, 0,
                           0, $index, 0);
}

// Driver Code
$arr = array( 2, 5, 1, 4, 0 );
$index = 4;
$n = count($arr);

if (canPartition($arr, $n, $index))
    echo ("Yes");
else
    echo ("No");

// This code is contributed by
// Manish Shaw (manishshaw1)
?>

Javascript

<script>

// JavaScript program to determine whether an array
// can be partitioned into two equal sum
// sets when an index is always excluded

    // Utility function to partition array
    // into two sets and check whether sum
    // of both sets are equal or not.
    function isSubsetSumPoss(arr,
          n, set1Sum, set2Sum,
                       index, pos)
    {

        // If the entire array is traversed,
        // then check whether sum of both
        // the sets are equal or not.
        if (pos == n)
            return (set1Sum == set2Sum);

        // If current position is the index
        // to be excluded then call the
        // function for next position without
        // updating any sum.
        if (pos == index)
            isSubsetSumPoss(arr, n, set1Sum,
                    set2Sum, index, pos + 1);

        // Each element can be included
        // either in set 1 or in set 2.
        // Call function for both the cases.
        return isSubsetSumPoss(arr, n, set1Sum
           + arr[pos], set2Sum, index, pos + 1)
            || isSubsetSumPoss(arr, n, set1Sum,
           set2Sum + arr[pos], index, pos + 1);
    }

    // Function that calls the main utility
    // function and returns whether array can
    // be partitioned into two sets or not.
    function canPartition(arr, n, index)
    {

        // Calculate sum of entire array
        // excluding position index.
        let sum = 0;

        for (let i = 0; i < n; i++) {
            if (i == index)
                continue;
            sum += arr[i];
        }

        // If sum is not even then array
        // cannot be partitioned into two
        // equal sum sets.
        if (sum % 2 != 0)
            return false;

        // If sum is even call utility function.
        return isSubsetSumPoss(arr, n, 0, 0,
                                index, 0);
    }
// Driver code

        let arr = [ 2, 5, 1, 4, 0 ];
        let index = 4;
        let n = arr.length;

        if (canPartition(arr, n, index))
            document.write("Yes");
        else
            document.write("No");

</script>

Output

Yes

Time Complexity : O(2^n) ,The time complexity of this algorithm is O(2^n) where n is the size of the array. This is because the recursive function is called twice for each element, leading to an exponential time complexity.

Space complexity: O(n),The space complexity of this algorithm is O(n) where n is the size of the array. This is because the recursive call stack is of size n.

Exercise : Try to solve this problem iteratively.