Ai
1 Star 2 Fork 5

LilithSangreal/LeetCode-Solutions

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
numbers-with-repeated-digits.py 1.00 KB
一键复制 编辑 原始数据 按行查看 历史
kamyu 提交于 2019-03-17 19:14 +08:00 . Create numbers-with-repeated-digits.py
# Time: O(logn)
# Space: O(logn)
class Solution(object):
def numDupDigitsAtMostN(self, N):
"""
:type N: int
:rtype: int
"""
def P(m, n):
result = 1
while n > 0:
result *= m-n+1
n -= 1
return result
digits = map(int, str(N+1))
result = 0
# Given 321
#
# 1. count numbers without repeated digits:
# - X
# - XX
for i in xrange(1, len(digits)):
result += P(9, 1)*P(9, i-1)
# 2. count numbers without repeated digits:
# - 1XX ~ 3XX
# - 30X ~ 32X
# - 320 ~ 321
prefix_set = set()
for i, x in enumerate(digits):
for y in xrange(1 if i == 0 else 0, x):
if y in prefix_set:
continue
result += P(9-i, len(digits)-i-1)
if x in prefix_set:
break
prefix_set.add(x)
return N-result
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/LilithSangreal/LeetCode-Solutions.git
git@gitee.com:LilithSangreal/LeetCode-Solutions.git
LilithSangreal
LeetCode-Solutions
LeetCode-Solutions
master

搜索帮助