Codeforces Round #436 (Div. 2) Fire

01背包的变形问题。

#include <bits\stdc++.h>
#define pb push_back
#define mp make_pair

using namespace std;
struct node
{
    int t,d,p,i;
}a[105];
bool cmp(node a,node b)
{
    return a.d<b.d;
}
int b[105];
int dp[2005],path[105][2005];
int main()
{
    int n;
    int cnt=0;
    scanf("%d",&n);
  for(int i=0;i<n;i++)
    scanf("%d%d%d",&a[i].t,&a[i].d,&a[i].p),a[i].i=i+1;
  sort(a,a+n,cmp);
  int maxn=0,tmp=0;
  memset(path,-1,sizeof(path));
  memset(dp,0,sizeof(dp));

  for(int i=0;i<n;i++)
  {
      if(i!=0)
      for(int j=1;j<a[i].t;j++)
        path[i][j]=path[i-1][j];
      for(int j=a[i].d-1;j>=a[i].t;j--)
      {
          if(dp[j]<=dp[j-a[i].t]+a[i].p)//01背包如果放入，路径就是自己
          {
              dp[j]=dp[j-a[i].t]+a[i].p;
              path[i][j]=i;
          }
      else if(i>0)//如果不放入，路径就是前一个的路径
      {
          path[i][j]=path[i-1][j];
      }
      }
  }

  for(int i=1;i<=2000;i++)
  {
      if(dp[i]>maxn)
      {
          maxn=dp[i];
          tmp=i;
      }
  }
  //for(int i=1;i<=7;i++)printf("%d\n",path[n-1][i]);
  int tmp1=n-1;
  while(tmp>0)
  {
     // printf("%d\n",tmp);
      b[cnt++]=a[path[tmp1][tmp]].i;
      tmp=tmp-a[path[tmp1][tmp]].t;
      tmp1--;
      tmp1=path[tmp1][tmp];
  }
  //sort(b,b+cnt);
  printf("%d\n",maxn);
  printf("%d\n",cnt);
  for(int i=cnt-1;i>=0;i--)//按顺序输出，应该是倒叙输出
    printf("%d ",b[i]);
    puts("");
    return 0;
}

/*
3
3 7 4
2 6 5
3 7 6
*/