题目链接:
http://acm.xmu.edu.cn/JudgeOnline/problem.php?id=1560
题目大意:
给定n(n<=200)个任务及每个任务的耗时,问m(m<=200)时间能够获得的最大收益(收益为解决连续任务数的平方的和,具体例子见题目)
题目思路:
【动态规划】
设f[i][j]表示前i个任务,当前时间为j的最优值。
枚举第i个任务是前有几个和i连续的任务,状态转移方程很好推。
时间复杂度比O(n3)小很多,大概O(n2)级别。
//
//by coolxxx
//
#include<iostream>
#include<algorithm>
#include<string>
#include<iomanip>
#include<memory.h>
#include<time.h>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<stdbool.h>
#include<math.h>
#define min(a,b) ((a)<(b)?(a):(b))
#define max(a,b) ((a)>(b)?(a):(b))
#define abs(a) ((a)>0?(a):(-(a)))
#define lowbit(a) (a&(-a))
#define sqr(a) ((a)*(a))
#define swap(a,b) ((a)^=(b),(b)^=(a),(a)^=(b))
#define eps 1e-8
#define J 10000
#define MAX 0x7f7f7f7f
#define PI 3.1415926535897
#define N 204
using namespace std;
int n,m,lll,ans,cas;
int a[N],sum[N];
int f[N][N];
int main()
{
#ifndef ONLINE_JUDGE
// freopen("1.txt","r",stdin);
// freopen("2.txt","w",stdout);
#endif
int i,j,k;
scanf("%d",&cas);
while(cas--)
// while(~scanf("%s",s1))
// while(~scanf("%d",&n))
{
scanf("%d%d",&n,&m);
memset(sum,,sizeof(sum));
memset(f,,sizeof(f));
for(i=;i<=n;i++)
{
scanf("%d",&a[i]);
sum[i]=sum[i-]+a[i];
}
for(i=;i<=n;i++)f[i][a[i]]=;
for(i=;i<=n;i++)
{
for(j=;j<=m;j++)f[i][j]=max(f[i][j],f[i-][j]);
for(j=a[i];j<=m;j++)
{
f[i][j]=max(f[i][j],f[i][j-]);
for(k=i;k && j>=sum[i]-sum[k-];k--)
{
f[i][j]=max(f[i][j],f[k-][j-sum[i]+sum[k-]]+sqr(i-k+));
}
}
}
printf("%d\n",f[n][m]);
}
return ;
} /*
// //
*/