[洛谷]P1336 最佳课题选择 (#线性dp)

题目描述

Matrix67要在下个月交给老师n篇论文，论文的内容可以从m个课题中选择。由于课题数有限，Matrix67不得不重复选择一些课题。完成不同课题的论文所花的时间不同。具体地说，对于某个课题i，若Matrix67计划一共写x篇论文，则完成该课题的论文总共需要花费Ai*x^Bi个单位时间（系数Ai和指数Bi均为正整数）。给定与每一个课题相对应的Ai和Bi的值，请帮助Matrix67计算出如何选择论文的课题使得他可以花费最少的时间完成这n篇论文。

输入格式

第一行有两个用空格隔开的正整数n和m，分别代表需要完成的论文数和可供选择的课题数。

以下m行每行有两个用空格隔开的正整数。其中，第i行的两个数分别代表与第i个课题相对应的时间系数Ai和指数Bi。

输出格式

输出完成n篇论文所需要耗费的最少时间。

输入输出样例

输入 #1复制

10 3
2 1
1 2
2 1

输出 #1复制

19

说明/提示

【样例说明】

4篇论文选择课题一，5篇论文选择课题三，剩下一篇论文选择课题二，总耗时为2*4^1+1*1^2+2*5^1=8+1+10=19。可以证明，不存在更优的方案使耗时小于19。

【数据规模与约定】

对于30%的数据，n<=10,m<=5；

对于100%的数据，n<=200，m<=20，Ai<=100，Bi<=5。

---

思路

令dp[i][j]为表示前i个课题中完成j篇论文的最小耗时，其中k为对于课题i选择j个论文，因此k需要枚举(0<=k<=j)，则：

dp[i][j]=min(dp[i][j],dp[i-1][j-k]+a[i]×k^b[i])

根据设计的状态，答案为dp[m][n]。

#include <stdio.h>
#include <iostream>
#include <memory.h>
#include <math.h>
#define ll long long int
#define inf 2e9+7
using namespace std;
ll n,m,s,dp[21][201],a[21],b[21];
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	register ll i,j,k;
	cin>>n>>m;
	for(i=1;i<=m;i++)
	{
		cin>>a[i]>>b[i];
	}
	memset(dp,inf,sizeof(dp));
	for(i=0;i<=20;i++)
	{
		dp[i][0]=0;//完成论文数为0时值为0
	}
	for(i=1;i<=n;i++)//枚举论文 
	{
		for(j=1;j<=m;j++)//枚举课题 
		{
			for(k=0;k<=i;k++)//枚举课题j选择多少个论文 
			{
				dp[j][i]=min(dp[j][i],dp[j-1][i-k]+a[j]*(ll)pow((double)k,(double)b[j]));
			}
		}
	}
	cout<<dp[m][n]<<endl;
	return 0;
}