20分求助

#include<bits/stdc++.h>
using namespace std;
int n,m,ans;
int a[100][5]; 
int w[100],c[100];
int dp[100][100005];
bool f[100];
int find(int i){
    for(int j=i-1;j>=1;j--){
        if(f[j]==1)return j;
    }
    return 0;
}
int main(){
    cin>>m>>n;
    for(int i=1;i<=n;i++){
        cin>>w[i]>>c[i];
        int x;
        cin>>x;
        if(x!=0){
            a[x][0]++;
            a[x][a[x][0]]=i; //记录附件 
        }else{
            f[i]=1;
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=0;j<=a[i][0];j++)cout<<a[i][j]<<" ";
        cout<<endl; 
    }
    for(int i=1;i<=n;i++){
        if(f[i]==1){ //它是主件 
            for(int j=0;j<=m;j++){
                int last=find(i);  //上一个主件的编号 
                if(j<w[i]){//主件都买不起
                    dp[i][j]=dp[last][j];
                    continue;
                }

                dp[i][j]=max(dp[last][j-w[i]]+w[i]*c[i],dp[last][j]); 

                if(a[i][0]>=1&&j>=w[i]+w[a[i][1]])
                    dp[i][j]=max(dp[i][j],dp[last][j-w[i]-w[a[i][1]]]+w[i]*c[i]+w[a[i][1]]*c[a[i][1]]);
                if(a[i][0]>=2&&j>=w[i]+w[a[i][2]])
                    dp[i][j]=max(dp[i][j],dp[last][j-w[i]-w[a[i][2]]]+w[i]*c[i]+w[a[i][2]]*c[a[i][2]]); 
                if(a[i][0]>=3&&j>=w[i]+w[a[i][3]])
                    dp[i][j]=max(dp[i][j],dp[last][j-w[i]-w[a[i][3]]]+w[i]*c[i]+w[a[i][3]]*c[a[i][3]]);

            }
        }
    }
    for(int i=0;i<=m;i++)ans=max(ans,dp[n][i]);
    cout<<ans;
    return 0;
}
/*
10 3
5 2 0
3 2 1
3 1 1
*/