Open In App

Count unique numbers that can be generated from N by adding one and removing trailing zeros

Given a number N. Add one to the number in the first step and if the number has trailing zeros, remove all the trailing zeros in the second step. Continue the process for the next generated number. The task is to count the number of unique numbers that can be generated from these operations. 
Examples: 
 

Input: N = 5 
Output:
5 -> 6 -> 7 -> 8 -> 9 -> 1 -> 2 -> 3 -> 4 -> 5 (same sequence repeats) 
Note that 10 is not included as it contained trailing zero 
and removing the zero gave 1 as the next element.
Input: N = 28 
Output: 11 
 



Approach: The problem can be solved using recursion. Use a unordered_set to store all the unique numbers. In case a number is reached twice, we end the recursion as the same sequence will be repeated and we will not be getting any more unique numbers. Else insert the number to the set and in the first step increase the number by 1 and remove all trailing zeros in the next step if there are any.
Below is the implementation of the above approach: 
 




// C++ implementation of the approach
#include <bits/stdc++.h>
using namespace std;
 
// Function to count the unique numbers
void count_unique(unordered_set<int>& s, int n)
{
 
    // If the number has
    // already been visited
    if (s.count(n))
        return;
 
    // Insert the number to the set
    s.insert(n);
 
    // First step
    n += 1;
 
    // Second step
    // remove trailing zeros
    while (n % 10 == 0) {
        n = n / 10;
    }
 
    // Recur again for the new number
    count_unique(s, n);
}
 
// Driver code
int main()
{
    int n = 10;
    unordered_set<int> s;
    count_unique(s, n);
 
    cout << s.size();
 
    return 0;
}




// Java implementation of the approach
import java.util.*;
 
class GFG
{
 
// Function to count the unique numbers
static void count_unique(HashSet<Integer>s, int n)
{
 
    // If the number has
    // already been visited
    if (s.contains(n))
        return;
 
    // Insert the number to the set
    s.add(n);
 
    // First step
    n += 1;
 
    // Second step
    // remove trailing zeros
    while (n % 10 == 0)
    {
        n = n / 10;
    }
 
    // Recur again for the new number
    count_unique(s, n);
}
 
// Driver code
public static void main(String[] args)
{
    int n = 10;
    HashSet<Integer>s = new HashSet<>();
    count_unique(s, n);
    System.out.println(s.size());
}
}
 
// This code has been contributed by 29AjayKumar




# Python3 implementation of the approach
 
# Function to count the unique numbers
def count_unique(s, n) :
 
    # If the number has
    # already been visited
    if (s.count(n)) :
        return;
 
    # Insert the number to the set
    s.append(n);
 
    # First step
    n += 1;
 
    # Second step
    # remove trailing zeros
    while (n % 10 == 0) :
        n = n // 10;
 
    # Recur again for the new number
    count_unique(s, n);
 
 
# Driver code
if __name__ == "__main__" :
 
    n = 10
    s = []
     
    count_unique(s, n)
     
    print(len(s))
 
# This code is contributed by Ryuga




// C# implementation of the approach
using System;
using System.Collections.Generic;
 
class GFG
{
 
// Function to count the unique numbers
static void count_unique(HashSet<int>s, int n)
{
 
    // If the number has
    // already been visited
    if (s.Contains(n))
        return;
 
    // Insert the number to the set
    s.Add(n);
 
    // First step
    n += 1;
 
    // Second step
    // remove trailing zeros
    while (n % 10 == 0)
    {
        n = n / 10;
    }
 
    // Recur again for the new number
    count_unique(s, n);
}
 
// Driver code
public static void Main(String[] args)
{
    int n = 10;
    HashSet<int>s = new HashSet<int>();
    count_unique(s, n);
    Console.WriteLine(s.Count);
}
}
 
// This code contributed by Rajput-Ji




<script>
 
// JavaScript implementation of the approach
 
// Function to count the unique numbers
function count_unique(s,n)
{
    // If the number has
    // already been visited
    if (s.has(n))
        return;
   
    // Insert the number to the set
    s.add(n);
   
    // First step
    n += 1;
   
    // Second step
    // remove trailing zeros
    while (n % 10 == 0)
    {
        n = Math.floor(n / 10);
    }
   
    // Recur again for the new number
    count_unique(s, n);
}
 
// Driver code
let n = 10;
let s = new Set();
count_unique(s, n);
document.write(s.size);
 
 
// This code is contributed by rag2127
 
</script>

Output: 

19

 

Time Complexity: O(9*logN), as we are using recursion in the worst case we will call the function 9 times before we decrement N by floor division of 10, which will effectively cost logN time.

Auxiliary Space: O(9*logN), as we are using extra space for the set. 


Article Tags :