HNUCM-OJ-2023年春季学期《算法分析与设计》练习9

目录

问题A:最大子段和

问题B:最大子段和升级版：

问题C:斜线最大最小值：

问题D:矩阵连乘问题-备忘录法求最优值

问题E:矩阵连乘问题-动态规划求最优值

问题F:矩阵连乘问题-构造最优解

问题G:石子合并问题

问题H:XP的矩阵

---

问题A:最大子段和

题目描述

给定n个整数（可能是负数）组成的序列a[1], a[2], a[3], …, a[n]，求该序列的子段和如a[i]+a[i+1]+…+a[j]的最大值。

输入

每组输入包括两行，第一行为序列长度n，第二行为序列。

输出

输出字段和的最大值。

样例输入 Copy

5
-1 0 1 2 3

样例输出 Copy

6

while True:
    n=int(input())
    a=list(map(int,input().split()))
    b=[0 for i in range(n)]
    b[0]=a[0]
    maxlen=b[0]
    for j in range(1,n):
        if b[j-1]>0:
            b[j]=b[j-1]+a[j]
        else:
            b[j]=a[j]
    print(max(b))

---

问题B:最大子段和升级版：

题目描述

使用动态规划算法求整数数组（可能包含负整数）的最大子段和，以及和最大子段的起始位置和结束位置：
例如：输入数组(6,-1,5,4,-7)，输出14, 1, 4，其中14表示最大子段和，1表示和最大的子段从第1个数字开始，4表示和最大的子段到第4个数字结束，即(6, -1 , 5, 4)。

输入

每组输入两行，第1行为数组中包含的整数个数n，第2行为n个整数（可能包含负整数），两两之间用空格隔开。

输出

输出最大子段和，以及和最大子段的起始位置和结束位置，两两之间用空格隔开。

样例输入 Copy

5
6 -1 5 4 -7

样例输出 Copy

14 1 4

while True:
    n=int(input())
    a=list(map(int,input().split()))
    b=[0 for i in range(n)]
    b[0]=a[0]
    maxlen=b[0]
    for j in range(1,n):
        if b[j-1]>0:
            b[j]=b[j-1]+a[j]
        else:
            b[j]=a[j]
    end=b.index(max(b))
    maxsum=max(b)
    t=maxsum
    for i in range(end,-1,-1):
        t-=a[i]
        if t==0:
            t=i
            break
    print(maxsum,t+1,end+1)

---

问题C:斜线最大最小值：

题目描述

求如图所示一个上三角矩阵中每一条斜线中的最大元素(L)和最小元素(S)。

输入

每组输入包括两部分，一部分为数字n，表示三角矩阵的行数。
第二部分即为三角矩阵。

输出

每一个对角线输出一行，每行包括Lx=Max, Sx=Min，其中x为斜线序号（序号从1开始），Max为该斜线上的最大值，Min为该斜线上的最小值。

样例输入 Copy

6
1 3 5 7 11 20
0 6 8 2 3 13
0 0 7 4 8 9
0 0 0 18 3 10
0 0 0 0 12 6
0 0 0 0 0 15

样例输出 Copy

L1=18, S1=1
L2=8, S2=3
L3=10, S3=2
L4=9, S4=3
L5=13, S5=11
L6=20, S6=20

---

def slove(n):
    global a
    for r in range(n):
        max=int(a[0][r])
        min=int(a[0][r])
        for i in range(0,n-r):
            j=int(r+i)
            if max<a[i][j]:
                max=a[i][j]
            if min>a[i][j]:
                min=a[i][j]
        print("L%d=%d, S%d=%d"%(r+1,max,r+1,min))
while True:
    n=int(input())
    a=[]
    for i in range(n):
        a.append(list(map(int,input().split())))
    slove(n)

问题D:矩阵连乘问题-备忘录法求最优值

题目描述

使用备忘录法求解矩阵连乘问题，输出最少乘法次数。

输入

每组数据包括两行，第一行为数组长度n，第二行为存储矩阵维数的一维数组。

输出

矩阵连乘最优计算次数。

样例输入 Copy

7
30 35 15 5 10 20 25

样例输出 Copy

15125

def slove(x,y):
    for r in range(2,y+1):
        for i in range(1,y-r+2):
            j=i+r-1
            dp[i][j]=dp[i+1][j]+p[i-1]*p[i]*p[j];
            for k in range(i+1,j):
                dp[i][j]=min(dp[i][j],dp[i][k]+dp[k+1][j]+p[i-1]*p[k]*p[j])
    return dp[x][y]
while True:
    try:
        n= int(input())
        p=[int(i) for i in input().split()]
        dp = [[0] * n for _ in range(n)]
        print(slove(1, n - 1))
    except:
        break

问题E:矩阵连乘问题-动态规划求最优值

题目描述

使用动态规划算法求解矩阵连乘问题，输出最少乘法次数。

输入

每组数据包括两行，第一行为数组长度n，第二行为存储矩阵维数的一维数组。

输出

矩阵连乘最优计算次数。

样例输入 Copy

7
30 35 15 5 10 20 25

样例输出 Copy

15125

def slove(x,y):
    for r in range(2,y+1):
        for i in range(1,y-r+2):
            j=i+r-1
            dp[i][j]=dp[i+1][j]+p[i-1]*p[i]*p[j];
            for k in range(i+1,j):
                dp[i][j]=min(dp[i][j],dp[i][k]+dp[k+1][j]+p[i-1]*p[k]*p[j])
    return dp[x][y]
while True:
    try:
        n= int(input())
        p=[int(i) for i in input().split()]
        dp = [[0] * n for _ in range(n)]
        print(slove(1, n - 1))
    except:
        break

问题F:矩阵连乘问题-构造最优解

题目描述

使用动态规划算法求解矩阵连乘问题。

输入

每组数据包括两行，第一行为数组长度n，第二行为存储矩阵维数的一维数组。

输出

矩阵连乘最优计算次序。

样例输入 Copy

7
30 35 15 5 10 20 25

样例输出 Copy

A[2:2] * A[3:3]
A[1:1] * A[2:3]
A[4:4] * A[5:5]
A[4:5] * A[6:6]
A[1:3] * A[4:6]

def slove(x,y):
    for r in range(2,y+1):
        for i in range(1,y-r+2):
            j=i+r-1
            dp[i][j]=dp[i+1][j]+p[i-1]*p[i]*p[j];
            s[i][j]=i
            for k in range(i+1,j):
                t=dp[i][k]+dp[k+1][j]+p[i-1]*p[k]*p[j]
                if t<dp[i][j]:
                    dp[i][j]=t
                    s[i][j]=k
def tr(i,j):
    if i==j:
        return
    tr(i,s[i][j])
    tr(s[i][j]+1,j)
    m=s[i][j]+1
    print(f"A[{i}:{s[i][j]}] * A[{m}:{j}]")
while True:
    try:
        n= int(input())
        p=[int(i) for i in input().split()]
        dp = [[0] * n for _ in range(n)]
        s = [[0] * n for _ in range(n)]
        slove(1, n - 1)
        tr(1,n-1)
    except:
        break

问题G:石子合并问题

题目描述

在一条直线上有n堆石子，每堆有一定的数量，每次可以将两堆相邻的石子合并，合并后放在两堆的中间位置，合并的费用为两堆石子的总数。求把所有石子合并成一堆的最小花费。例如：输入{1，2，3，4，5}，输出33。【3+6+9+15=33】

输入

本题应该处理到文件尾，每组输入包括两行，第一行为石子堆的个数n，第二行则为每堆石子的个数。

输出

输出最小花费。

样例输入 Copy

5
1 2 3 4 5

样例输出 Copy

33

while True:
    try:
        n=int(input())
        a=[int(i) for i in input().split()]
        dp=[[float('inf')]*n for _ in range(n)]
        for i in range(n):
            dp[i][i]=0
        m=[0]*(n+1)
        for i in range(1,n+1):
            m[i]=m[i-1]+a[i-1]
        for r in range(2,n+1):
            for i in range(n-r+1):
                j=r+i-1
                for k in range(i,j):
                    dp[i][j]=min(dp[i][k]+dp[k+1][j],dp[i][j])
                dp[i][j]+=m[j+1]-m[i]
        print(dp[0][n-1])
    except:
        break
        

问题H:XP的矩阵

题目描述

XP学长觉得矩阵很美，虽然他也不知道美在哪里，因此，他决定挖掘一下矩阵的美。现在给定一个m行n列的矩阵，从左上角开始每次只能向右或者向下移动，最后到达右下角的位置，将路径上的所有数字累加起来作为这条路径的路径和。XP学长决定编写一个程序来求所有路径和中的最小路径和。例如，下面矩阵中的路径1-3-1-0-6-1-0是所有路径中路径和最小的，返回结果是12。
1 3 5 9
8 1 3 4
5 0 6 1
8 8 4 0
 

输入

输入包含多组测试用例，第一行输入一个T表示测试数据组数，(1<=T<=15)
接下来有T组数据，每组先输入两个整数M,N接下来输入M*N的矩阵(1<=N,M<=1000)，且最终结果在int范围内。

输出

输出路径和的最小值。

样例输入 Copy

1
4 4
1 3 5 9
8 1 3 4
5 0 6 1
8 8 4 0

样例输出 Copy

12

t=int(input())
for i in range(t):
    m,n=map(int,input().split())
    a=[[float('inf') for j in range(n)] for x in range(m)]
    dp=[[float('inf') for j in range(n)] for x in range(m)]
    for i in range(m):
        a[i]=list(map(int,input().split()))
    for i in range(m):
        for j in range(n):
            if i==0 and j==0:
                dp[i][j]=a[i][j]
                continue
            dp[i][j]=min(dp[i-1][j]+a[i][j],dp[i][j-1]+a[i][j])
    print(dp[m-1][n-1])