#include <iostream> #include <array> #include <algorithm> std::array <int, 1'000'000> val; std::array <int, 1'000'000> poz; int main() { int n; std::cin >> n; std::cout << 2*n + 1 << " "; int n1 = n - 1; for (int i = 0; i < n; ++i) { int a; std::cin >> a; --a; val[i] = a; poz[a] = i; } uint64_t total = 1; int l = poz[n1], r = poz[n1]; int poprz_miotla = 1; for (int i = n - 2; i >= 0; --i) { //std::cout << "[PRE " << i << "] l/r " << l << "-" << r << " pm " << poprz_miotla // << " total " << total << std::endl; int p = poz[i]; if (p < l) l = p; else if (p > r) r = p; int curr_przedzial = r - l + 1; int max_miotla = 2*(n1 - i) + 1; //std::cout << "[MID " << i << "] l/r " << l << "-" << r << " cp " << curr_przedzial << " mm " << max_miotla << std::endl; for (int miotla = poprz_miotla + 1; miotla <= max_miotla; ++miotla) { //std::cout << "[MIOT " << miotla << "] "; if (miotla < curr_przedzial) { //std::cout << "lipper" << std::endl; continue; } int max_lewa = std::max(0, r - miotla + 1); int max_prawa = std::min(n1, l + miotla - 1); int tplus = max_prawa - max_lewa - miotla + 2; total += tplus; //std::cout << max_lewa << "-" << max_prawa << " aka " << tplus << " miotel" << std::endl; if (miotla == n) { i = 0; //std::cout << " also spierdalam bo duza miotla xd" << std::endl; break; } } poprz_miotla = max_miotla; //std::cout << "[POST " << i << "] total " << total << std::endl; } std::cout << total << std::endl; }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 | #include <iostream> #include <array> #include <algorithm> std::array <int, 1'000'000> val; std::array <int, 1'000'000> poz; int main() { int n; std::cin >> n; std::cout << 2*n + 1 << " "; int n1 = n - 1; for (int i = 0; i < n; ++i) { int a; std::cin >> a; --a; val[i] = a; poz[a] = i; } uint64_t total = 1; int l = poz[n1], r = poz[n1]; int poprz_miotla = 1; for (int i = n - 2; i >= 0; --i) { //std::cout << "[PRE " << i << "] l/r " << l << "-" << r << " pm " << poprz_miotla // << " total " << total << std::endl; int p = poz[i]; if (p < l) l = p; else if (p > r) r = p; int curr_przedzial = r - l + 1; int max_miotla = 2*(n1 - i) + 1; //std::cout << "[MID " << i << "] l/r " << l << "-" << r << " cp " << curr_przedzial << " mm " << max_miotla << std::endl; for (int miotla = poprz_miotla + 1; miotla <= max_miotla; ++miotla) { //std::cout << "[MIOT " << miotla << "] "; if (miotla < curr_przedzial) { //std::cout << "lipper" << std::endl; continue; } int max_lewa = std::max(0, r - miotla + 1); int max_prawa = std::min(n1, l + miotla - 1); int tplus = max_prawa - max_lewa - miotla + 2; total += tplus; //std::cout << max_lewa << "-" << max_prawa << " aka " << tplus << " miotel" << std::endl; if (miotla == n) { i = 0; //std::cout << " also spierdalam bo duza miotla xd" << std::endl; break; } } poprz_miotla = max_miotla; //std::cout << "[POST " << i << "] total " << total << std::endl; } std::cout << total << std::endl; } |