Course Selection System ZOJ - 3956 【01背包变形】好题~

最新推荐文章于 2022-03-26 15:30:09 发布

原创最新推荐文章于 2022-03-26 15:30:09 发布 · 272 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

动态规划之基础DP 专栏收录该内容

55 篇文章

订阅专栏

本文介绍了一个课程选择系统的优化问题，旨在帮助学生通过选取合适数量的课程来最大化学期满意度。通过将问题转化为01背包问题，并利用动态规划求解，文章详细阐述了算法的设计与实现过程。

Course Selection System

Time Limit: 1 Second Memory Limit: 65536 KB

There are n courses in the course selection system of Marjar University. The i-th course is described by two values: happiness H_i and credit C_i. If a student selects m courses x₁, x₂, ..., x_m, then his comfort level of the semester can be defined as follows:

$(\sum_{i=1}^{m} H_{x_i})^2-(\sum_{i=1}^{m} H_{x_i})\times(\sum_{i=1}^{m} C_{x_i})-(\sum_{i=1}^{m} C_{x_i})^2$

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?

Input

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 H_i and C_i (1 ≤ H_i ≤ 10000, 1 ≤ C_i ≤ 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++.

Output

For each case, you should output one integer denoting the maximum comfort.

Sample Input

Sample Output

191
0

Hint

For the first case, Edward should select the first and second courses.

For the second case, Edward should select no courses.

Author: WANG, Yucheng
Source: The 17th Zhejiang University Programming Contest Sponsored by TuSimple

题意：已知每个物品的幸福值H[i]和信用值C[i],要求实现 $(\sum_{i=1}^{m} H_{x_i})^2-(\sum_{i=1}^{m} H_{x_i})\times(\sum_{i=1}^{m} C_{x_i})-(\sum_{i=1}^{m} C_{x_i})^2$ 输出其最大值

思路：

幸福值→价值，信用值→重量。假若背包容量给定，那么通过01背包DP可以求出在给定容积下可获得的最大价值。

观察数据，发现容量不超过50000，暴力枚举容积，可求得最大的价值。并且该方程对称轴在x轴左侧，开口朝上。

那么当容积确定后，幸福值越大，该方程的值越大。复杂度为O(T*n*50000)

AC代码：

#include <stdio.h>
#include <string.h>
using namespace std;
typedef long long ll;

const int MAX_N=1e6+3;
const int MOD=1e9+7;
int dp[MAX_N];
int w[MAX_N],v[MAX_N];
int n,m;

ll max(ll a,ll b){
    return a>b ? a :b;
}

int main(void){
    int T;
    scanf("%d",&T);
    while(T--){
        scanf("%d",&n);
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=n;++i)   scanf("%d%d",&v[i],&w[i]);
        ///求出每个背包容量对应能获得的最大价值，枚举容积的原因在于背包容量小
        for(int i=1;i<=n;i++)
            for(int j=50000;j>=w[i];j--)    dp[j]=max(dp[j],dp[j-w[i]]+v[i]);
        ll ans=0;
        for(int i=1;i<=50000;i++)   ans=max(ans,1LL*dp[i]*dp[i]-1LL*dp[i]*i-1LL*i*i);
        printf("%lld\n",ans);
    }
}