이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
int main()
{
int64_t m, l;
cin >> m >> l;
vector<int64_t> a(2 * m + 1);
for (int64_t &x : a)
cin >> x;
if (l < 0)
{
l = -l;
for (int64_t i = 0; i < m; i++)
swap(a[m - i - 1], a[m + i + 1]);
}
int64_t p = 0, total_added = 0;
vector<int64_t> b(a.begin(), a.end());
for (int64_t i = -m; i <= 0; i++)
p += a[i + m] * i, b[i + m] = 0, total_added += a[i + m];
for (int64_t i = 1; i <= m; i++)
{
for (int64_t j = 62; j >= 0; j--)
if (1LL << j <= b[i + m] && p + (1LL << j) * i <= l)
{
p += (1LL << j) * i;
b[i + m] -= 1LL << j;
total_added += 1LL << j;
}
}
if (p < l - m)
{
p = 0;
total_added = 0;
b = vector<int64_t>(a.begin(), a.end());
for (int64_t i = 0; i <= m; i++)
p += a[i + m] * i, b[i + m] = 0, total_added += a[i + m];
for (int64_t i = -1; i >= -m; i--)
{
for (int64_t j = 62; j >= 0; j--)
if (1LL << j <= b[i + m] && p + (1LL << j) * i >= l)
{
p += (1LL << j) * i;
b[i + m] -= 1LL << j;
total_added += 1LL << j;
}
}
if (p < l - m)
{
cout << "impossible\n";
return 0;
}
}
// l resides at m * m.
vector<int64_t> dp1(2 * m * m + 1, 0), dp2(2 * m * m + 1, 0);
dp1[m * m - (l - p)] = total_added;
for (int64_t i = -m; i <= m; i++)
{
for (int64_t j = 0; j < dp1.size(); j++)
{
if (!dp1[j])
continue;
int64_t max_add = min<int64_t>(b[i + m], m),
max_remove = min<int64_t>(a[i + m] - b[i + m], m);
for (int64_t k = 0;
1LL << k < max_add &&
0 <= j + (1LL << k) * i && j + (1LL << k) * i < dp1.size();
k++)
{
dp2[j + (1LL << k) * i] = max<int64_t>(dp2[j + (1LL << k) * i], dp1[j] + (1LL << k));
max_add -= 1LL << k;
}
if (0 <= j + max_add * i && j + max_add * i < dp1.size())
dp2[j + max_add * i] = max<int64_t>(dp2[j + max_add * i], dp1[j] + max_add);
for (int64_t k = 0;
1LL << k < max_remove &&
0 <= j - (1LL << k) * i && j - (1LL << k) * i < dp1.size();
k++)
{
dp2[j - (1LL << k) * i] = max<int64_t>(dp2[j - (1LL << k) * i], dp1[j] - (1LL << k));
max_remove -= 1LL << k;
}
if (0 <= j - max_remove * i && j - max_remove * i < dp1.size())
dp2[j - max_remove * i] = max<int64_t>(dp2[j - max_remove * i], dp1[j] - max_remove);
}
swap(dp1, dp2);
for (size_t i = 0; i < dp1.size(); i++)
dp1[i] = max(dp1[i], dp2[i]), dp2[i] = max(dp1[i], dp2[i]);
}
if (!dp1[m * m])
cout << "impossible\n";
else
cout << dp1[m * m] << '\n';
}
컴파일 시 표준 에러 (stderr) 메시지
vault.cpp: In function 'int main()':
vault.cpp:66:31: warning: comparison of integer expressions of different signedness: 'int64_t' {aka 'long int'} and 'std::vector<long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
66 | for (int64_t j = 0; j < dp1.size(); j++)
| ~~^~~~~~~~~~~~
vault.cpp:75:64: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
75 | 0 <= j + (1LL << k) * i && j + (1LL << k) * i < dp1.size();
| ~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~
vault.cpp:81:57: warning: comparison of integer expressions of different signedness: 'int64_t' {aka 'long int'} and 'std::vector<long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
81 | if (0 <= j + max_add * i && j + max_add * i < dp1.size())
| ~~~~~~~~~~~~~~~~^~~~~~~~~~~~
vault.cpp:86:64: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
86 | 0 <= j - (1LL << k) * i && j - (1LL << k) * i < dp1.size();
| ~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~
vault.cpp:92:63: warning: comparison of integer expressions of different signedness: 'int64_t' {aka 'long int'} and 'std::vector<long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
92 | if (0 <= j - max_remove * i && j - max_remove * i < dp1.size())
| ~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |