There are n courses in the course selection system of Marjar University. The i-th course is described by two values: happiness Hi and credit Ci. If a student selectsm courses x1, x2, ..., xm, then his comfort level of the semester can be defined as follows:
Edward, a student in Marjar University, wants to select some courses (also he can select no courses, then his comfort level is 0) to maximize his comfort level. Can you help him?
There are multiple test cases. The first line of input contains an integer T, indicating the number of test cases. For each test case:
The first line contains a integer n (1 ≤ n ≤ 500) -- the number of cources.
Each of the next n lines contains two integers Hi and Ci (1 ≤ Hi ≤ 10000, 1 ≤ Ci≤ 100).
It is guaranteed that the sum of all n does not exceed 5000.
We kindly remind you that this problem contains large I/O file, so it's recommended to use a faster I/O method. For example, you can use scanf/printf instead of cin/cout in C++.
For each case, you should output one integer denoting the maximum comfort.
2 3 10 1 5 1 2 10 2 1 10 2 10
191 0
For the first case, Edward should select the first and second courses.
For the second case, Edward should select no courses.
思路 :
这个题的公式涉及两个变量,令 x = sum(Hi),y = sum(Ci) 就是求 f = x^2 - x*y - y^2 的最大值,如果把y当做常熟对称轴就是 x = y/2 显然如果 x < y ,则 f < 0,所以x越大越好。即,在某一个y下,它构成的x越大越好。因此就可以这样考虑了,每个课程,挑或者不挑,使得所有可能的y得到的x最大,这就转化成背包问题。只不过这个背包是个必须要装满的背包罢了。
#include <cstdio>
#include <map>
#include <algorithm>
#include <cmath>
#include <iostream>
#include <cstring>
using namespace std;
typedef long long ll;
int x[100000];
int y[100000];
ll dp[100000];
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
int n;
memset(dp,0,sizeof(dp));
scanf("%d",&n);
for(int i = 1; i <= n; i++)
{
scanf("%d%d",&x[i],&y[i]);
}
for(int i = 1; i <= n; i++)
{
for(int j = 101*n; j >= y[i]; j--)
{
if(dp[j-y[i]] || j == y[i])
{
dp[j] = max(dp[j-y[i]] + x[i],dp[j]);
}
}
}
ll max_ = 0;
for(int i = 0; i <= n*101; i++)
{
if(dp[i]) max_ = max(max_,dp[i]*dp[i] - dp[i]*i - i*i);
}
printf("%lld\n",max_);
}
return 0;
}