每日一题 合唱团（动规）顺时针打印矩阵

题目

有 n 个学生站成一排，每个学生有一个能力值，牛牛想从这 n 个学生中按照顺序选取 k 名学生，要求相邻两个学生的位置编号的差不超过 d，使得这 k 个学生的能力值的乘积最大，你能返回最大的乘积吗？

思路

链接：https://www.nowcoder.com/questionTerminal/661c49118ca241909add3a11c96408c8
来源：牛客网

链接：https://www.nowcoder.com/questionTerminal/661c49118ca241909add3a11c96408c8
来源：牛客网

乘积有正有负，负负得正，所以要维护两个dp数组，一个存储最大，一个存储最小。
定义max[k][i]表示当选中了k个学生，并且以第i个学生为结尾，所产生的最大乘积；
min[k][i]表示 当选中了k个学生，并且以第i个学生为结尾，所产生的最小乘积；
那么max[k+1][i+1]=max(max[k][i]*stu[i+1],max[k][i]*stu[i+1])，
即当选中了k个学生后，再选择第i+1编号学生，所产生的最大乘积；
然而，并不能保证上一次选择的就是第i个学生，所以要遍历子数组max[k]，
令j从i到1，并且j与i+1之间小于间隔D，遍历max[k][j]，以及min[k][j]；
同理min[k+1][i+1]=min(min[k][i]*stu[i+1],max[k][i]*stu[i+1])。
最后，遍历一遍max[K][i]求得最大值（i从1～N）。

代码

import java.util.*;
public class Main{
 
    public static void main(String[]args){
           Scanner sc=new Scanner( System.in);
        while(sc.hasNext()){
            int n=sc.nextInt();
            int []arr=new int[n+1];
          for (int i = 1; i <=n ; i++) {
                arr[i] = sc.nextInt();
            }
            int K=sc.nextInt();
            int d=sc.nextInt();
            long [][]max=new long[K+1][n+1];
            long [][]min=new long[K+1][n+1];
            long  res = Integer.MIN_VALUE; 
            for(int i=1;i<=n;i++){
                max[1][i]=arr[i];
                min[1][i]=arr[i];
                for(int k=2;k<=K;k++){
                    for(int j=i-1;j>0&&i-j<=d;j--){
                        min[k][i]=Math.min(min[k][i],Math.min(min[k-1][j]*arr[i],max[k-1][j]*arr[i]));
                        max[k][i]=Math.max(max[k][i],Math.max(max[k-1][j]*arr[i],min[k-1][j]*arr[i]));
                 }
                }
                res = Math.max(res ,max[K][i]);
            }
            System.out.println(res);
        }
    }
}

代码

public class Printer {
    public int[] clockwisePrint(int[][] mat, int n, int m) {
        // write code here
        int startX=0;
        int startY=0;
        int endX=n-1;
        int endY=m-1;
        int []arr=new int[m*n];
        int index=0;
        while(endX>=startX&&endY>=startY){
            for(int y=startY;y<=endY;y++){
                arr[index++]=mat[startX][y];
            }
            for(int i=startX+1;i<=endX;i++){
                arr[index++]=mat[i][endY];
            }
            if(endX>startX){
                for(int i=endY-1;i>=startY;i--){
                    arr[index++]=mat[endX][i];
                }
            }
            if(endY>startY){
                for(int i=endX-1;i>startX;i--){
                    arr[index++]=mat[i][startY];
                }
            }
            startX++;
            startY++;
            endX--;
            endY--;
        }
        return arr;
    }
    
}