網易2017內推筆試題
阿新 • • 發佈:2019-01-01
有 n 個學生站成一排,每個學生有一個能力值,牛牛想從這 n 個學生中按照順序選取 k 名學生,要求相鄰兩個學生的位置編號的差不超過 d,使得這 k 個學生的能力值的乘積最大,你能返回最大的乘積嗎?
輸入描述:
每個輸入包含 1 個測試用例。每個測試資料的第一行包含一個整數 n (1 <= n <= 50),表示學生的個數,接下來的一行,包含 n 個整數,按順序表示每個學生的能力值 ai(-50 <= ai <= 50)。接下來的一行包含兩個整數,k 和 d (1 <= k <= 10, 1 <= d <= 50)。
輸出描述:
輸出一行表示最大的乘積。
輸入例子:
3 7 4 7 2 50
輸出例子:
49
動態規劃:考慮正負,需儲存乘積最大值和乘積最小值。 dp0[i][j]表示選中第i個數長度為j的乘積最大, dp1[i][j]表示選中第i個數長度為j的乘積最小。 長度為j的乘積由長度為j-1轉化而來,即dp[m][j-1]*a[i] m的取值為[max(1,i-d),i-1]。
#include <iostream> using namespace std; typedef long long ll; ll dp0[51][11]; ll dp1[51][11]; int a[51]; ll ans; int main() { int n,k,d; cin>>n; for(int i=1;i<=n;i++){ cin>>a[i]; } cin>>k>>d; for(int i=1;i<=n;i++){ dp0[i][1]=a[i]; dp1[i][1]=a[i]; for(int j=2;j<=k;j++){ for(int m=max(1,i-d);m<i;m++){ dp0[i][j]=max(dp0[i][j],max(a[i]*dp0[m][j-1],a[i]*dp1[m][j-1])); dp1[i][j]=min(dp1[i][j],min(a[i]*dp1[m][j-1],a[i]*dp0[m][j-1])); } } ans=max(ans,dp0[i][k]); cout<<ans<<endl; } cout<<ans; return 0; }
參考:http://www.nowcoder.com/test/question/done?tid=4826956&qid=45830