Submission #226900

#	Time	Username	Problem	Language	Result	Execution time	Memory
226900		DavidDamian	Sličice (COCI19_slicice)	C++11	90 / 90	65 ms	1528 KiB

slicice

#include <bits/stdc++.h>
using namespace std;
///Dynamic Programming with answer
///Determine the maximum number of points when adding some extra images
int memo[505][505];
int p[505][505];
int A[505];
int B[1005];
int sum[505];
int extra[505];
int n,m,K;
void answer(int i,int k)
{
    if(i==0 || k==0) return;
    extra[i]=p[i][k];
    answer(i-1,k-extra[i]);
}
int main()
{
    ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    cin>>n>>m>>K;
    for(int i=1;i<=n;i++){
        cin>>A[i];
    }
    for(int i=0;i<=m;i++){
        cin>>B[i];
    }
    for(int i=1;i<=n;i++){
        sum[i]=sum[i-1]+B[ A[i] ];
    }
    for(int i=m+1;i<=2*m+1;i++){
        B[i]=B[i-1];
    }
    for(int i=1;i<=n;i++){
        memo[i%2][0]=sum[i];
        for(int k=1;k<=K;k++){
            for(int j=0;j<=k;j++){
                int points=memo[(i-1)%2][k-j]+B[ A[i]+j ];
                if(points>memo[i%2][k]){
                    memo[i%2][k]=points;
                    p[i][k]=j;
                }
            }
        }
    }
    cout<<memo[n%2][K]<<'\n';
    //answer(n,K);
    //for(int i=1;i<=n;i++){
    //    cout<<extra[i]<<" ";
    //}
    return 0;
}

• Subtask 1: 90 / 90