python编程------迅雷2019秋招在线笔试题

1.勾股数，是由三个正整数组成的数组；能符合勾股定理 a*a + b*b = c*c ， (a, b, c) 的正整数解。如果 (a, b, c) 是勾股数，它们的正整数倍数，也是勾股数。如果 (a, b, c) 互质，它们就称为素勾股数。给定正整数N，计算出小于或等于N的素勾股数个数。(0 < a <= b <= c <= N)

示例：

输入
10

输出
1

我写的代码如下，望大佬们批评指正：

num = int(input('num:'))
res = 0
n = 0
M = [i for i in range(1,num+1)]
N = [i**2 for i in range(1,num+1)]
for i in N:
    for j in N[n:]:
        if i + j in N:
            a = M[N.index(i)]
            b = M[N.index(j)]
            c = M[N.index(i+j)]
            for k in range(2,a+1):
                if a%k == 0 and b%k == 0 and c%k == 0:
                    break
            else:
                res += 1
    n += 1
print(res)

（第二题题干是这样，后面不记得了，动态规划这一块不是很熟练，当时没写出来，估计凉了）
2.有红黑两种颜色的方块积木，红色代表正数A，黑色代表负数B。选出17块积木排成一排，使得任意相邻7块积木之和都小于0。如何挑选才能使17块积木之和最大，最大值是多少？

---