#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;
for(int j = 1; j <= n; j++){
if(!boo[j]) {
mn = j;
break;
}
}
for(int i = 1; i <= n; i++){
int xx = b[i];
if(a[i] == -1) xx = mn;
for(int j = 0; j < xx; j++) dp[i][j] = -1e18;
for(int j = xx; j <= n; j++){
if(a[i] == -1){
if(!boo[j]) dp[i][j] = dp[i - 1][j - 1] + c[j];
else dp[i][j] = dp[i - 1][j];
}else{
if(j == a[i]) dp[i][j] = dp[i - 1][j - 1] + c[i];
else if(i != 1)dp[i][j] = dp[i - 1][j];
else dp[i][j] = -1e18;
}
dp[i][j] = max(dp[i][j - 1], dp[i][j]);
if(i != 1) dp[i][j] = max(dp[i - 1][j], 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 << ' ';
}
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
340 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
1364 KB |
Output is correct |
2 |
Correct |
34 ms |
39664 KB |
Output is correct |
3 |
Runtime error |
503 ms |
267716 KB |
Execution killed with signal 11 |
4 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
340 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
340 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
1064 ms |
29844 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
340 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |