第一題:有 n 個學生站成一排,每個學生有一個能力值,牛牛想從這 n 個學生中按照順序選取 k 名學生,要求相鄰兩個學生的位置編號的差不超過 d,使得這 k 個學生的能力值的乘積最大,你能返回最大的乘積嗎?
采用了兩個矩陣mx,mn
mx[i][j]是從i個選出j個,並以i為結束,滿足相鄰位置不大於j的最大乘積
mn[i][j]是從i個選出j個,並以i為結束,滿足相鄰
#include<iostream> #include<cstdio> #include<algorithm> using namespace std; #define N 110 const long long INF=1e8; long long a[N],mx[N][N],mn[N][N];//事先聲明了數組,因為最后數可能會很大,所以定義的longlong類型 long long max(long long a,long long b) { return a>b?a:b; } long long min(long long a,long long b) { return a>b?b:a; } int main() { int n; scanf("%d",&n); for(int i=1;i<=n;i++)//這種輸入數組的方法 scanf("%lld",&a[i]); int k,d; scanf("%d%d",&k,&d); for(int i=1;i<=n;i++)//初始化mx,mn數組 { mx[i][0]=1; mn[i][0]=1; for(int j=1;j<=k;j++) { mx[i][j]=-INF; mn[i][j]=INF; } } long long mmx=-100,mnx=100;//mmx主要是用於當k=1時選出數組中最大的元素 long long ans=-INF; for(int i=1;i<=n;i++) { mmx=max(mmx,a[i]); mx[i][1]=a[i]; mn[i][1]=a[i]; if(k==1) ans=max(ans,mmx); } for(int i=1;i<=n;i++) { for(int j=2;j<=k;j++)//j=1已經討論過了 { for(int r=i-1;r>=max(1,i-d)&&r>=j-1;r--)//mx[r][j-1]表示從前r個中選出j-1個最大的乘積,所以r>=j-1;r只能最多回溯d個或者回溯到底 { mx[i][j]=max(mx[i][j],mx[r][j-1]*a[i]); mx[i][j]=max(mx[i][j],mn[r][j-1]*a[i]); mn[i][j]=min(mn[i][j],mx[r][j-1]*a[i]); mn[i][j]=min(mn[i][j],mn[r][j-1]*a[i]); if(j==k) ans=max(ans,mx[i][j]); } } } cout<<ans; }
位置不大於j的最小乘積
