51nod-1636 教育改革

[题目链接]

思路：不难的一道dp题目，状态都想好了，略估一下复杂度想可能会TLE，结果却一直是WA，头疼~

dp[i][j][k]:表示第i天选第j门课作业量为k时的最大总作业量。
第i天选第j门课的状态可由第i-1天选复杂度小于第j门课的状态转移得到。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;

ll n,m,d;
ll dp[55][55][220];
struct Node{
    ll a,b,c;
    bool operator<(const Node& n)const{
        return c<n.c;
    }
}node[55];

int main()
{
    cin>>n>>m>>d;
    for(int i=0;i<m;i++)
        cin>>node[i].a>>node[i].b>>node[i].c;
    sort(node,node+m);
    memset(dp,0,sizeof(dp));
    for(int i=0;i<m;i++){
        for(int j=0;j<=node[i].b-node[i].a;j++)
            dp[1][i][j]=j+node[i].a;
    }
    for(int i=1;i<n;i++){
        for(int j=1;j<m;j++){ //枚举k课程去更新j课程
            for(int k=0;k<j;k++){
                if(node[j].c==node[k].c)continue;
                for(ll l=node[k].a;l<=node[k].b;l++){
                    if(dp[i][k][l-node[k].a]==0)continue;
                    if(l+d>=node[j].a&&l+d<=node[j].b){
                        if(dp[i+1][j][l+d-node[j].a]<dp[i][k][l-node[k].a]+l+d)
                        dp[i+1][j][l+d-node[j].a]=dp[i][k][l-node[k].a]+l+d;
                    }
                    if(l*d>=node[j].a&&l*d<=node[j].b){
                        if(dp[i+1][j][l*d-node[j].a]<dp[i][k][l-node[k].a]+l*d)
                        dp[i+1][j][l*d-node[j].a]=dp[i][k][l-node[k].a]+l*d;
                    }
                }
            }
        }
    }
    ll Max=0;
    for(int i=0;i<m;i++){
        for(int j=0;j<=node[i].b-node[i].a;j++)
            if(dp[n][i][j]>Max)Max=dp[n][i][j];
    }
    if(Max){
        cout<<"YES\n";
        cout<<Max<<endl;
    }
    else cout<<"NO\n";
    return 0;
}