codevs1166 矩阵取数游戏

题目描述 Description

【问题描述】
帅帅经常跟同学玩一个矩阵取数游戏：对于一个给定的n*m 的矩阵，矩阵中的每个元素aij均
为非负整数。游戏规则如下：
1. 每次取数时须从每行各取走一个元素，共n个。m次后取完矩阵所有元素；
2. 每次取走的各个元素只能是该元素所在行的行首或行尾；
3. 每次取数都有一个得分值，为每行取数的得分之和，每行取数的得分= 被取走的元素值*2^i，
其中i 表示第i 次取数（从1 开始编号）；
4. 游戏结束总得分为m次取数得分之和。
帅帅想请你帮忙写一个程序，对于任意矩阵，可以求出取数后的最大得分。

输入描述 Input Description

第1行为两个用空格隔开的整数n和m。
第2~n+1 行为n*m矩阵，其中每行有m个用单个空格隔开的非负整数。

输出描述 Output Description

输出 仅包含1 行，为一个整数，即输入矩阵取数后的最大得分。

样例输入 Sample Input

2 3
1 2 3
3 4 2

样例输出 Sample Output

82

数据范围及提示 Data Size & Hint

样例解释

第 1 次：第1 行取行首元素，第2 行取行尾元素，本次得分为1*2^1+2*2^1=6
第2 次：两行均取行首元素，本次得分为2*2^2+3*2^2=20
第3 次：得分为3*2^3+4*2^3=56。总得分为6+20+56=82

 

【限制】
60%的数据满足：1<=n, m<=30, 答案不超过10^16
100%的数据满足：1<=n, m<=80, 0<=aij<=1000

分析：

行与行间取数方案是相互独立的，只要每行进行一次dp再将各行结果相加即可。

对任意一行，设a[i]为该行第i个元素。

dp[i][j]表示该行剩余[i,j]列时已得到的最大得分。

方程为

dp[i][j] = max{ dp[i-1][j] + a[i-1]*2^(m-j+i-1),  dp[i][j+1] + a[j+1]*2^(m-j+i-1)  }

最后该行的最大得分是 max{ dp[i][i] + a[i]*2^m } ,其中 1<=i<=n

代码如下：

public class Test {

  public static void main(String[] args) {
    Scanner scanner =  new Scanner(System.in);
    System.out.println("请输入行数和列数（中间用空格隔开）：");
    String[] nm = scanner.nextLine().split(" ");
    int n = Integer.parseInt(nm[0]);
    int m = Integer.valueOf(nm[1]);
    System.out.println("请输入每行数字（数字中间用空格隔开）：");
    int[][] arr = new int[n+1][m+1];
    int[][] dp = new int[1000][1000];
    int sum = 0;
    for (int i = 1; i <= n; i++){
      String[] tmp = scanner.nextLine().split(" ");
      for (int j = 1; j <= tmp.length; j++){
        arr[i][j] = Integer.parseInt(tmp[j-1]);
      }
    }
    scanner.close();

    for (int i = 1; i <= n; i++){
      int ans = 0;
      for (int j = 1; j <= m; j++){

        for (int k = m; k >= j; k--){
          int t1 = 0;
          int t2 = 0;
          if (j > 1){
            t1 = dp[j-1][k] + (arr[i][j-1] << (m-k+j-1));
          }
          if (k < m){
            t2 = dp[j][k+1] + (arr[i][k+1] << (m-k+j-1));
          }
          if (t1 > t2){
            dp[j][k] = t1;
          } else {
            dp[j][k] = t2;
          }
        }
        int tmp = dp[j][j] + (arr[i][j] << m);
        if (ans < tmp){
          ans = tmp;
        }

      }

      sum += ans;
    }
    System.out.println(sum);
  }

}

或者使用高精度运算，使用BigInteger,替换一下：

public class Test2 {

  public static void main(String[] args) {
    Scanner scanner =  new Scanner(System.in);
    System.out.println("请输入行数和列数（中间用空格隔开）：");
    String[] nm = scanner.nextLine().split(" ");
    int n = Integer.parseInt(nm[0]);
    int m = Integer.valueOf(nm[1]);
    System.out.println("请输入每行数字（数字中间用空格隔开）：");
    int[][] arr = new int[n+1][m+1];
    BigInteger[][] dp = new BigInteger[1000][1000];
    BigInteger sum = BigInteger.valueOf(0);
    for (int i = 1; i <= n; i++){
      String[] tmp = scanner.nextLine().split(" ");
      for (int j = 1; j <= tmp.length; j++){
        arr[i][j] = Integer.parseInt(tmp[j-1]);
      }
    }
    scanner.close();

    for (int i = 1; i <= n; i++){
      BigInteger ans = BigInteger.valueOf(0);
      for (int j = 1; j <= m; j++){

        for (int k = m; k >= j; k--){
          BigInteger t1 = BigInteger.valueOf(0);
          BigInteger t2 = BigInteger.valueOf(0);
          if (j > 1){
            t1 = dp[j-1][k].add(BigInteger.valueOf((arr[i][j-1] << (m-k+j-1)))) ;
          }
          if (k < m){
            t2 = dp[j][k+1].add(BigInteger.valueOf((arr[i][k+1] << (m-k+j-1))));
          }
          if (t1.compareTo(t2) > 0){
            dp[j][k] = t1;
          } else {
            dp[j][k] = t2;
          }
        }
        BigInteger tmp = dp[j][j].add(BigInteger.valueOf((arr[i][j] << m)));
        if (ans.compareTo(tmp) < 0){
          ans = tmp;
        }

      }

      sum = sum.add(ans);
    }
    System.out.println(sum);
  }

}

题目来源：http://codevs.cn/problem/1166/