Welcome to Subscribe On Youtube

3032. Count Numbers With Unique Digits II

Description

Given two positive integers a and b, return the count of numbers having unique digits in the range [a, b] (inclusive).

 

Example 1:

Input: a = 1, b = 20
Output: 19
Explanation: All the numbers in the range [1, 20] have unique digits except 11. Hence, the answer is 19.

Example 2:

Input: a = 9, b = 19
Output: 10
Explanation: All the numbers in the range [9, 19] have unique digits except 11. Hence, the answer is 10. 

Example 3:

Input: a = 80, b = 120
Output: 27
Explanation: There are 41 numbers in the range [80, 120], 27 of which have unique digits.

 

Constraints:

  • 1 <= a <= b <= 1000

Solutions

Solution 1: State Compression + Digit DP

The problem asks to count how many numbers in the range $[a, b]$ have unique digits. We can solve this problem using state compression and digit DP.

We can use a function $f(n)$ to count how many numbers in the range $[1, n]$ have unique digits. Then the answer is $f(b) - f(a - 1)$.

In addition, we can use a binary number to record the digits that have appeared in the number. For example, if the digits $1, 3, 5$ have appeared in the number, we can use $10101$ to represent this state.

Next, we use memoization search to implement digit DP. We search from the starting point to the bottom layer to get the number of schemes, return the answer layer by layer and accumulate it, and finally get the final answer from the search starting point.

The basic steps are as follows:

  1. We convert the number $n$ into a string $num$, where $num[0]$ is the highest digit and $num[len - 1]$ is the lowest digit.
  2. Based on the problem information, we design a function $dfs(pos, mask, limit)$, where $pos$ represents the current processing position, $mask$ represents the digits that have appeared in the current number, and $limit$ represents whether there is a limit at the current position. If $limit$ is true, then the digit at the current position cannot exceed $num[pos]$.

The answer is $dfs(0, 0, true)$.

The time complexity is $O(m \times 2^{10} \times 10)$, and the space complexity is $O(m \times 2^{10})$. Where $m$ is the number of digits in $b$.

Solution 2

This implementation uses set. It traverses the relevant values and updates its state as each value is processed. A keyed container records values that must be found or updated efficiently. After all required states have been considered, the maintained result is returned.

  • class Solution {
        private String num;
        private Integer[][] f;
    
        public int numberCount(int a, int b) {
            num = String.valueOf(a - 1);
            f = new Integer[num.length()][1 << 10];
            int x = dfs(0, 0, true);
            num = String.valueOf(b);
            f = new Integer[num.length()][1 << 10];
            int y = dfs(0, 0, true);
            return y - x;
        }
    
        private int dfs(int pos, int mask, boolean limit) {
            if (pos >= num.length()) {
                return mask > 0 ? 1 : 0;
            }
            if (!limit && f[pos][mask] != null) {
                return f[pos][mask];
            }
            int up = limit ? num.charAt(pos) - '0' : 9;
            int ans = 0;
            for (int i = 0; i <= up; ++i) {
                if ((mask >> i & 1) == 1) {
                    continue;
                }
                int nxt = mask == 0 && i == 0 ? 0 : mask | 1 << i;
                ans += dfs(pos + 1, nxt, limit && i == up);
            }
            if (!limit) {
                f[pos][mask] = ans;
            }
            return ans;
        }
    }
    
    
    // Solution 2
    class Solution {
        public int numberCount(int a, int b) {
            int res = 0;
            for (int i = a; i <= b; ++i) {
                if (isValid(i)) {
                    ++res;
                }
            }
            return res;
        }
        private boolean isValid(int n) {
            boolean[] present = new boolean[10];
            Arrays.fill(present, false);
            while (n > 0) {
                int dig = n % 10;
                if (present[dig]) {
                    return false;
                }
                present[dig] = true;
                n /= 10;
            }
            return true;
        }
    }
    
    
  • class Solution {
    public:
        int numberCount(int a, int b) {
            string num = to_string(b);
            int f[num.size()][1 << 10];
            memset(f, -1, sizeof(f));
            function<int(int, int, bool)> dfs = [&](int pos, int mask, bool limit) {
                if (pos >= num.size()) {
                    return mask ? 1 : 0;
                }
                if (!limit && f[pos][mask] != -1) {
                    return f[pos][mask];
                }
                int up = limit ? num[pos] - '0' : 9;
                int ans = 0;
                for (int i = 0; i <= up; ++i) {
                    if (mask >> i & 1) {
                        continue;
                    }
                    int nxt = mask == 0 && i == 0 ? 0 : mask | 1 << i;
                    ans += dfs(pos + 1, nxt, limit && i == up);
                }
                if (!limit) {
                    f[pos][mask] = ans;
                }
                return ans;
            };
    
            int y = dfs(0, 0, true);
            num = to_string(a - 1);
            memset(f, -1, sizeof(f));
            int x = dfs(0, 0, true);
            return y - x;
        }
    };
    
    
    // Solution 2
    class Solution {
    public:
        bool isvalid(int n) {
            vector<bool> present(10, false);
            while (n) {
                const int dig = n % 10;
                if (present[dig])
                    return false;
                present[dig] = true;
                n /= 10;
            }
            return true;
        }
        int numberCount(int a, int b) {
            int res = 0;
            for (int i = a; i <= b; ++i) {
                if (isvalid(i)) {
                    ++res;
                }
            }
            return res;
        }
    };
    
    
  • class Solution:
        def numberCount(self, a: int, b: int) -> int:
            @cache
            def dfs(pos: int, mask: int, limit: bool) -> int:
                if pos >= len(num):
                    return 1 if mask else 0
                up = int(num[pos]) if limit else 9
                ans = 0
                for i in range(up + 1):
                    if mask >> i & 1:
                        continue
                    nxt = 0 if mask == 0 and i == 0 else mask | 1 << i
                    ans += dfs(pos + 1, nxt, limit and i == up)
                return ans
    
            num = str(a - 1)
            x = dfs(0, 0, True)
            dfs.cache_clear()
            num = str(b)
            y = dfs(0, 0, True)
            return y - x
    
    
    # Solution 2
    class Solution:
        def numberCount(self, a: int, b: int) -> int:
            return sum(len(set(str(num))) == len(str(num)) for num in range(a, b + 1))
    
    
  • func numberCount(a int, b int) int {
    	num := strconv.Itoa(b)
    	f := make([][1 << 10]int, len(num))
    	for i := range f {
    		for j := range f[i] {
    			f[i][j] = -1
    		}
    	}
    	var dfs func(pos, mask int, limit bool) int
    	dfs = func(pos, mask int, limit bool) int {
    		if pos >= len(num) {
    			if mask != 0 {
    				return 1
    			}
    			return 0
    		}
    		if !limit && f[pos][mask] != -1 {
    			return f[pos][mask]
    		}
    		up := 9
    		if limit {
    			up = int(num[pos] - '0')
    		}
    		ans := 0
    		for i := 0; i <= up; i++ {
    			if mask>>i&1 == 1 {
    				continue
    			}
    			nxt := mask | 1<<i
    			if mask == 0 && i == 0 {
    				nxt = 0
    			}
    			ans += dfs(pos+1, nxt, limit && i == up)
    		}
    		if !limit {
    			f[pos][mask] = ans
    		}
    		return ans
    	}
    	y := dfs(0, 0, true)
    	num = strconv.Itoa(a - 1)
    	for i := range f {
    		for j := range f[i] {
    			f[i][j] = -1
    		}
    	}
    	x := dfs(0, 0, true)
    	return y - x
    }
    
    
    // Solution 2
    func numberCount(a int, b int) int {
    	count := 0
    	for num := a; num <= b; num++ {
    		if hasUniqueDigits(num) {
    			count++
    		}
    	}
    	return count
    }
    func hasUniqueDigits(num int) bool {
    	digits := strconv.Itoa(num)
    	seen := make(map[rune]bool)
    	for _, digit := range digits {
    		if seen[digit] {
    			return false
    		}
    		seen[digit] = true
    	}
    	return true
    }
    
    
  • function numberCount(a: number, b: number): number {
        let num: string = b.toString();
        const f: number[][] = Array(num.length)
            .fill(0)
            .map(() => Array(1 << 10).fill(-1));
        const dfs: (pos: number, mask: number, limit: boolean) => number = (pos, mask, limit) => {
            if (pos >= num.length) {
                return mask ? 1 : 0;
            }
            if (!limit && f[pos][mask] !== -1) {
                return f[pos][mask];
            }
            const up: number = limit ? +num[pos] : 9;
            let ans: number = 0;
            for (let i = 0; i <= up; i++) {
                if ((mask >> i) & 1) {
                    continue;
                }
                let nxt: number = mask | (1 << i);
                if (mask === 0 && i === 0) {
                    nxt = 0;
                }
                ans += dfs(pos + 1, nxt, limit && i === up);
            }
            if (!limit) {
                f[pos][mask] = ans;
            }
            return ans;
        };
    
        const y: number = dfs(0, 0, true);
        num = (a - 1).toString();
        f.forEach(v => v.fill(-1));
        const x: number = dfs(0, 0, true);
        return y - x;
    }
    
    
    // Solution 2
    function numberCount(a: number, b: number): number {
        let count: number = 0;
        for (let num = a; num <= b; num++) {
            if (hasUniqueDigits(num)) {
                count++;
            }
        }
        return count;
    }
    function hasUniqueDigits(num: number): boolean {
        const digits: Set<string> = new Set(num.toString().split(''));
        return digits.size === num.toString().length;
    }
    
    
  • class Solution {
        public int numberCount(int a, int b) {
            int res = 0;
            for (int i = a; i <= b; ++i) {
                if (isValid(i)) {
                    ++res;
                }
            }
            return res;
        }
        private boolean isValid(int n) {
            boolean[] present = new boolean[10];
            Arrays.fill(present, false);
            while (n > 0) {
                int dig = n % 10;
                if (present[dig]) {
                    return false;
                }
                present[dig] = true;
                n /= 10;
            }
            return true;
        }
    }
    
    
  • class Solution {
    public:
        bool isvalid(int n) {
            vector<bool> present(10, false);
            while (n) {
                const int dig = n % 10;
                if (present[dig])
                    return false;
                present[dig] = true;
                n /= 10;
            }
            return true;
        }
        int numberCount(int a, int b) {
            int res = 0;
            for (int i = a; i <= b; ++i) {
                if (isvalid(i)) {
                    ++res;
                }
            }
            return res;
        }
    };
    
    
  • class Solution:
        def numberCount(self, a: int, b: int) -> int:
            return sum(len(set(str(num))) == len(str(num)) for num in range(a, b + 1))
    
    
  • func numberCount(a int, b int) int {
    	count := 0
    	for num := a; num <= b; num++ {
    		if hasUniqueDigits(num) {
    			count++
    		}
    	}
    	return count
    }
    func hasUniqueDigits(num int) bool {
    	digits := strconv.Itoa(num)
    	seen := make(map[rune]bool)
    	for _, digit := range digits {
    		if seen[digit] {
    			return false
    		}
    		seen[digit] = true
    	}
    	return true
    }
    
    
  • function numberCount(a: number, b: number): number {
        let count: number = 0;
        for (let num = a; num <= b; num++) {
            if (hasUniqueDigits(num)) {
                count++;
            }
        }
        return count;
    }
    function hasUniqueDigits(num: number): boolean {
        const digits: Set<string> = new Set(num.toString().split(''));
        return digits.size === num.toString().length;
    }
    
    

All Problems

All Solutions