QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#105100 | #5374. 数圈 | HgCl2 | 30 | 208ms | 5704kb | C++14 | 3.2kb | 2023-05-12 23:04:25 | 2023-05-12 23:04:28 |
Judging History
answer
#include <algorithm>
#include <array>
#include <cstddef>
#include <cstdint>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/detail/container_base_dispatch.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#include <fstream>
#include <functional>
#include <iostream>
#include <string>
namespace __gnu_pbds {
struct null_type;
struct rb_tree_tag;
} // namespace __gnu_pbds
namespace {
using std::cin;
using std::cout;
using std::int64_t;
using std::size_t;
using Tree = __gnu_pbds::tree<int64_t, __gnu_pbds::null_type, std::less<>,
__gnu_pbds::rb_tree_tag,
__gnu_pbds::tree_order_statistics_node_update>;
namespace base {
template <typename T, size_t... sizes>
struct NestedArray {};
template <typename T, size_t size, size_t... sizes>
struct NestedArray<T, size, sizes...> {
using Type = std::array<typename NestedArray<T, sizes...>::Type, size>;
};
template <typename T>
struct NestedArray<T> {
using Type = T;
};
template <typename T, size_t... sizes>
using Array = typename NestedArray<T, sizes...>::Type;
void OptimizeIO() {
std::ios::sync_with_stdio(false);
cin.tie(nullptr), cout.tie(nullptr);
}
void OptimizeIO(const std::string &input_file, const std::string &output_file) {
static std::ifstream input_stream(input_file);
static std::ofstream output_stream(output_file);
cin.rdbuf(input_stream.rdbuf());
cout.rdbuf(output_stream.rdbuf());
cin.tie(nullptr), cout.tie(nullptr);
}
} // namespace base
using base::Array;
const int kMaxN = 1.0e5 + 5;
const int64_t kBase = 1 << 20;
int n, m;
Array<int, kMaxN> a, sum_a;
Tree tree;
struct Node {
int val, num;
};
Array<Node, kMaxN> b;
int Divide(int x, int y) { return x >= 0 ? x / y : (x - y + 1) / y; }
void Solve() {
cin >> n;
for (int i = 1; i <= n; i++) cin >> a[i];
sum_a[0] = 0;
for (int i = 1; i <= n; i++) sum_a[i] = sum_a[i - 1] + a[i];
int s = sum_a[n];
if (s < 0) {
cout << "-1\n";
return;
}
if (s == 0) {
if (std::count(a.begin() + 1, a.begin() + n + 1, 0) == n) {
cout << "0\n";
} else {
cout << "-1\n";
}
return;
}
int64_t ans = 0;
for (int i = 0; i < n; i++) {
ans += tree.size() - tree.order_of_key((sum_a[i] + 1) * kBase);
tree.insert(sum_a[i] * kBase + i);
}
tree.clear();
std::sort(sum_a.begin(), sum_a.begin() + n);
m = 0;
for (int i = 0; i < n;) {
int j = i;
while (j < n && sum_a[i] == sum_a[j]) j++;
b[++m] = {sum_a[i], j - i};
i = j;
}
int64_t sum_q = 0;
int num = 0;
for (int i = 1; i <= m; i++) {
int q = Divide(b[i].val, s);
int r = b[i].val - q * s;
ans += static_cast<int64_t>(b[i].num) * (num * q - sum_q);
ans -= static_cast<int64_t>(b[i].num) *
(tree.size() - tree.order_of_key((r + 1) * kBase));
r++;
if (r == s) r = 0, q++;
sum_q += static_cast<int64_t>(b[i].num) * q;
for (int j = 0; j < b[i].num; j++) tree.insert(r * kBase + (++num));
}
tree.clear();
cout << ans << "\n";
}
int Main() {
base::OptimizeIO();
int t;
cin >> t;
while (t--) Solve();
return 0;
}
} // namespace
int main() { return Main(); }
Details
Tip: Click on the bar to expand more detailed information
Subtask #1:
score: 10
Accepted
Test #1:
score: 10
Accepted
time: 0ms
memory: 3440kb
input:
10 3 2 -9 -4 3 4 6 0 3 3 -10 0 3 7 -6 3 3 -6 7 10 3 -3 9 -2 3 6 1 -2 3 5 2 -2 3 -9 -5 7 3 -4 -5 6
output:
-1 0 -1 3 1 4 2 1 -1 -1
result:
ok 10 lines
Subtask #2:
score: 20
Accepted
Dependency #1:
100%
Accepted
Test #2:
score: 20
Accepted
time: 2ms
memory: 3296kb
input:
10 50 -5 -5 3 8 0 0 3 8 5 -7 6 -9 5 5 2 7 -9 1 -7 5 0 10 -6 -2 -6 -8 9 7 4 3 -9 9 5 9 8 1 7 0 0 -5 -1 -3 5 -7 5 -4 6 8 -1 1 50 -6 4 3 -6 -9 -8 -5 -2 -10 7 -4 1 -1 -5 2 1 -10 8 8 7 -8 -5 3 -6 10 3 2 -1 10 0 4 -6 9 3 -6 3 9 -4 4 -2 -3 4 -9 -7 7 1 5 2 -4 5 50 2 -10 5 3 -1 1 9 -1 -5 3 -2 -10 7 0 5 -5 -1...
output:
161 -1 249 -1 21873 -1 452 -1 -1 314
result:
ok 10 lines
Subtask #3:
score: 0
Wrong Answer
Dependency #2:
100%
Accepted
Test #3:
score: 0
Wrong Answer
time: 9ms
memory: 3584kb
input:
10 2000 -4438 -448 2902 3873 -5348 1821 -5284 2787 -1369 -4712 3298 2808 1651 -4568 4377 870 2217 -2683 1217 120 -3854 1156 -2129 -3757 -2704 3026 -1745 -5327 -1315 405 3944 340 -1510 2213 -24 -32 -5414 -2330 760 3715 -4871 2831 1917 3148 1360 -3662 -4281 -1248 788 1334 -3401 2050 4174 3163 -2456 33...
output:
90206708583 9272643195 2640993721 148400379 20504656 2904294 -1 -6402916481728 61998 67150
result:
wrong answer 8th lines differ - expected: '6666669000000', found: '-6402916481728'
Subtask #4:
score: 0
Wrong Answer
Test #4:
score: 0
Wrong Answer
time: 208ms
memory: 5704kb
input:
10 30000 -3879 -556 4570 1863 2815 -4010 2471 -270 2835 3071 -3331 -1251 -2243 4221 -5249 -4134 3376 1978 858 2545 -4207 386 3875 2029 1706 1119 3065 -3097 4399 4385 -3021 2473 2506 2157 3946 -886 3929 1478 2728 -4239 4091 -151 -4762 -2136 -1424 2162 -669 267 190 -1180 2640 -757 -2078 -1409 3165 216...
output:
-21282471247843 22341295988767 -6003284067240 36468201766247 -8979140171837 19189348837140 199398449992115 36547177979137 -43520712753021 42868047325028
result:
wrong answer 1st lines differ - expected: '121860198793245', found: '-21282471247843'
Subtask #5:
score: 0
Skipped
Dependency #4:
0%
Subtask #6:
score: 0
Skipped
Dependency #5:
0%
Subtask #7:
score: 0
Skipped
Dependency #1:
100%
Accepted
Dependency #2:
100%
Accepted
Dependency #3:
0%