Divide 1 to n into two groups with minimum sum difference
Last Updated :
15 Mar, 2025
Given a positive integer n, such that n > 2. Your task is to divide the numbers from 1 to n in two groups, such that the absolute difference between the sum of the elements of the these two groups is minimum possible.
Examples:
Input: n = 5
Output: [ [ 2, 5 ], [ 1, 3, 4 ] ]
Explanation: The sum of first and second group are 7 and 8 respectively, and the difference is 8 - 7 = 1, which is the minimum possible. [ [ 1, 2, 5 ], [ 3, 4 ] ] is also the possible solution.
Input: n = 7
Output: [ [ 1, 6, 7 ], [ 2, 3, 4, 5 ] ]
Explanation: The sum of first and second group are 14 and 14 respectively, and the difference is 14 - 14 = 0, which is the minimum possible.
- It can be observed that numbers 1 to n, can always be divided in to two groups, such that the difference of sum of elements is either 0 or 1.
- The idea is to firstly compute the sum of elements from 1 to n (which is equal to n * (n + 1) / 2), then find the half of it.
[Naive Solution] - Dynamic Programming - O(n^3) Time
This problem can be reduced to the subset sum problem. We mainly need to find a subset of [1, 2, ... n] whose sum is closest to half of n*(n+1)/2. The time complexity of this solution would be O(n^3).
C++
#include <bits/stdc++.h>
using namespace std;
vector<vector<int>> minSubsetDiff(int n) {
int S = n * (n + 1) / 2;
int target = S / 2;
vector<bool> dp(target + 1, false);
vector<vector<int>> subset(target + 1);
dp[0] = true;
// Find all possible subset sums and store
// subsets also.
for (int i = 1; i <= n; i++) {
for (int j = target; j >= i; j--) {
if (dp[j - i]) {
dp[j] = true;
subset[j] = subset[j - i];
subset[j].push_back(i);
}
}
}
// Find the subset closest (smaller or equal)
// to target
int s1 = 0;
for (int i = target; i >= 0; i--) {
if (dp[i]) {
s1 = i;
break;
}
}
vector<int> subset1 = subset[s1];
unordered_set<int> set1(subset1.begin(), subset1.end());
vector<int> subset2;
for (int i = 1; i <= n; i++) {
if (!set1.count(i)) subset2.push_back(i);
}
return {subset1, subset2};
}
int main() {
int n = 4;
vector<vector<int>> res = minSubsetDiff(n);
cout << "Subset 1: ";
for (int x : res[0]) cout << x << " ";
cout << "\nSubset 2: ";
for (int x : res[1]) cout << x << " ";
cout << endl;
}
Java
import java.util.HashSet;
import java.util.Set;
public class SubsetDifference {
public static int[][] minSubsetDiff(int n) {
int S = n * (n + 1) / 2;
int target = S / 2;
boolean[] dp = new boolean[target + 1];
int[][] subset = new int[target + 1][];
dp[0] = true;
// Find all possible subset sums and store
for (int i = 1; i <= n; i++) {
for (int j = target; j >= i; j--) {
if (dp[j - i]) {
dp[j] = true;
subset[j] = new int[subset[j - i].length + 1];
System.arraycopy(subset[j - i], 0, subset[j], 0, subset[j - i].length);
subset[j][subset[j - i].length] = i;
}
}
}
// Find the subset closest (smaller or equal)
// to target
int s1 = 0;
for (int i = target; i >= 0; i--) {
if (dp[i]) {
s1 = i;
break;
}
}
int[] subset1 = subset[s1];
Set<Integer> set1 = new HashSet<>();
for (int num : subset1) {
set1.add(num);
}
int[] subset2 = new int[n - subset1.length];
int index = 0;
for (int i = 1; i <= n; i++) {
if (!set1.contains(i)) subset2[index++] = i;
}
return new int[][]{subset1, subset2};
}
public static void main(String[] args) {
int n = 4;
int[][] res = minSubsetDiff(n);
System.out.print("Subset 1: ");
for (int x : res[0]) System.out.print(x + " ");
System.out.println();
System.out.print("Subset 2: ");
for (int x : res[1]) System.out.print(x + " ");
System.out.println();
}
}
Python
def min_subset_diff(n):
S = n * (n + 1) // 2
target = S // 2
dp = [False] * (target + 1)
subset = [[] for _ in range(target + 1)]
dp[0] = True
# Find all possible subset sums and store
# subsets also.
for i in range(1, n + 1):
for j in range(target, i - 1, -1):
if dp[j - i]:
dp[j] = True
subset[j] = subset[j - i] + [i]
# Find the subset closest (smaller or equal)
# to target
s1 = 0
for i in range(target, -1, -1):
if dp[i]:
s1 = i
break
subset1 = subset[s1]
set1 = set(subset1)
subset2 = [i for i in range(1, n + 1) if i not in set1]
return [subset1, subset2]
n = 4
res = min_subset_diff(n)
print("Subset 1:", res[0])
print("Subset 2:", res[1])
C#
using System;
using System.Collections.Generic;
class Program {
public static List<List<int>> MinSubsetDiff(int n) {
int S = n * (n + 1) / 2;
int target = S / 2;
bool[] dp = new bool[target + 1];
List<int>[] subset = new List<int>[target + 1];
for (int i = 0; i <= target; i++) {
subset[i] = new List<int>();
}
dp[0] = true;
// Find all possible subset sums and store
// subsets also.
for (int i = 1; i <= n; i++) {
for (int j = target; j >= i; j--) {
if (dp[j - i]) {
dp[j] = true;
subset[j] = new List<int>(subset[j - i]);
subset[j].Add(i);
}
}
}
// Find the subset closest (smaller or equal)
// to target
int s1 = 0;
for (int i = target; i >= 0; i--) {
if (dp[i]) {
s1 = i;
break;
}
}
List<int> subset1 = subset[s1];
HashSet<int> set1 = new HashSet<int>(subset1);
List<int> subset2 = new List<int>();
for (int i = 1; i <= n; i++) {
if (!set1.Contains(i)) subset2.Add(i);
}
return new List<List<int>> { subset1, subset2 };
}
static void Main() {
int n = 4;
var res = MinSubsetDiff(n);
Console.WriteLine("Subset 1: " + string.Join(" ", res[0]));
Console.WriteLine("Subset 2: " + string.Join(" ", res[1]));
}
}
JavaScript
function minSubsetDiff(n) {
let S = n * (n + 1) / 2;
let target = Math.floor(S / 2);
let dp = Array(target + 1).fill(false);
let subset = Array.from({ length: target + 1 }, () => []);
dp[0] = true;
// Find all possible subset sums and store
// subsets also.
for (let i = 1; i <= n; i++) {
for (let j = target; j >= i; j--) {
if (dp[j - i]) {
dp[j] = true;
subset[j] = [...subset[j - i], i];
}
}
}
// Find the subset closest (smaller or equal)
// to target
let s1 = 0;
for (let i = target; i >= 0; i--) {
if (dp[i]) {
s1 = i;
break;
}
}
let subset1 = subset[s1];
let set1 = new Set(subset1);
let subset2 = [];
for (let i = 1; i <= n; i++) {
if (!set1.has(i)) subset2.push(i);
}
return [subset1, subset2];
}
let n = 4;
let res = minSubsetDiff(n);
console.log("Subset 1:", res[0]);
console.log("Subset 2:", res[1]);
OutputSubset 1: 1 4
Subset 2: 2 3
[Expected Solution[ - Greedy Approach - O(n) Time
We traverse numbers from n to 1. And we try adding every number to the first subset. If adding the number does not make the sum more than half, we add the number to first subset, else to the second.
We create two auxiliary arrays to store the elements of two groups, and initiate two counters to store the sum of elements of both the groups. Now start iterating from 1 to n, and insert the element in the first group if it does not exceed the first group's sum more than the calculated half sum, else insert the element in second group.
C++
#include <bits/stdc++.h>
using namespace std;
// Function to divide the elements from 1
// to n in two groups with minimum difference
vector<vector<int>> findGroup(int n) {
// to store the two groups
vector<vector<int>> res(2);
// to store the sum of both groups
int sum1 = 0, sum2 = 0;
// find half of sum of elements from 1 to n
int half = n * (n + 1) / 4;
for(int i = n; i >= 1; i--) {
// if adding the current element to first
// group doesn't exceed half, add it
if(sum1 + i <= half) {
res[0].push_back(i);
sum1 += i;
}
// else add it to the second group
else {
res[1].push_back(i);
sum2 += i;
}
}
return res;
}
int main() {
int n = 7;
vector<vector<int>> res = findGroup(n);
for(int i = 0; i < 2; i++) {
for(int j = 0; j < res[i].size(); j++) {
cout << res[i][j] << " ";
}
cout << endl;
}
return 0;
}
Java
// Function to divide the elements from 1
// to n in two groups with minimum difference
import java.util.*;
class GfG {
// Function to divide the elements from 1
// to n in two groups with minimum difference
static int[][] findGroup(int n) {
// to store the two groups
ArrayList<Integer> group1 = new ArrayList<>();
ArrayList<Integer> group2 = new ArrayList<>();
// to store the sum of both groups
int sum1 = 0, sum2 = 0;
// find half of sum of elements from 1 to n
int half = n * (n + 1) / 4;
for (int i = n; i >= 1; i--) {
// if adding the current element to first
// group doesn't exceed half, add it
if (sum1 + i <= half) {
group1.add(i);
sum1 += i;
}
// else add it to the second group
else {
group2.add(i);
sum2 += i;
}
}
// convert groups to int[] arrays
int[] arr1 = new int[group1.size()];
for (int i = 0; i < group1.size(); i++) {
arr1[i] = group1.get(i);
}
int[] arr2 = new int[group2.size()];
for (int i = 0; i < group2.size(); i++) {
arr2[i] = group2.get(i);
}
// create the result 2D array
int[][] res = new int[2][];
res[0] = arr1;
res[1] = arr2;
return res;
}
public static void main(String[] args) {
int n = 7;
int[][] res = findGroup(n);
for (int i = 0; i < 2; i++) {
for (int j = 0; j < res[i].length; j++) {
System.out.print(res[i][j] + " ");
}
System.out.println();
}
}
}
Python
# Function to divide the elements from 1
# to n in two groups with minimum difference.
def findGroup(n):
# to store the two groups
res = [[], []]
# to store the sum of both groups
sum1 = 0
sum2 = 0
# find half of sum of elements from 1 to n
half = n * (n + 1) // 4
for i in range(n, 0, -1):
# if adding the current element to first
# group doesn't exceed half, add it
if sum1 + i <= half:
res[0].append(i)
sum1 += i
# else add it to the second group
else:
res[1].append(i)
sum2 += i
return res
if __name__ == "__main__":
n = 7
res = findGroup(n)
for group in res:
for x in group:
print(x, end=" ")
print()
C#
// Function to divide the elements from 1
// to n in two groups with minimum difference
using System;
using System.Collections.Generic;
class GfG {
// Function to divide the elements from 1
// to n in two groups with minimum difference
static List<List<int>> findGroup(int n) {
// to store the two groups
List<int> group1 = new List<int>();
List<int> group2 = new List<int>();
// to store the sum of both groups
int sum1 = 0, sum2 = 0;
// find half of sum of elements from 1 to n
int half = n * (n + 1) / 4;
for (int i = n; i >= 1; i--) {
// if adding the current element to first
// group doesn't exceed half, add it
if (sum1 + i <= half) {
group1.Add(i);
sum1 += i;
}
// else add it to the second group
else {
group2.Add(i);
sum2 += i;
}
}
// to store the two groups
List<List<int>> res = new List<List<int>>();
res.Add(group1);
res.Add(group2);
return res;
}
static void Main() {
int n = 7;
List<List<int>> res = findGroup(n);
for (int i = 0; i < res.Count; i++) {
for (int j = 0; j < res[i].Count; j++) {
Console.Write(res[i][j] + " ");
}
Console.WriteLine();
}
}
}
JavaScript
// Function to divide the elements from 1
// to n in two groups with minimum difference.
function findGroup(n) {
// to store the two groups
let res = [[], []];
// to store the sum of both groups
let sum1 = 0, sum2 = 0;
// find half of sum of elements from 1 to n
let half = n * (n + 1) / 4;
for (let i = n; i >= 1; i--) {
// if adding the current element to first
// group doesn't exceed half, add it
if (sum1 + i <= half) {
res[0].push(i);
sum1 += i;
}
// else add it to the second group
else {
res[1].push(i);
sum2 += i;
}
}
return res;
}
let n = 7;
let res = findGroup(n);
for (let i = 0; i < res.length; i++) {
console.log(res[i].join(" "));
}
Time Complexity: O(n), as we are iterating from n to 1 only once.
Space Complexity: O(n), to store the elements from 1 to n in two groups.
Similar Reads
Greedy Algorithms Greedy algorithms are a class of algorithms that make locally optimal choices at each step with the hope of finding a global optimum solution. At every step of the algorithm, we make a choice that looks the best at the moment. To make the choice, we sometimes sort the array so that we can always get
3 min read
Greedy Algorithm Tutorial Greedy is an algorithmic paradigm that builds up a solution piece by piece, always choosing the next piece that offers the most obvious and immediate benefit. Greedy algorithms are used for optimization problems. An optimization problem can be solved using Greedy if the problem has the following pro
9 min read
Greedy Algorithms General Structure A greedy algorithm solves problems by making the best choice at each step. Instead of looking at all possible solutions, it focuses on the option that seems best right now.Example of Greedy Algorithm - Fractional KnapsackProblem structure:Most of the problems where greedy algorithms work follow thes
5 min read
Difference between Greedy Algorithm and Divide and Conquer Algorithm Greedy algorithm and divide and conquer algorithm are two common algorithmic paradigms used to solve problems. The main difference between them lies in their approach to solving problems. Greedy Algorithm:The greedy algorithm is an algorithmic paradigm that follows the problem-solving heuristic of m
3 min read
Greedy Approach vs Dynamic programming Greedy approach and Dynamic programming are two different algorithmic approaches that can be used to solve optimization problems. Here are the main differences between these two approaches: Greedy Approach:The greedy approach makes the best choice at each step with the hope of finding a global optim
2 min read
Comparison among Greedy, Divide and Conquer and Dynamic Programming algorithm Greedy algorithm, divide and conquer algorithm, and dynamic programming algorithm are three common algorithmic paradigms used to solve problems. Here's a comparison among these algorithms:Approach:Greedy algorithm: Makes locally optimal choices at each step with the hope of finding a global optimum.
4 min read
Standard Greedy algorithms
Activity Selection Problem | Greedy Algo-1Given n activities with start times in start[] and finish times in finish[], find the maximum number of activities a single person can perform without overlap. A person can only do one activity at a time. Examples: Input: start[] = [1, 3, 0, 5, 8, 5], finish[] = [2, 4, 6, 7, 9, 9]Output: 4Explanatio
13 min read
Job Sequencing ProblemGiven two arrays: deadline[] and profit[], where the index of deadline[] represents a job ID, and deadline[i] denotes the deadline for that job and profit[i] represents profit of doing ith job. Each job takes exactly one unit of time to complete, and only one job can be scheduled at a time. A job ea
13 min read
Huffman Coding | Greedy Algo-3Huffman coding is a lossless data compression algorithm. The idea is to assign variable-length codes to input characters, lengths of the assigned codes are based on the frequencies of corresponding characters. The variable-length codes assigned to input characters are Prefix Codes, means the codes (
12 min read
Huffman DecodingWe have discussed Huffman Encoding in a previous post. In this post, decoding is discussed. Examples: Input Data: AAAAAABCCCCCCDDEEEEEFrequencies: A: 6, B: 1, C: 6, D: 2, E: 5 Encoded Data: 0000000000001100101010101011111111010101010 Huffman Tree: '#' is the special character usedfor internal nodes
15 min read
Water Connection ProblemYou are given n houses in a colony, numbered from 1 to n, and p pipes connecting these houses. Each house has at most one outgoing pipe and at most one incoming pipe. Your goal is to install tanks and taps efficiently.A tank is installed at a house that has one outgoing pipe but no incoming pipe.A t
8 min read
Greedy Algorithm for Egyptian FractionEvery positive fraction can be represented as sum of unique unit fractions. A fraction is unit fraction if numerator is 1 and denominator is a positive integer, for example 1/3 is a unit fraction. Such a representation is called Egyptian Fraction as it was used by ancient Egyptians. Following are a
11 min read
Policemen catch thievesGiven an array arr, where each element represents either a policeman (P) or a thief (T). The objective is to determine the maximum number of thieves that can be caught under the following conditions:Each policeman (P) can catch only one thief (T).A policeman can only catch a thief if the distance be
12 min read
Fitting Shelves ProblemGiven length of wall w and shelves of two lengths m and n, find the number of each type of shelf to be used and the remaining empty space in the optimal solution so that the empty space is minimum. The larger of the two shelves is cheaper so it is preferred. However cost is secondary and first prior
9 min read
Assign Mice to HolesThere are N Mice and N holes are placed in a straight line. Each hole can accommodate only 1 mouse. A mouse can stay at his position, move one step right from x to x + 1, or move one step left from x to x -1. Any of these moves consumes 1 minute. Assign mice to holes so that the time when the last m
8 min read
Greedy algorithm on Array
Minimum product subset of an arrayINTRODUCTION: The minimum product subset of an array refers to a subset of elements from the array such that the product of the elements in the subset is minimized. To find the minimum product subset, various algorithms can be used, such as greedy algorithms, dynamic programming, and branch and boun
13 min read
Maximize array sum after K negations using SortingGiven an array of size n and an integer k. We must modify array k number of times. In each modification, we can replace any array element arr[i] by -arr[i]. The task is to perform this operation in such a way that after k operations, the sum of the array is maximum.Examples : Input : arr[] = [-2, 0,
10 min read
Minimum sum of product of two arraysFind the minimum sum of Products of two arrays of the same size, given that k modifications are allowed on the first array. In each modification, one array element of the first array can either be increased or decreased by 2.Examples: Input : a[] = {1, 2, -3} b[] = {-2, 3, -5} k = 5 Output : -31 Exp
14 min read
Minimum sum of absolute difference of pairs of two arraysGiven two arrays a[] and b[] of equal length n. The task is to pair each element of array a to an element in array b, such that sum S of absolute differences of all the pairs is minimum.Suppose, two elements a[i] and a[j] (i != j) of a are paired with elements b[p] and b[q] of b respectively, then p
7 min read
Minimum increment/decrement to make array non-IncreasingGiven an array a, your task is to convert it into a non-increasing form such that we can either increment or decrement the array value by 1 in the minimum changes possible. Examples : Input : a[] = {3, 1, 2, 1}Output : 1Explanation : We can convert the array into 3 1 1 1 by changing 3rd element of a
11 min read
Sorting array with reverse around middleConsider the given array arr[], we need to find if we can sort array with the given operation. The operation is We have to select a subarray from the given array such that the middle element(or elements (in case of even number of elements)) of subarray is also the middle element(or elements (in case
6 min read
Sum of Areas of Rectangles possible for an arrayGiven an array, the task is to compute the sum of all possible maximum area rectangles which can be formed from the array elements. Also, you can reduce the elements of the array by at most 1. Examples: Input: a = {10, 10, 10, 10, 11, 10, 11, 10} Output: 210 Explanation: We can form two rectangles o
13 min read
Largest lexicographic array with at-most K consecutive swapsGiven an array arr[], find the lexicographically largest array that can be obtained by performing at-most k consecutive swaps. Examples : Input : arr[] = {3, 5, 4, 1, 2} k = 3 Output : 5, 4, 3, 2, 1 Explanation : Array given : 3 5 4 1 2 After swap 1 : 5 3 4 1 2 After swap 2 : 5 4 3 1 2 After swap 3
9 min read
Partition into two subsets of lengths K and (N - k) such that the difference of sums is maximumGiven an array of non-negative integers of length N and an integer K. Partition the given array into two subsets of length K and N - K so that the difference between the sum of both subsets is maximum. Examples : Input : arr[] = {8, 4, 5, 2, 10} k = 2 Output : 17 Explanation : Here, we can make firs
7 min read
Greedy algorithm on Operating System
Program for First Fit algorithm in Memory ManagementPrerequisite : Partition Allocation MethodsIn the first fit, the partition is allocated which is first sufficient from the top of Main Memory.Example : Input : blockSize[] = {100, 500, 200, 300, 600}; processSize[] = {212, 417, 112, 426};Output:Process No. Process Size Block no. 1 212 2 2 417 5 3 11
8 min read
Program for Best Fit algorithm in Memory ManagementPrerequisite : Partition allocation methodsBest fit allocates the process to a partition which is the smallest sufficient partition among the free available partitions. Example: Input : blockSize[] = {100, 500, 200, 300, 600}; processSize[] = {212, 417, 112, 426}; Output: Process No. Process Size Bl
8 min read
Program for Worst Fit algorithm in Memory ManagementPrerequisite : Partition allocation methodsWorst Fit allocates a process to the partition which is largest sufficient among the freely available partitions available in the main memory. If a large process comes at a later stage, then memory will not have space to accommodate it. Example: Input : blo
8 min read
Program for Shortest Job First (or SJF) CPU Scheduling | Set 1 (Non- preemptive)The shortest job first (SJF) or shortest job next, is a scheduling policy that selects the waiting process with the smallest execution time to execute next. SJN, also known as Shortest Job Next (SJN), can be preemptive or non-preemptive. Â Characteristics of SJF Scheduling: Shortest Job first has th
13 min read
Job Scheduling with two jobs allowed at a timeGiven a 2d array jobs[][] of order n * 2, where each element jobs[i], contains two integers, representing the start and end time of the job. Your task is to check if it is possible to complete all the jobs, provided that two jobs can be done simultaneously at a particular moment. Note: If a job star
6 min read
Optimal Page Replacement AlgorithmIn operating systems, whenever a new page is referred and not present in memory, page fault occurs, and Operating System replaces one of the existing pages with newly needed page. Different page replacement algorithms suggest different ways to decide which page to replace. The target for all algorit
3 min read
Greedy algorithm on Graph