| # |
Submission time |
Handle |
Problem |
Language |
Result |
Execution time |
Memory |
| 737224 |
2023-05-07T00:08:52 Z |
Olympia |
Zoltan (COCI16_zoltan) |
C++17 |
|
662 ms |
40264 KB |
#include <iostream>
#include <set>
#include <cmath>
#include <queue>
#include <vector>
#include <cstdlib>
#include <ctime>
#include <cmath>
#include <algorithm>
#include <cassert>
#include <map>
#include <deque>
using namespace std;
const int MOD = 1e9 + 7;
struct SegmentTree {
vector<pair<int,int64_t>> vec;
pair<int,int> id = make_pair(-1, 1);
pair<int,int> merge (pair<int,int64_t> left, pair<int,int64_t> right) {
if (left.first == -1) return right;
if (right.first == -1) return left;
if (left.first > right.first) return left;
if (left.first < right.first) return right;
left.second = ((int64_t)right.second + (int64_t)left.second) % MOD;
return left;
}
pair<int,int> query (int dum, int tl, int tr, int l, int r) {
if (tl >= l and tr <= r) {
return vec[dum];
}
if (r < tl or tr < l) {
return id;
}
return merge(query(2 * dum + 1, tl, (tl + tr)/2, l, r), query(2 * dum + 2, (tl + tr)/2 + 1, tr, l, r));
}
pair<int,int> query (int l, int r) {
return query(0, 0, (int)vec.size()/2 - 1, l, r);
}
void update (int dum, pair<int,int64_t> node) {
dum += (int)vec.size()/2 - 1;
vec[dum] = node;
while (dum != 0) {
dum = (dum - 1)/2;
vec[dum] = merge(vec[2 * dum + 1], vec[2 * dum + 2]);
}
}
SegmentTree (int n) {
n = (1 << (int)log2(n)) * 2;
vec.assign(2 * n, id);
}
};
SegmentTree st = SegmentTree(1);
vector<pair<int,int64_t>> lis (vector<int>& v) {
st.vec.assign(st.vec.size(), st.id);
map<int,vector<int>> oc;
for (int i = 0; i < v.size(); i++) {
oc[-v[i]].push_back(i);
}
vector<pair<int,int64_t>> vec;
vec.push_back(make_pair(0, 1));
int used = pow(2, v.size() - 1);
for (auto& p: oc) {
vector<pair<int,int64_t>> upd;
for (int i: p.second) {
pair<int,int64_t> node = st.query(i + 1, (int)v.size() - 1);
if (node.first == -1) {
node = make_pair(1, 1);
} else {
node.first += 1;
}
upd.push_back(node);
}
reverse(upd.begin(), upd.end());
for (int i: p.second) {
st.update(i, upd.back()), upd.pop_back();
}
pair<int,int64_t> g = st.id;
for (int i: p.second) {
g = st.merge(g, st.query(i, i));
}
vec.emplace_back(g);
}
return vec;
}
vector<pair<int,int64_t>> lds (vector<int>& v) {
for (int i = 0; i < st.vec.size(); i++) {
st.vec[i] = st.id;
}
map<int,vector<int>> oc;
for (int i = 0; i < v.size(); i++) {
oc[v[i]].push_back(i);
}
vector<pair<int,int64_t>> vec;
vec.push_back(make_pair(0, 1));
for (auto& p: oc) {
vector<pair<int,int64_t>> upd;
for (int i: p.second) {
pair<int,int64_t> node = st.query(i + 1, (int)v.size() - 1);
if (node.first == -1) {
node = make_pair(1, 1);
} else {
node.first += 1;
}
upd.push_back(node);
}
reverse(upd.begin(), upd.end());
for (int i: p.second) {
st.update(i, upd.back()), upd.pop_back();
}
vec.emplace_back(st.query(0, (int)v.size() - 1));
}
reverse(vec.begin(), vec.end());
return vec;
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
int n;
cin >> n;
st = SegmentTree(n);
vector<int> v(n);
for (int i = 0; i < n; i++) {
cin >> v[i];
}
auto a1 = lds(v);
auto a2 = lis(v);
int mx = 0;
int64_t cnt = 0;
for (int i = 0; i < a1.size(); i++) {
if (a1[i].first + a2[i].first > mx) {
mx = a1[i].first + a2[i].first;
cnt = ((int64_t)a1[i].second * (int64_t)a2[i].second) % MOD;
cnt %= MOD;
} else if (a1[i].first + a2[i].first == mx) {
cnt += ((int64_t)a1[i].second * (int64_t)a2[i].second) % MOD;
cnt %= MOD;
}
}
for (int i = 0; i < n - mx; i++) {
cnt *= 2;
cnt %= MOD;
}
if (cnt % 2 == 0) cnt /= 2;
else cnt = (cnt + MOD)/2 % MOD;
cout << mx << " " << cnt;
}
Compilation message
zoltan.cpp: In function 'std::vector<std::pair<int, long int> > lis(std::vector<int>&)':
zoltan.cpp:55:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
55 | for (int i = 0; i < v.size(); i++) {
| ~~^~~~~~~~~~
zoltan.cpp:60:9: warning: unused variable 'used' [-Wunused-variable]
60 | int used = pow(2, v.size() - 1);
| ^~~~
zoltan.cpp: In function 'std::vector<std::pair<int, long int> > lds(std::vector<int>&)':
zoltan.cpp:85:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
85 | for (int i = 0; i < st.vec.size(); i++) {
| ~~^~~~~~~~~~~~~~~
zoltan.cpp:89:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
89 | for (int i = 0; i < v.size(); i++) {
| ~~^~~~~~~~~~
zoltan.cpp: In function 'int main()':
zoltan.cpp:128:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
128 | for (int i = 0; i < a1.size(); i++) {
| ~~^~~~~~~~~~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
0 ms |
212 KB |
Output is correct |
| 2 |
Correct |
0 ms |
212 KB |
Output is correct |
| 3 |
Correct |
0 ms |
212 KB |
Output is correct |
| 4 |
Correct |
0 ms |
212 KB |
Output is correct |
| 5 |
Correct |
0 ms |
212 KB |
Output is correct |
| 6 |
Correct |
0 ms |
212 KB |
Output is correct |
| 7 |
Correct |
2 ms |
468 KB |
Output is correct |
| 8 |
Correct |
2 ms |
468 KB |
Output is correct |
| 9 |
Correct |
2 ms |
468 KB |
Output is correct |
| 10 |
Correct |
2 ms |
468 KB |
Output is correct |
| 11 |
Runtime error |
366 ms |
35000 KB |
Memory limit exceeded |
| 12 |
Correct |
348 ms |
32436 KB |
Output is correct |
| 13 |
Correct |
267 ms |
24172 KB |
Output is correct |
| 14 |
Correct |
460 ms |
32384 KB |
Output is correct |
| 15 |
Runtime error |
604 ms |
36740 KB |
Memory limit exceeded |
| 16 |
Runtime error |
662 ms |
40264 KB |
Memory limit exceeded |
| 17 |
Runtime error |
473 ms |
36404 KB |
Memory limit exceeded |
| 18 |
Runtime error |
466 ms |
36392 KB |
Memory limit exceeded |
| 19 |
Runtime error |
480 ms |
36480 KB |
Memory limit exceeded |
| 20 |
Runtime error |
449 ms |
36376 KB |
Memory limit exceeded |
