Sum of first K natural numbers missing in given Array

Last Updated : 08 Apr, 2022

Given an array arr[] of size N and a number K, the task is to find the sum of the first K natural numbers that are not present in the given array.

Examples:

Input: arr[] = {2, 3, 4}, K = 3
Output: 12
Explanation: First 3 Missing numbers are: [1, 5, 6]

Input: arr[] = {-2, -3, 4}, K = 2
Output: 3
Explanation: Missing numbers are: [1 2]

Approach: The problem can be solved based on the following idea:

Store the positive numbers in the set and calculate the sums of the numbers present between two numbers of set using the formula for sum of all numbers from L to R (say X) = (R-1)*R/2 - (L+1)*L/2.

Check for all the intervals between two numbers and calculate their sum till K elements are not considered.

Follow the steps mentioned below to solve the problem:

Create a sorted set from array elements to remove all duplicates.
Then for each positive numbers from starting in set:
- Find count of numbers in between 2 positive numbers in the set
- If the count is smaller than K, find the sum of numbers in between those 2 numbers in the set and reduce K by count
- If the count is greater than K, find the sum of only K numbers from the previous positive number and reduce K to 0
Return the sum

Below is the implementation of the above approach:

C++

// C++ program for the above approach

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

// Function to find the sum between 0 to n
long long sumN(long long n)
{
    return n * (n + 1) / 2;
}

// Function to calculate the subsequence sum
long long printKMissingSum(vector<int>& nums,
                           int k)
{
    set<int> s(nums.begin(), nums.end());
    int a, b, prev, cnt;

    // Create one variable ans
    long long ans = 0;

    // Loop to calculate the sum
    for (auto itr = s.begin();
         itr != s.end() && k > 0; itr++) {
        b = *itr;
        if (b < 0) {
            a = 0;
            prev = 0;
            continue;
        }

        a = (itr == s.begin() ? 0 : prev);
        cnt = b - 1 - a;

        if (cnt <= k) {
            ans += sumN(b - 1) - sumN(a);
            k -= cnt;
        }
        else {
            ans += sumN(a + k) - sumN(a);
            k -= k;
        }
        prev = b;
    }
    if (k > 0) {
        ans += sumN(prev + k) - sumN(prev);
        k -= k;
    }
    return ans;
}

// Driver code
int main()
{
    vector<int> arr = { -2, -3, -4 };
    int K = 2;

    cout << printKMissingSum(arr, K);
    return 0;
}

Java

// Java program for the above approach
import java.io.*;
import java.util.*;

class GFG {
    // Function to find the sum between 0 to n
    public static long sumN(long n)
    {
        return n * (n + 1) / 2;
    }

    // Function to calculate the subsequence sum
    public static long printKMissingSum(int nums[], int k)
    {
        HashSet<Integer> s = new HashSet<Integer>();
        for (int i : nums)
            s.add(i);
        int a = 0, b = 0, prev = 0, cnt = 0;

        // Create one variable ans
        long ans = 0;

        // Loop to calculate the sum
        Iterator<Integer> it = s.iterator();
        int tempk = k;
        while (it.hasNext() && k > 0) {
            b = it.next();
            if (b < 0) {
                a = 0;
                prev = 0;
                continue;
            }

            a = (k == tempk ? 0 : prev);
            cnt = b - 1 - a;

            if (cnt <= k) {
                ans += sumN(b - 1) - sumN(a);
                k -= cnt;
            }
            else {
                ans += sumN(a + k) - sumN(a);
                k -= k;
            }
            prev = b;
        }
        if (k > 0) {
            ans += sumN(prev + k) - sumN(prev);
            k -= k;
        }
        return ans;
    }
    public static void main(String[] args)
    {
        int arr[] = { -2, -3, -4 };
        int K = 2;

        System.out.print(printKMissingSum(arr, K));
    }
}

// This code is contributed by Rohit Pradhan

Python3

# Python program for the above approach

# Function to find the sum between 0 to n
def sumN(n):
    return (n * (n + 1)) / 2

# Function to calculate the subsequence sum
def printKMissingSum(nums, k):
    s = set(nums)
    s = list(s)
    ans = 0
    itr = 0
    while itr < len(s) and k > 0:
        b = s[itr]
        if b < 0:
            a = 0
            prev = 0
            break

        a = 0 if itr == 0 else prev
        cnt = b - 1 - a

        if cnt <= k:
            ans += sumN(b - 1) - sumN(a)
            k -= cnt
        else:
            ans += sumN(a + k) - sumN(a)
            k -= k
        prev = b

        itr += 1
    if k > 0:
        ans += sumN(prev + k) - sumN(a)
        k -= k
    return int(ans)

# Driver code
nums = [-2, -3, -4]
k = 2
print(printKMissingSum(nums, k))

# This code is contributed by amnindersingh1414.

// C# program for the above approach
using System;
using System.Collections.Generic;

public class GFG{

  // Function to find the sum between 0 to n
  static long sumN(long n)
  {
    return n * (n + 1) / 2;
  }

  // Function to calculate the subsequence sum
  static long printKMissingSum(int[] nums, int k)
  {
    HashSet<int> s = new HashSet<int>();
    foreach (var i in nums)
      s.Add(i);
    int a = 0, b = 0, prev = 0, cnt = 0;

    // Create one variable ans
    long ans = 0;

    // Loop to calculate the sum
    HashSet<int>.Enumerator it = s.GetEnumerator();
    int tempk = k;
    while (it.MoveNext() && k > 0) {
      b = it.Current;
      if (b < 0) {
        a = 0;
        prev = 0;
        continue;
      }

      a = (k == tempk ? 0 : prev);
      cnt = b - 1 - a;

      if (cnt <= k) {
        ans += sumN(b - 1) - sumN(a);
        k -= cnt;
      }
      else {
        ans += sumN(a + k) - sumN(a);
        k -= k;
      }
      prev = b;
    }
    if (k > 0) {
      ans += sumN(prev + k) - sumN(prev);
      k -= k;
    }
    return ans;
  }

  static public void Main ()
  {
    int[] arr = { -2, -3, -4 };
    int K = 2;

    Console.Write(printKMissingSum(arr, K));
  }
}

// This code is contributed by hrithikgarg03188.

JavaScript

    <script>
        // JavaScript program for the above approach

        // Function to find the sum between 0 to n
        const sumN = (n) => {
            return n * (n + 1) / 2;
        }

        // Function to calculate the subsequence sum
        const printKMissingSum = (nums, k) => {
            let s = new Set();
            for (let itm in nums) s.add(nums[itm]);
            s = Array.from(s);
            let a, b, prev, cnt;

            // Create one variable ans
            let ans = 0;

            // Loop to calculate the sum
            for (let itr = 0; itr < s.length && k > 0; itr++) {
                b = s[itr];
                if (b < 0) {
                    a = 0;
                    prev = 0;
                    continue;
                }

                a = (itr == 0 ? 0 : prev);
                cnt = b - 1 - a;

                if (cnt <= k) {
                    ans += sumN(b - 1) - sumN(a);
                    k -= cnt;
                }
                else {
                    ans += sumN(a + k) - sumN(a);
                    k -= k;
                }
                prev = b;
            }
            if (k > 0) {
                ans += sumN(prev + k) - sumN(prev);
                k -= k;
            }
            return ans;
        }

        // Driver code

        let arr = [-2, -3, -4];
        let K = 2;

        document.write(printKMissingSum(arr, K));

    // This code is contributed by rakeshsahni

    </script>

Output

Time Complexity: O(N * logN)
Auxiliary Space: O(N)

Find if given number is sum of first n natural numbers

code_r

Improve

Article Tags :

Practice Tags :

Sum of first K natural numbers missing in given Array

Similar Reads

Thank You!

What kind of Experience do you want to share?