Home » Interview coding problems/challenges

Combinational sum problem with repetition of digits

Combinational sum problem with repetition of digits: Here, we are going to learn to make some combination of the numbers whose sum equals to a given number using backtracking.
Submitted by Souvik Saha, on February 07, 2020

Description:

This is a standard interview problem to make some combination of the numbers whose sum equals to a given number using backtracking.

Problem statement:

Given a set of positive numbers and a number, your task is to find out the combinations of the numbers from the set whose summation equals to the given number. You can use the numbers repeatedly to make the combination.

    Input:
    Test case T
    T no. of N values and corresponding N positive numbers and the Number.
    
    E.g.
    3
    
    3
    2 3 5
    12
    
    3
    2 7 5
    14
    
    4
    1 2 3 4
    5
    
    Constrains:
    1 <= T <= 500
    1 <= N <= 20
    1 <= A[i] <= 9
    1 <= Number<= 50
    
    Output:
    Print all the combination which summation equals to the given number.

Example

    Input:
    N=3
    Set[ ]=2 3 5
    Number=12

    Output:
    2 2 2 2 2 2
    2 2 2 3 3
    2 2 3 5
    2 5 5
    3 3 3 3

Explanation with example:

Let there is a set S of positive numbers N and a positive number.

For pre-requisite, we are recommending you to go to our article combinational sum problem.

Making some combinations repeatedly using a number in such a way that the summation of that combination results that given number is a problem of combination and we will solve this problem using backtracking approach.

    Let, 
    f(i) = function to insert the isth number into the combinational subset

In this case we will consider two cases to solve the problem,

1. We will consider i^th element into the part of our combination subset until the sum greater the given number (f(i)).
2. We will not consider ith element into the part of our combination subset (not f(i)).

And every time we will check the current sum with the number. Each of the time we will count the number of occurrence and the also the combinations.

    Let,
    f(i) = function to insert the ith number into the combinational subset

For the input:

    S[] = {2, 3, 5}
    Number = 12

Here from every (not f(2)) there will be two edges (f(3)) and (not f(3)) again from (not f(3)) there will be ( f(5)) and (not f(5)). Here in this case we will discard that edges which have a current sum greater than the given number and make a count to those numbers which are equal to the given number.

C++ implementation:

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

void combination(int* arr, int n, int num, int pos, int curr_sum, vector<vector<int> >& v, vector<int> vi, set<vector<int> >& s)
{
    if (num < curr_sum)
        return;
    if (num == curr_sum && s.find(vi) == s.end()) {
        s.insert(vi);
        v.push_back(vi);
        return;
    }
    //go for the next elements for combination
    for (int i = pos; i < n; i++) {
        if (curr_sum + arr[i] <= num) {
            vi.push_back(arr[i]);
            combination(arr, n, num, i, curr_sum + arr[i], v, vi, s);
            vi.pop_back();
        }
    }
}

//print the vector
void print(vector<vector<int> > v)
{
    for (int i = 0; i < v.size(); i++) {
        for (int j = 0; j < v[i].size(); j++) {
            cout << v[i][j] << " ";
        }
        cout << endl;
    }
}

void combinational_sum(int* arr, int n, int num)
{
    vector<vector<int> > v;
    vector<int> vi;
    int pos = 0;
    int curr_sum = 0;
    set<vector<int> > s;
    combination(arr, n, num, pos, curr_sum, v, vi, s);
    print(v);
}

int main()
{
    int t;
    cout << "Test Case : ";
    cin >> t;
    while (t--) {
        int n, num;
        cout << "Enter the value of N : ";
        cin >> n;
        int arr[n];
        cout << "Enter the values : ";
        for (int i = 0; i < n; i++) {
            cin >> arr[i];
        }
        sort(arr, arr + n);
        cout << "Enter the number : ";
        cin >> num;
        combinational_sum(arr, n, num);
    }
    return 0;
}

Output

Test Case : 3
Enter the value of N : 3
Enter the values : 2 3 5
Enter the number : 12
2 2 2 2 2 2
2 2 2 3 3
2 2 3 5
2 5 5
3 3 3 3
Enter the value of N : 3
Enter the values : 2 7 5
Enter the number : 14
2 2 2 2 2 2 2
2 2 5 5
2 5 7
7 7
Enter the value of N : 4
Enter the values : 1 2 3 4
Enter the number : 5
1 1 1 1 1
1 1 1 2
1 1 3
1 2 2
1 4
2 3