题意:上一篇博客~~
做法:
根据上一篇博客我们知道,dp[i][0]的值与dp[i-k][k]的最大值有关系。
dp[i][j]只与dp[i-1][j-1]有关系。
那么我们就建立一个数组fan[x],fan[x]代表到现在的i为止,dp[i-k][k]的最大值(i-k+k=x)。
每执行一分钟,就更新一次数组fan[x]。
两个做法的结果比较:
第一种做法,二维数组:
19828K | 172MS |
第二种做法,滚动数组:
248K | 188MS |
我们可以发现用时差不多,但是数组小了很多,所以说当n比较大的时候,可以选择用第二种做法。
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
#include<queue>
#include<stack>
#include<map>
#include<string>
#include<stdlib.h>
#define INF_MAX 0x7fffffff
#define INF 999999
#define max3(a,b,c) (max(a,b)>c?max(a,b):c)
#define min3(a,b,c) (min(a,b)<c?min(a,b):c)
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
struct node
{int u;int v;int w;bool friend operator < (node a, node b){return a.w < b.w;}
}edge[1001];
int gcd(int n,int m){if(n<m) swap(n,m);return n%m==0?m:gcd(m,n%m);}
int lcm(int n,int m){if(n<m) swap(n,m);return n/gcd(n,m)*m;}
int main()
{int n,m,i,j;int d[10001];scanf("%d%d",&n,&m);for(i=1;i<=n;i++){scanf("%d",&d[i]);}int dp[2][501];int fan[10001];memset(fan,0,sizeof(fan));for(j=0;j<=m;j++)dp[0][j]=0;for(i=1;i<=n;i++){dp[i%2][0]=max(fan[i],dp[(i-1)%2][0]);dp[i%2][1]=dp[(i-1)%2][0]+d[i];for(j=2;j<=m;j++){if(dp[(i-1)%2][j-1]!=0)dp[i%2][j]=dp[(i-1)%2][j-1]+d[i];else dp[i%2][j]=0;}for(j=1;j<=m;j++){if(i+j<=n&&dp[i%2][j])fan[i+j]=max(fan[i+j],dp[i%2][j]);}}printf("%d\n",dp[n%2][0]);return 0;
}