1. 程式人生 > >網易2017內推筆試題

網易2017內推筆試題

有 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