Submission #476073

#TimeUsernameProblemLanguageResultExecution timeMemory
476073arwaeystoamnegGroup Photo (JOI21_ho_t3)C++17
100 / 100
2732 ms728 KiB
// EXPLOSION! #define _CRT_SECURE_NO_WARNINGS #include<bits/stdc++.h> #include<unordered_set> #include<unordered_map> #include<chrono> using namespace std; typedef pair<int, int> pii; typedef long long ll; typedef pair<ll, ll> pll; typedef long double ld; typedef vector<int> vi; typedef vector<ll> vll; typedef vector<pair<int, int>> vpi; typedef vector<pair<ll, ll>> vpll; #define FOR(i,a,b) for (int i = (a); i < (b); ++i) #define F0R(i,a) FOR(i,0,a) #define ROF(i,a,b) for (int i = (b)-1; i >= (a); --i) #define R0F(i,a) ROF(i,0,a) #define trav(a,x) for (auto& a: x) #define pb push_back #define mp make_pair #define rsz resize #define sz(x) int(x.size()) #define all(x) x.begin(),x.end() #define f first #define s second #define cont continue #define endl '\n' //#define ednl '\n' #define test int testc;cin>>testc;while(testc--) #define pr(a, b) trav(x,a)cerr << x << b; cerr << endl; #define message cout << "Hello World" << endl; const int dx[4] = { 1,0,-1,0 }, dy[4] = { 0,1,0,-1 }; // for every grid problem!! const ll linf = 4000000000000000000LL; const ll inf = 998244353;// 1000000007 void pv(vi a) { trav(x, a)cout << x << " "; cout << endl; }void pv(vll a) { trav(x, a)cout << x << " "; cout << endl; }void pv(vector<vi>a) { F0R(i, sz(a)) { cout << i << endl; pv(a[i]); cout << endl; } }void pv(vector<vll>a) { F0R(i, sz(a)) { cout << i << endl; pv(a[i]); }cout << endl; }void pv(vector<string>a) { trav(x, a)cout << x << endl; cout << endl; } void setIO(string s) { ios_base::sync_with_stdio(0); cin.tie(0); #ifdef arwaeystoamneg if (sz(s)) { freopen((s + ".in").c_str(), "r", stdin); if (s != "test1") freopen((s + ".out").c_str(), "w", stdout); } #endif } template<class T>class segment_tree { struct item { T sum; }; item single(T i) { return { i }; } item merge(item x, item y) { item ans; ans.sum = x.sum + y.sum; return ans; } vector<item> tree; vector<item>A; int height; item neutral = { 0 }; public:void build(vector<T>& B) { int n = B.size(); height = log2(n + 1) + 1; A.rsz(n); tree.rsz((1 << height + 1) - 1); F0R(i, n)A[i] = single(B[i]); A.rsz(1 << height, neutral); build(A, 0, 0, (1 << height) - 1); } void build(vector<item>& A, int v, int tl, int tr) { if (tl == tr) tree[v] = A[tl]; else { int mid = (tl + tr) / 2; build(A, 2 * v + 1, tl, mid); build(A, 2 * v + 2, mid + 1, tr); tree[v] = merge(tree[2 * v + 1], tree[2 * v + 2]); } } public:T query(int l, int r) { return query(0, 0, (1 << height) - 1, l, r).sum; } item query(int v, int tl, int tr, int l, int r) { if (r<tl || l>tr)return neutral; if (l <= tl && r >= tr) { return tree[v]; } int mid = (tl + tr) / 2; return merge(query(2 * v + 1, tl, mid, l, r), query(2 * v + 2, mid + 1, tr, l, r)); } //assign public:void update(int pos, T val) { update(0, 0, (1 << height) - 1, pos, single(val)); } void update(int v, int tl, int tr, int pos, item val) { if (tl == tr) { tree[v] = val; } else { int mid = (tl + tr) / 2; if (pos <= mid) update(2 * v + 1, tl, mid, pos, val); else update(2 * v + 2, mid + 1, tr, pos, val); tree[v] = merge(tree[2 * v + 1], tree[2 * v + 2]); } } public:int get(int k) { return get(0, 0, (1 << height) - 1, k); } int get(int v, int tl, int tr, int k) { if (tl == tr)return tl; int mid = (tl + tr) / 2; if (k > tree[2 * v + 1].sum) return get(2 * v + 2, mid + 1, tr, k - tree[2 * v + 1].sum); return get(2 * v + 1, tl, mid, k); } }; int n; vi a, dp; int main() { setIO("test1"); cin >> n; a.rsz(n); trav(x, a)cin >> x, x--; dp.rsz(n + 1, inf); dp[n] = 0; R0F(i, n + 1) { int val = dp[i]; vi b; trav(x, a)if (x < i)b.pb(x); vi idx(i); F0R(j, i)idx[b[j]] = j; vi t(i); segment_tree<int>sum; sum.build(t); R0F(j, i) { val += i - idx[j] - 1; val -= sum.query(0, idx[j]); sum.update(idx[j], 1); dp[j] = min(dp[j], val); } } //trav(x, dp)cerr << x << " "; //cerr << endl; cout << dp[0] << endl; }

Compilation message (stderr)

Main.cpp: In instantiation of 'void segment_tree<T>::build(std::vector<_Tp>&) [with T = int]':
Main.cpp:166:14:   required from here
Main.cpp:81:24: warning: suggest parentheses around '+' inside '<<' [-Wparentheses]
   81 |  tree.rsz((1 << height + 1) - 1);
      |                 ~~~~~~~^~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...