当前位置: 代码迷 >> 综合 >> poj-3661-另一种做法(滚动数组)
  详细解决方案

poj-3661-另一种做法(滚动数组)

热度:4   发布时间:2023-12-19 11:18:49.0

题意:上一篇博客~~

做法:

根据上一篇博客我们知道,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;
}