拼多多2021笔试真题--python编程两道

题目来源牛客网

1.

多多的数字组合

多多君最近在研究某种数字组合：
定义为：每个数字的十进制表示中(0~9)，每个数位各不相同且各个数位之和等于N。
满足条件的数字可能很多，找到其中的最小值即可。

数据范围： 1≤n≤1000 1≤n≤1000 
进阶：空间复杂度 O(1) O(1)  ，时间复杂度 O(n) O(n) 

时间限制：C/C++ 1秒，其他语言2秒

空间限制：C/C++ 256M，其他语言512M

输入描述：

共一行，一个正整数N，如题意所示，表示组合中数字不同数位之和。
（1 <= N <= 1,000）

输出描述：共一行，一个整数，表示该组合中的最小值。 如果组合中没有任何符合条件的数字，那么输出-1即可。

示例1

输入例子：

5

输出例子：

5

例子说明：符合条件的数字有：5，14，23，32，41其中最小值为5

示例2

输入例子：

12

输出例子：

39

例子说明：

示例3

输入例子：

50
输出例子：

-1

例子说明：没有符合条件的数字 (Ｔ▽Ｔ)

python代码

def find_min_number(n):
    if n > 45:  # 如果n大于45，那么没有满足条件的数字
        return -1

    # 从个位开始，填充数位
    result = []
    used_digits = [False] * 10  # 标记每个数位是否已被使用

    for i in range(9, -1, -1):  # 从9到0
        while n >= i and not used_digits[i]:  # 如果还有剩余的数位之和，且该数位未使用
            used_digits[i] = True
            n -= i
            result.append(str(i))  # 将该数位加入结果

        if n == 0:
            break

    # 如果n不为0，说明没有足够的数位组成这个数
    if n != 0:
        return -1

    # 将结果逆序并转化为整数
    return int(''.join(reversed(result)))

# 输入处理
if __name__ == "__main__":
    n = int(input())
    print(find_min_number(n))

多多路上从左到右有N棵树（编号1～N），其中第i个颗树有和谐值Ai。多多鸡认为，如果一段连续的树，它们的和谐值之和可以被M整除，那么这个区间整体看起来就是和谐的。现在多多鸡想请你帮忙计算一下，满足和谐条件的区间的数量。

时间限制：C/C++ 1秒，其他语言2秒

空间限制：C/C++ 256M，其他语言512M

输入描述：

第一行，有2个整数N和M，表示树的数量以及计算和谐值的参数。

（ 1 <= N <= 100,000, 1 <= M <= 100  ）

第二行，有N个整数Ai, 分别表示第i个颗树的和谐值。

（ 0 <= Ai <= 1,000,000,000 ）

输出描述：

共1行，每行1个整数，表示满足整体是和谐的区间的数量。

示例1

输入例子：

5 2
1 2 3 4 5
输出例子：
6
例子说明：
长度为1: [2], [4]
长度为2: 无
长度为3: [1,2,3], [3,4,5]
长度为4: [1,2,3,4], [2,3,4,5]
长度为5: 无
共6个区间的和谐值之和可以被2整除。

python代码：

def count_harmonious_subarrays(N, M, A):
    from collections import defaultdict

    # 初始化余数计数器，余数0初始化为1，因为数组空时，和为0
    remainder_counts = defaultdict(int)
    remainder_counts[0] = 1

    # 前缀和初始化为0
    prefix_sum = 0

    # 和谐区间的数量
    count = 0

    # 遍历数组A
    for value in A:
        # 更新前缀和
        prefix_sum += value

        # 计算前缀和对M取模的余数
        remainder = prefix_sum % M

        # 更新和谐区间的数量
        count += remainder_counts[remainder]

        # 更新余数计数器
        remainder_counts[remainder] += 1

    return count

# 读取输入
N, M = map(int, input().split())
A = list(map(int, input().split()))

# 调用函数并输出结果
print(count_harmonious_subarrays(N, M, A))