Submission #772821

#	Time	Username	Problem	Language	Result	Execution time	Memory
772821		birthdaycake	Fruits (NOI22_fruits)	C++17	30 / 100	772 ms	278428 KiB

Main

#ifndef _GLIBCXX_NO_ASSERT
#include <cassert>
#endif
#include <cctype>
#include <cerrno>
#include <cfloat>
#include <ciso646>
#include <climits>
#include <clocale>
#include <cmath>
#include <csetjmp>
#include <csignal>
#include <cstdarg>
#include <cstddef>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <ctime>
 
#if __cplusplus >= 201103L
#include <ccomplex>
#include <cfenv>
#include <cinttypes>
#include <cstdbool>
#include <cstdint>
#include <ctgmath>
#include <cwchar>
#include <cwctype>
#endif
 
#include <algorithm>
#include <bitset>
#include <complex>
#include <deque>
#include <exception>
#include <fstream>
#include <functional>
#include <iomanip>
#include <ios>
#include <iosfwd>
#include <iostream>
#include <istream>
#include <iterator>
#include <limits>
#include <list>
#include <locale>
#include <map>
#include <memory>
#include <new>
#include <numeric>
#include <ostream>
#include <queue>
#include <set>
#include <sstream>
#include <stack>
#include <stdexcept>
#include <streambuf>
#include <string>
#include <typeinfo>
#include <utility>
#include <valarray>
#include <vector>
 
 
#include <array>
#include <atomic>
#include <chrono>
#include <condition_variable>
#include <forward_list>
#include <future>
#include <initializer_list>
#include <mutex>
#include <random>
#include <ratio>
#include <regex>
#include <scoped_allocator>
#include <system_error>
#include <thread>
#include <tuple>
#include <typeindex>
#include <type_traits>
#include <unordered_map>
#include <unordered_set>
#include <fstream>
#define endl '\n'
#define int long long
#define mod 1000000007
using namespace std;
 
 
int a[400001],c[400001],b[40001];
int dp[4001][4001];
int boo[400001];
int m,sz;
vector<int>x;
signed main(){
    
    
    int n; cin >> n;
    b[0] = -1;
    for(int i = 1; i <= n; i++) {
        cin >> a[i];
        b[i] = max(a[i], b[i - 1]);
        if(a[i] > -1) boo[a[i]] = 1;
    }
    for(int i = 1; i <= n; i++) cin >> c[i];
    int mn = n + 1;
    vector<int>x;
    for(int j = 1; j <= n; j++){
        if(!boo[j]) {
            x.push_back(j);
        }
    }
    int k = 0;
    for(int i = 1; i <= n; i++){
        if(a[i] == -1) b[i] = max(b[i], x[k++]);
        b[i] = max(b[i - 1], b[i]);
    }
    for(int i = 1; i <= n; i++){
        for(int j = 0; j <= n; j++) dp[i][j] = -1e9;
    }
    // dp[i][j] = max score for the first i elements such that j is the max element...
    //
    for(int i = 1; i <= n; i++){
        int xx = b[i];
        for(int j = xx; j <= n; j++){
            if(a[i] == -1){
                if(!boo[j]) dp[i][j] = max(dp[i - 1][j - 1] + c[j],dp[i][j]);
            }else{
                if(j == a[i]) dp[i][j] = max(dp[i - 1][j - 1] + c[j],dp[i][j]);
                else dp[i][j] = max(dp[i - 1][j], dp[i][j]);

            }
            dp[i][j] = max(dp[i - 1][j], dp[i][j]);
            dp[i][j] = max(dp[i][j - 1], dp[i][j]);
        }
        /*for(int j = 1; j <= n; j++){
            cout << dp[i][j] << ' ';
        }
        cout << endl;*/
        
    }
    int mx = 0;
    for(int i = 1; i <= n; i++){
        mx = max(mx,dp[i][n]);
        cout << mx << ' ';
    }
}

Compilation message (stderr)

Main.cpp: In function 'int main()':
Main.cpp:107:9: warning: unused variable 'mn' [-Wunused-variable]
  107 |     int mn = n + 1;
      |         ^~

• Subtask 1: 6 / 6
• Subtask 2: 0 / 5
• Subtask 3: 11 / 11
• Subtask 4: 13 / 13
• Subtask 5: 0 / 23
• Subtask 6: 0 / 42