序列最大收益

题目描述
给定一个长度为 m 的整数序列 a1,a2,…,am。

序列中每个元素的值 ai 均满足 1≤ai≤n。

当一个值为 i 的元素和一个值为 j 的元素相邻时，可以产生的收益为 wi,j。

现在，我们可以从序列中删除最多 k 个元素，删除一些元素后，原本不相邻的元素可能会变得相邻。

序列的收益和为所有相邻元素对产生的收益之和，例如一个长度为 3 的整数序列 1,3,2 的收益和为 w1,3+w3,2。

请问，通过利用删除操作，能够得到的序列的最大收益和是多少？

输入格式
第一行包含三个整数 n,k,m。

第二行包含 m 个整数 a1,a2,…,am。

接下来 n 行，每行包含 n 个整数，其中第 i 行第 j 列的数表示 wi,j。

输出格式
输出序列的最大收益和。

数据范围
对于 30% 的数据，1≤n,k,m≤20。
对于 100% 的数据，1≤n,k,m≤200，0≤wi,j≤107，1≤ai≤n。
数据保证 wi,j=wj,i，wi,i=0。

样例
输入样例：
4 1 3
1 4 2
0 3 0 1
3 0 0 0
0 0 0 0
1 0 0 0
输出样例：
3
样例解释
初始序列收益和为 w1,4+w4,2=1+0=1。

删除中间的 4 后，序列 1,2 的收益和为 w1,2=3。

思路一： DP
回想下最长上升子序列的求法：

用 f[j] 表示前 j 个数，且以第 j 个数为结尾的最长上升子序列。
本题：
f[i][j] 表示考虑以第 i 个数为结尾，删除 j 个数的最大收益。

假设删除 j 个数后， 和 i 相邻的数是第 u 个数。

u 和 i 相邻, u + 1 到 i - 1 被删除，删除了 i - 1 - u 个数，还得在 1 到 u 中再删除 j - (i - 1 - u) 个数。

所以：f[i][j] = max (f[u, j - (i - 1 - u)] + w[a[u], a[i]]), //状态转移方程
其中：j - (i - 1 - u) >= 0 且 u <= i - 1 且 j - (i - 1 - u) <= u。

取值范围解释：

j - (i - 1 - u) >= 0：前 u 个数中，被删除的数的个数要大于等于 0 。

u <= i - 1 ： u 最大是 i - 1。

j - (i - 1 - u) <= u : 前 u 个数中，被删除的的数最多是 u 个。

时间复杂度 O(n^3)
参考文献
C++ 代码

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 240;
int n,m,k,res;
int a[N],w[N][N],f[N][N];
int main()
{
    cin>>n>>k>>m;
    for (int i = 1; i <= m; i ++ )cin>>a[i];
    for (int i = 1; i <= n; i ++ )
     for (int j = 1;j <= n; j ++ )
      cin>>w[i][j];
    f[1][0]=0;
    for (int i = 1; i <= m; i ++ )
      for (int j = 0; j <= k ; j ++ )
        for (int u = 1; u < i; u ++ )
            if(j>=i - u -1)
           f[i][j]=max(f[i][j],f[u][j - (i - u - 1)] + w[a[u]][a[i]]);
    for (int i = 0; i <= k; i ++ )
       res=max(res,f[m][i]);
    return cout<<res,0;
}

Java代码

import java.util.*;
import java.io.*;
public class Main{
    static int res;
    public static void main(String[] args){
        Scanner cin = new Scanner(System.in);
        int n=cin.nextInt(),k=cin.nextInt(),m=cin.nextInt();
        int a[]=new int [240];
        int f[][]=new int [240][240];
        int w[][]=new int [240][240];
        for(int i = 1; i <= m; i++ )a[i]=cin.nextInt();
        for(int i = 1; i<= n; i ++ )
           for(int j = 1; j<= n; j ++)
             w[i][j]=cin.nextInt();
        f[1][0]=0;
        for (int i = 1; i <= m; i ++ )
          for (int j = 0; j <= k ; j ++ )
            for (int u = 1; u < i; u ++ )
               if(j>=i - u -1)
               f[i][j]=Math.max(f[i][j],f[u][j - (i - u - 1)] + w[a[u]][a[i]]);
        for (int i = 0; i <= k; i ++ )
        res=Math.max(res,f[m][i]);
        System.out.println(res);
        return ;
    }
}

GO代码

package main
import "fmt"
const N=240 
var (
   a[N]int 
   w[N][N]int 
   f[N][N]int 
)
func main(){
    var n,k,m,res int
    fmt.Scanf("%d %d %d",&n,&k,&m)
    for i:= 1;i <= m; i ++ {
        fmt.Scanf("%d",&a[i])
    }
    for i:= 1; i <= n; i ++ {
        for j:= 1; j <= n; j ++ {
            fmt.Scanf("%d",&w[i][j])
        }
    }
    f[1][0]=0
    for i:= 1; i <=m; i ++ {
       for j:= 0; j<=k; j ++ {
          for u:= 1; u < i; u ++ {
              if j>= i - u - 1{
                  f[i][j] = max(f[i][j], f[u][j - (i - u - 1)] + w[a[u]][a[i]])
              }
          }
       }   
    }
    for i:= 0; i <= k; i ++ {
        res = max(res,f[m][i]) 
    }
    fmt.Println(res)
}
func max(a,b int)int{
    if a > b{
        return a
    }
    return b
}

思路二： DP
换一种思路：f[i][j]表示以i位结尾，最多删除j个数的最大收益
定义一个枚举变量p表示j
从i开始第一个未被删除的数的下标是i-p-1
剩下的最多还能删除j-p个数
权值就变成了w[a[i-p-1]][a[i]]
状态转移方程：f[i][j]=max(f[i][j],f[i-p-1][j-p]+w[a[i-p-1]a[i]]);
本题答案位f[m][k];

时间复杂度 O(n^3)
参考文献
C++ 代码

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1010;
int n,m,k;
int w[N][N],f[N][N],a[N];
int main()
{
    cin>>n>>k>>m;
    for (int i = 1; i <= m; i ++ )cin>>a[i];
    for (int i = 1; i <= n; i ++ )
      for (int j = 1; j <= n; j ++ )
         cin>>w[i][j];
    for (int i = 1; i <= m; i ++ )
      for (int j = 0; j <= k; j ++ )
         for(int p = 0; p <=j; p ++ )
           if(i-p-1>=0)
             f[i][j]=max(f[i][j],f[i-p-1][j-p]+w[a[i-p-1]][a[i]]);
    return cout<<f[m][k],0;
}

Java代码

import java.util.*;
import java.io.*;
public class Main{
    static int res;
    public static void main(String[] args){
        Scanner cin = new Scanner(System.in);
        int n=cin.nextInt(),k=cin.nextInt(),m=cin.nextInt();
        int a[]=new int [240];
        int f[][]=new int [240][240];
        int w[][]=new int [240][240];
        for(int i = 1; i <= m; i++ )a[i]=cin.nextInt();
        for(int i = 1; i<= n; i ++ )
           for(int j = 1; j<= n; j ++)
             w[i][j]=cin.nextInt();
        for (int i = 1; i <= m; i ++ )
          for (int j = 0; j <= k ; j ++ )
             for(int p = 0; p <=j; p ++ )
               if(i-p-1>=0)
             f[i][j]=Math.max(f[i][j],f[i-p-1][j-p]+w[a[i-p-1]][a[i]]);
        System.out.println(f[m][k]);
        return ;
    }
}

GO代码

package main
import "fmt"
const N=240 
var (
   a[N]int 
   w[N][N]int 
   f[N][N]int 
)
func main(){
    var n,k,m int
    fmt.Scanf("%d %d %d",&n,&k,&m)
    for i:= 1;i <= m; i ++ {
        fmt.Scanf("%d",&a[i])
    }
    for i:= 1; i <= n; i ++ {
        for j:= 1; j <= n; j ++ {
            fmt.Scanf("%d",&w[i][j])
        }
    }
    f[1][0]=0
    for i:= 1; i <=m; i ++ {
       for j:= 0; j<=k; j ++ {
          for p:= 0; p <= j; p ++ {
              if i - p - 1>= 0{
                   f[i][j]=max(f[i][j],f[i-p-1][j-p]+w[a[i-p-1]][a[i]])
              }
          }
       }   
    }
    fmt.Println(f[m][k])
}
func max(a,b int)int{
    if a > b{
        return a
    }
    return b
}

欢迎留言点赞