合唱團(網易2017校招)
阿新 • • 發佈:2019-02-08
時間限制:1秒 空間限制:32768K 熱度指數:49379本題知識點: 動態規劃 演算法知識視訊講解
題目描述
有 n 個學生站成一排,每個學生有一個能力值,牛牛想從這 n 個學生中按照順序選取 k 名學生,要求相鄰兩個學生的位置編號的差不超過 d,使得這 k 個學生的能力值的乘積最大,你能返回最大的乘積嗎?輸入描述:
每個輸入包含 1 個測試用例。每個測試資料的第一行包含一個整數 n (1 <= n <= 50),表示學生的個數,接下來的一行,包含 n 個整數,按順序表示每個學生的能力值 ai(-50 <= ai <= 50)。接下來的一行包含兩個整數,k 和 d (1 <= k <= 10, 1 <= d <= 50)。
輸出描述:
輸出一行表示最大的乘積。示例1
輸入
3 7 4 7 2 50
輸出
49
本題特殊的地方在於需要在動態規劃裡儲存兩個值 最大值和最小值
因為有可能負負得正,最小的負數乘一個負數變成最大正值
//#include "stdafx.h" #include<cstdio> #include<iostream> #include<algorithm> #include<cstring> using namespace std; const int maxn=55; const int inf=0x7fffffff; typedef long long ll; int n; ll a[maxn]; int k,d; ll dp1[55][12]; ll dp2[55][12]; ll ans=-50^10; ll getmax(ll a,ll b,ll c){ ll d=max(b,c); return max(a,d); } ll getmin(ll a,ll b,ll c){ ll d=min(b,c); return min(a,d); } int main(){ //freopen("c://jin.txt","r",stdin); cin>>n; for(int i=1;i<=n;i++) cin>>a[i]; cin>>k>>d; fill(dp1[0],dp1[0]+55*12,1); fill(dp2[0],dp2[0]+55*12,1); /* for(int i=0;i<=k;i++) {dp1[0][i]=a[0]; dp2[0][i]=a[0]; }*/ for(int i=1;i<=n;i++) for(int w=k;w>=1;w--) {for(int j=1;j<=d;j++) {if(i-j>=0){ dp1[i][w]=getmax(dp2[i-j][w-1]*a[i],dp1[i-j][w-1]*a[i],dp1[i][w]); dp2[i][w]=getmin(dp1[i-j][w-1]*a[i],dp2[i-j][w-1]*a[i],dp2[i][w]); } ans=getmax(ans,dp1[i][w],dp2[i][w]); } } cout<<ans<<endl; // freopen("CON","r",stdin); // system("pause"); return 0; }