第三周算法题

[NOIP2001 普及组] 数的计算

题目描述

我们要求找出具有下列性质数的个数（包含输入的正整数 $n$）。

先输入一个正整数 $n$（$n\le 1000$），然后对此正整数按照如下方法进行处理：

1. 不作任何处理；

2. 在它的左边拼接一个正整数，但该正整数不能超过原数，或者是上一个被拼接的数的一半；

3. 加上数后，继续按此规则进行处理，直到不能再加正整数为止。

• 输入格式

一行，一个正整数 $n$（$n\le 1000$）。

• 输出格式

一个整数，表示具有该性质数的个数。

• 样例 #1

• 样例输入 #1

6

• 样例输出 #1

6

• 提示

【样例解释】

满足条件的数为：$6$，$16$，$26$，$126$，$36$，$136$。

【题目来源】

NOIP 2001 普及组第一题

代码

• 思路一：利用函数递归计算(-_- 超时，过了五个)

def find(n):
    if n == 1:
        return 1
    return 1 + sum([find(i) for i in range(1, n // 2 + 1)])

n = int(input())
print(find(n))

• 思路二：动态规划，dp[i] = 1 + sum(dp[:i//2+1]，第 i 个状态会与 i // 2 之前的状态相关

n = int(input())
dp = [0, 1]
for i in range(2, n+1):
    dp.append(1 + sum(dp[:i//2+1]))
print(dp[-1])

[NOIP2002 普及组] 选数

题目描述

已知 $n$ 个整数 $x_1,x_2,\cdots ,x_n$，以及 $1$ 个整数 $k$（$k<n$）。从 $n$ 个整数中任选 $k$ 个整数相加，可分别得到一系列的和。例如当 $n=4$，$k=3$，$4$ 个整数分别为 $3,7,12,19$ 时，可得全部的组合与它们的和为：

$3+7+12=22$

$3+7+19=29$

$7+12+19=38$

$3+12+19=34$

现在，要求你计算出和为素数共有多少种。

例如上例，只有一种的和为素数：$3+7+19=29$。

• 输入格式

第一行两个空格隔开的整数 $n,k$（$1\le n\le 20$，$k<n$）。

第二行 $n$ 个整数，分别为 $x_1,x_2,\cdots ,x_n$（$1\le x_i\le 5\times 10^6$）。

• 输出格式

输出一个整数，表示种类数。

• 样例 #1

• 样例输入 #1

4 3
3 7 12 19

• 样例输出 #1

1

• 提示

【题目来源】

NOIP 2002 普及组第二题

代码

from itertools import combinations

def judge(num):
    for i in range(2, int(pow(num, 0.5) + 1)):
        if num % i == 0:
            return False
    return True

n, k = map(int, input().split())
lst = list(map(int, input().split()))

combs = combinations(lst, k)  # 求组合，返回一个迭代器

cnt = 0
for comb in combs:
    Sum = sum(comb)
    if judge(Sum):
        cnt += 1

print(cnt)

Function

题目描述

对于一个递归函数 $w(a,b,c)$

• 如果 $a\le 0$ 或 $b\le 0$ 或 $c\le 0$ 就返回值$ 1$。
• 如果 $a>20$ 或 $b>20$ 或 $c>20$ 就返回 $w(20,20,20)$
• 如果 $a<b$ 并且 $b<c$ 就返回$ w(a,b,c-1)+w(a,b-1,c-1)-w(a,b-1,c)$。
• 其它的情况就返回 $w(a-1,b,c)+w(a-1,b-1,c)+w(a-1,b,c-1)-w(a-1,b-1,c-1)$

这是个简单的递归函数，但实现起来可能会有些问题。当 $a,b,c$ 均为 $15$ 时，调用的次数将非常的多。你要想个办法才行。

注意：例如 $w(30,-1,0)$ 又满足条件 $1$ 又满足条件 $2$，请按照最上面的条件来算，答案为 $1$。

• 输入格式

会有若干行。

并以 $-1,-1,-1$ 结束。

保证输入的数在 $[-9223372036854775808,9223372036854775807]$ 之间，并且是整数。

• 输出格式

输出若干行，每一行格式：

w(a, b, c) = ans

注意空格。

• 样例 #1

• 样例输入 #1

1 1 1
2 2 2
-1 -1 -1

• 样例输出 #1

w(1, 1, 1) = 2
w(2, 2, 2) = 4

代码

思路一：递归，不出意外全部超时

def func(a, b, c):
    if a <= 0 or b <= 0 or c <= 0:
        return 1
    elif a > 20 or b > 20 or c > 20:
        return func(20, 20, 20)
    elif a < b and b < c:
        return func(a, b, c-1) + func(a, b-1, c-1) - func(a, b-1, c)
    else:
        return func(a-1, b, c) + func(a-1, b-1, c) + func(a-1, b, c-1) - func(a-1, b-1, c-1)

a, b, c = map(int, input().split())
while a != -1 and b != -1 and c != -1:
    print("w({}, {}, {}) = {}".format(a, b, c, func(a, b, c)))
    a, b, c = map(int, input().split())

思路二：记忆化搜索

def func(a, b, c):
    global dic
    # 记忆化存储
    if a <= 0 or b <= 0 or c <= 0:
        if dic.get((a, b, c), 0) == 0:
            dic[(a, b, c)] = 1
        return dic[(a, b, c)]

    elif a > 20 or b > 20 or c > 20:
        # 记忆化存储
        if dic.get((a, b, c), 0) == 0:
            if dic.get((20, 20, 20), 0) == 0:
                dic[(20, 20, 20)] = func(20, 20, 20)
            dic[(a, b, c)] = dic[(20, 20, 20)]

        return dic[(a, b, c)]

    elif a < b and b < c:
        # 记忆化存储
        if dic.get((a, b, c-1), 0) == 0:
            dic[(a, b, c-1)] = func(a, b, c-1)
        if dic.get((a, b-1, c-1), 0) == 0:
            dic[(a, b-1, c-1)] = func(a, b-1, c-1)
        if dic.get((a, b-1, c), 0) == 0:
            dic[(a, b-1, c)] = func(a, b-1, c)

        return dic[(a, b, c-1)] + dic[(a, b-1, c-1)] - dic[(a, b-1, c)]

    else:
        # 记忆化存储
        if dic.get((a-1, b, c), 0) == 0:
            dic[(a-1, b, c)] = func(a-1, b, c)
        if dic.get((a-1, b-1, c), 0) == 0:
            dic[(a-1, b-1, c)] = func(a-1, b-1, c)
        if dic.get((a-1, b, c-1), 0) == 0:
            dic[(a-1, b, c-1)] = func(a-1, b, c-1)
        if dic.get((a-1, b-1, c-1), 0) == 0:
            dic[(a-1, b-1, c-1)] = func(a-1, b-1, c-1)


        return dic[(a-1, b, c)] + dic[(a-1, b-1, c)] + dic[(a-1, b, c-1)] - dic[(a-1, b-1, c-1)]

a, b, c = map(int, input().split())
dic = {}
while a != -1 or b != -1 or c != -1:
    print("w({}, {}, {}) = {}".format(a, b, c, func(a, b, c)))
    a, b, c = map(int, input().split())

【XR-3】等差数列

题目描述

小 X 给了你一个等差数列的前两项以及项数，请你求出这个等差数列各项之和。

等差数列：对于一个 $n$ 项数列 $a$，如果满足对于任意 $i\in [1,n)$，有 $a_{i+1}-a_i=d$，其中 $d$ 为定值，则称这个数列为一个等差数列。

• 输入格式

一行 $3$ 个整数 $a_1,a_2,n$，表示等差数列的第 $1,2$ 项以及项数。

数据范围：

• $|a_1|,|a_2|\le 10^6$。

• $3\le n\le 10^6$。

• 输出格式

一行一个整数，表示答案。

• 样例 #1

• 样例输入 #1

1 2 3

• 样例输出 #1

6

• 样例 #2

• 样例输入 #2

-5 -10 5

• 样例输出 #2

-75

• 提示

【样例 $1$ 说明】

这个等差数列为 1 2 3，其各项之和为 $6$。

代码

a1, a2, n = map(int, input().split())
d = a2 - a1
Sum = (a1 + a1 + (n-1)*d) * n // 2  # 等差数列求和公式
print(Sum)