#include <bits/stdc++.h>
using namespace std;
vector<int> convolute(vector<int> a, vector<int> b) {
int n = a.size(), m = b.size();
vector<int> ans(n + m + 1);
for (int i = 0; i < n; i++)
for (int j = 0; j < m; j++)
ans[i + j] = max(ans[i + j], a[i] + b[j]);
}
vector<int> ans(vector<int> p, int L, int D) {
if(L <= 2 * D) {
vector<int> res(D + 1);
for(int i = 0; i <= D; i++)
res[i] = p[i + L - D];
return res;
}
vector<int> q = ans(p, L / 2, D);
if(L % 2) {
vector<int> res = convolute(q, q);
res.resize(D + 1);
return res;
}
else {
vector<int> res = convolute(q, p);
vector<int> newq(D + 1);
for (int i = 0; i <= D; i++)
newq[i] = res[i + 1];
return newq;
}
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
int m, L, jnk;
cin >> m >> L;
for(int i = 0; i < m; i++) {
int x;
cin >> x;
if(x != 0)
return 1;
}
cin >> jnk;
vector<int> v, w;
for(int i = 1; i <= m; i++) {
int l, tpow = 1;
cin >> l;
while(l) {
if(l % 2)
v.push_back(tpow), w.push_back(tpow * i);
tpow *= 2;
l /= 2;
}
}
int n = v.size();
vector<vector<int>> dp(n + 1, vector<int>(2 * m + 1));
for(int i = 1; i <= n; i++) {
for(int j = 1; j <= 2 * m; j++) {
dp[i][j] = dp[i - 1][j];
if(j >= w[i - 1])
dp[i][j] = max(dp[i][j],dp[i][j - w[i - 1]] + v[i - 1]);
}
}
vector<int> p(2 * m + 1);
for(int i = 0; i <= 2 * m; i++)
p[i] = dp[n][i];
cout << ans(p, L, m).back() + jnk;
}
Compilation message
vault.cpp: In function 'std::vector<int> convolute(std::vector<int>, std::vector<int>)':
vault.cpp:10:1: warning: no return statement in function returning non-void [-Wreturn-type]
10 | }
| ^
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
0 ms |
348 KB |
Execution failed because the return code was nonzero |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
0 ms |
348 KB |
Execution failed because the return code was nonzero |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
348 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
348 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
348 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
0 ms |
348 KB |
Execution failed because the return code was nonzero |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
348 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
0 ms |
348 KB |
Execution failed because the return code was nonzero |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
348 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
0 ms |
348 KB |
Execution failed because the return code was nonzero |
2 |
Halted |
0 ms |
0 KB |
- |