这是昨晚codeforce的第三个题,没有做出来,后来听雪哥讲解,是一个dp,今天还做了好长时间。
注意题目给的a与b只差小于100,这样我们可以定义状态:f[i][j][k]表示前i天,在第i天选择第j门课程布置k个练习所得到的最多的练习。
我们可以用此状态来更新第i+1天选择l课程的状态。首先p[l].c > p[j].c && f[i ][j][k - p[j].a],如果满足k + K >= p[l].a && k + K <= p[l].b,则f[i + 1][l][k + K - p[l].a] = max(f[i + 1][l][k + K - p[l].a], f[i ][j][k - p[j].a] + k + K);如果满足k * K >= p[l].a && k * K <= p[l].b,则f[i + 1][l][k * K - p[l].a] = max(f[i + 1][l][k * K - p[l].a], f[i ][j][k - p[j].a] + k * K)。最后结果在f[n][j][k]中选择最大的,如果为0则无解。
初始化:让f为0,并让f[1][j][k - p[j].a] = k,第一天选课程的练习不受其他影响。Dp之前,首先要对课程按照复杂度从小到大排序,这也是能用dp的一个原因。Dp过程中可用一个g[i][j][k][2]来记录更新f[i][j][k]的课程编号和练习的数量,在利用递归求解答案。
#include <iostream>
#include <cstring>
#include <cstdio>
#include <string>
#include <vector>
#include <map>
#include <set>
#include <stack>
#include <cmath>
#include <queue>
#include <limits.h>
#include <utility>
#include <sstream>
#include <algorithm>
using namespace std;
#define LL long long
struct NODE{
LL a, b, c;
int num;
};
NODE p[55];
int n, m;
LL K;
LL f[55][55][110];
int g[55][55][110][2];
int cmp(NODE a, NODE b)
{
if(a.c == b.c){
return a.num < b.num;
}else return a.c < b.c;
}
void print_path(int dep, int j, int k)
{
if(!dep) return ;
print_path(dep - 1, g[dep][j][k][0], g[dep][j][k][1]);
printf("%d %I64d\n", p[j].num, k + p[j].a);
}
int main()
{
//freopen("input.txt", "r", stdin);
scanf("%d %d %I64d", &n, &m, &K);
for(int i = 1; i <= m; i++){
scanf("%I64d %I64d %I64d", &p[i].a, &p[i].b, &p[i].c);
p[i].num = i;
}
sort(p + 1, p + 1 + m, cmp);
memset(f, 0, sizeof(f));
for(int j = 1; j <= m; j++)
for(LL k = p[j].a; k <= p[j].b; k++)
f[1][j][k - p[j].a] = k;
for(int i = 2; i <= n; i++){
for(int j =i - 1; j <= m; j++){
for(LL k = p[j].a; k <= p[j].b; k++){
if(f[i - 1][j][k - p[j].a]){
for(int l = j + 1; l <= m; l++){
if(p[l].c > p[j].c){
LL tmp;
if(k + K >= p[l].a && k + K <= p[l].b){
tmp = k + K - p[l].a;
if(f[i][l][tmp] < f[i - 1][j][k - p[j].a] + k + K){
f[i][l][tmp] = f[i - 1][j][k - p[j].a] + k + K;
g[i][l][tmp][0] = j;
g[i][l][tmp][1] = k - p[j].a;
}
}
if(k * K >= p[l].a && k * K <= p[l].b){
tmp = k * K - p[l].a;
if(f[i][l][tmp] < f[i - 1][j][k - p[j].a] + k * K){
f[i][l][tmp] = f[i - 1][j][k - p[j].a] + k * K;
g[i][l][tmp][0] = j;
g[i][l][tmp][1] = k - p[j].a;
}
}
}
}
}
}
}
}
LL Maxx = 0;
int id, jd;
for(int j = n; j <= m; j++){
for(int i = 0; i <= p[j].b - p[j].a; i++){
if(Maxx < f[n][j][i]){
Maxx = f[n][j][i];
id = j;
jd = i;
}
}
}
if(Maxx){
//printf("%d %d %d\n", Maxx, id, jd);
printf("YES\n");
print_path(n, id, jd);
}else printf("NO\n");
return 0;
}
ps:昨晚的第二题貌似出题者的标程错了,不过确实是个很好的trip,太多人没有想到……