Submission #770571

#	Time	Username	Problem	Language	Result	Execution time	Memory
770571		birthdaycake	Fruits (NOI22_fruits)	C++17	6 / 100	1083 ms	267072 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];
int dp[4001][4001];
int m,sz;
vector<int>x;
int solve(int array_plc, int prev){
    if(array_plc == m) return 0;
    if(dp[array_plc][prev] != -1) return dp[array_plc][prev];
    dp[array_plc][prev] = 0;
    if(a[array_plc] == -1){
        int j = upper_bound(x.begin(), x.end(), prev) - x.begin();
        for(int k = j; k < sz; k++) {
            int xx = solve(array_plc + 1, max(x[k],prev));
            if(x[k] > prev) xx += c[x[k]];
            dp[array_plc][prev] = max(dp[array_plc][prev], xx);
        }
        if(j == sz){
            dp[array_plc][prev] = max(dp[array_plc][prev],solve(array_plc + 1, prev));
        }
        return dp[array_plc][prev];
    }else{
        dp[array_plc][prev] = solve(array_plc + 1, max(prev, a[array_plc]));
        if(a[array_plc] > prev) dp[array_plc][prev] += c[a[array_plc]];
        return dp[array_plc][prev];
    }
}
signed main(){
    
    
    int n; cin >> n;
    for(int i = 0; i < n; i++) cin >> a[i];
    for(int i = 1; i <= n; i++) cin >> c[i];
    
    
    for(int i = 0; i <= n; i++){
        for(int j = 0; j <= n; j++) dp[i][j] = -1;
    }
    set<int>d;
    for(int j = 1; j <= n; j++) d.insert(j);
    for(int j = 0; j < n; j++) d.erase(a[j]);
    for(auto s: d) x.push_back(s);
    sz = x.size();
    for(int i = 0; i < n; i++){
        m = i + 1;
        cout << solve(0,0) << ' ';
        for(int k = 0; k <= n; k++){
            for(int j = 0; j <= n; j++) dp[k][j] = -1;
        }
    }
}

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