Submission #776555

#	Time	Username	Problem	Language	Result	Execution time	Memory
776555		vjudge1	Knapsack (NOI18_knapsack)	C++17	100 / 100	88 ms	36344 KiB

knapsack

#include <bits/stdc++.h>
#define int long long
using namespace std;

const int maxn = 2010;
const int mod = 1e9 + 7;
typedef pair<int,int> ii;
typedef pair<int,ii> iii;
#define ff first
#define ss second

int n,S;
vector <ii> a[maxn];
int dp[maxn][maxn];


signed main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr); cout.tie(nullptr);
    cin>>S>>n;
    for (int i=1; i<=n; i++)
    {
        int v,w,k; cin>>v>>w>>k;
        a[w].push_back({v,k});
    }
    for (int i=1; i<=S; i++)
    {
        for (int j=0; j<=S; j++) dp[i][j] = dp[i-1][j];
        if (a[i].empty()) continue;
        sort(a[i].begin(),a[i].end(),greater<ii> ());
        for (int j=0; j<=S-i; j++)
        {
            int it,sum,left,cnt;
            it = cnt = 0;
            left = sum = 0;
            while (true)
            {
                cnt++;
                if (j+cnt*i > S) break;
                if (left==0)
                {
                    if (it==a[i].size()) break;
                    left = a[i][it++].ss;
                }
                sum+=a[i][it-1].ff;
                left--;
                dp[i][j+cnt*i] = max(dp[i][j+cnt*i],sum + dp[i-1][j]);
            }
        }
    }
    cout<<dp[S][S]<<'\n';
}

Compilation message (stderr)

knapsack.cpp: In function 'int main()':
knapsack.cpp:43:27: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   43 |                     if (it==a[i].size()) break;
      |                         ~~^~~~~~~~~~~~~

• Subtask 1: 12 / 12
• Subtask 2: 17 / 17
• Subtask 3: 20 / 20
• Subtask 4: 24 / 24
• Subtask 5: 27 / 27