网易2017笔试-合唱团

题目https://www.nowcoder.com/questionTerminal/661c49118ca241909add3a11c96408c8

题目重述

一排数字,可正可负;从前往后,挑出K个;可有间隔,不超过D;
K个数字,最大乘积,到底多少.

想法

不会...参考的.
如果这写数字全部都是正的或者是负的,那就比较好办了.问题是有正有负难度就比较大了.现在令两个dp数组出来,maxdp[k][i]表示挑出k个数字,且以第i个数字结尾的最大乘积,同理mindp[k][i]表示挑出k个数字,且以第i个数字结尾的最小乘积.
那么现在考虑maxdp[k+1][i] ,代表以第i个数字结尾,挑出k+1个数字的最大乘积,但是i位前面一个数字的位置是不确定的,这就需要一个前向遍历,令前向遍历的下标为j,则罗列出所有maxdp[k][j] * vec[i]的值挑选一个最大的就是maxdp[k+1][i]了,真的是这样吗? naive! 还需要考虑可能负数的情况,这部分实现参考代码部分,其实就是把mindp[k][j]*vec[i]情况考虑下然后比较谁大.mindp[k][?]在执行过程中的作用可以理解为为maxdp[k+1][?]产生铺垫,最终找出maxdp[K][0~(n-1)]的值就是最终答案(这里的大写的K代表输入K).

Code

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <vector>
using namespace std;
inline long long max(long long a,long long b){return (a>b?a:b);}
inline long long min(long long a,long long b){return (a>b?b:a);}
int main()
{
  int n,K,d;
  scanf("%d", &n);
  vector<long long int> vec;
  vec.resize(n+1);
  for(int i=0;i<n;i++) scanf("%lld", &vec[i]);
  scanf("%d %d",&K, &d);
  long long int ans = -1;
  long long maxdp[11][51];
  long long mindp[11][51];
  for(int i=1;i<=K;i++)
    for(int j=0;j<n;j++ ) maxdp[i][j]=mindp[i][j]=0;
  for(int i=0;i<n;i++){
    maxdp[1][i] = mindp[1][i] = vec[i];
    for(int k=2;k<=K;k++){
      // 前向遍历,同时注意条件 i-j<d
       for(int j=i-1; j>=0 && i-j<d; j--){
         maxdp[k][i] = max(max(maxdp[k-1][j]*vec[i],mindp[k-1][j]*vec[i]),maxdp[k][i]);
         mindp[k][i] = min(min(mindp[k-1][j]*vec[i],maxdp[k-1][j]*vec[i]),mindp[k][i]);
       }
    }
    ans = max(maxdp[K][i],ans);
  }
  printf("%lld\n",ans);
  return 0;
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 背景 一年多以前我在知乎上答了有关LeetCode的问题, 分享了一些自己做题目的经验。 张土汪:刷leetcod...
    土汪阅读 12,779评论 0 33
  • 欧拉旋转、四元数、矩阵旋转之间的差异 除了欧拉旋转以外,还有两种表示旋转的方式:矩阵旋转和四元数旋转。接下来我们比...
    AndrewFan阅读 2,611评论 0 3
  • 动态规划(Dynamic Programming) 本文包括: 动态规划定义 状态转移方程 动态规划算法步骤 最长...
    廖少少阅读 3,343评论 0 18
  • 纯净水、怪癖、茄子、胸卡,字数不超过300字 小赵是一家情趣用品店的老板,他有一个不为人知的怪癖——喜欢在顾客买单...
    MioH阅读 377评论 0 0
  • 寒冬衬容颜 白雪添新衣 风中舞飘逸 香甜入人心
    丁香夜语阅读 326评论 1 1