제출 #1299599

#제출 시각아이디문제언어결과실행 시간메모리
1299599Zbyszek99Construction of Highway (JOI18_construction)C++20
0 / 100
29 ms67760 KiB
#include <bits/stdc++.h> #include <ext/pb_ds/assoc_container.hpp> #include <ext/pb_ds/tree_policy.hpp> #define ll long long #define ld long double #define ull unsigned long long #define ff first #define ss second #define pii pair<int,int> #define pll pair<long long, long long> #define vi vector<int> #define vl vector<long long> #define pb push_back #define rep(i, b) for(int i = 0; i < (b); ++i) #define rep2(i,a,b) for(int i = a; i <= (b); ++i) #define rep3(i,a,b,c) for(int i = a; i <= (b); i+=c) #define count_bits(x) __builtin_popcountll((x)) #define all(x) (x).begin(),(x).end() #define siz(x) (int)(x).size() #define forall(it,x) for(auto& it:(x)) using namespace __gnu_pbds; using namespace std; typedef tree<int, null_type, less<int>, rb_tree_tag,tree_order_statistics_node_update> ordered_set; //mt19937 mt;void random_start(){mt.seed(chrono::time_point_cast<chrono::milliseconds>(chrono::high_resolution_clock::now()).time_since_epoch().count());} //ll los(ll a, ll b) {return a + (mt() % (b-a+1));} const int INF = 1e9+50; const ll INF_L = 1e18+40; const ll MOD = 1e9+7; const int tree_siz = 1024*256-1; ll sum[tree_siz+1]; ll get_sum(int akt, int p1, int p2, int s1, int s2) { if(p2 < s1 || p1 > s2) return 0; if(p1 >= s1 && p2 <= s2) return sum[akt]; return get_sum(akt*2,p1,(p1+p2)/2,s1,s2)+get_sum(akt*2+1,(p1+p2)/2+1,p2,s1,s2); } void upd(int v) { sum[v] = sum[v*2]+sum[v*2+1]; if(v != 1) upd(v/2); } void change(int ind, int val) { sum[tree_siz/2+ind+1] += val; upd((tree_siz/2+1+ind)/2); } vi graph[100001]; int val[100001]; int path[100001]; int path_end[100001]; stack<pii> path_events[100001]; int P[100001]; int pre[100001]; int maxpre[100001]; int siz_[100001]; int depth[100001]; int cur_pre; int cur_path; void dfs_siz(int v, int d = 0) { siz_[v] = 1; pre[v] = cur_pre++; maxpre[v] = pre[v]; depth[v] = d++; forall(it,graph[v]) { dfs_siz(it,d); siz_[v] += siz_[it]; maxpre[v] = maxpre[it]; } } void dfs_HLD(int v, int p) { path[v] = p; pii best = {-1e9,-1}; forall(it,graph[v]) best = max(best,{siz_[it],it}); if(best.ss != -1) { dfs_HLD(best.ss,p); } forall(it,graph[v]) if(it != best.ss) { path_end[cur_path] = it; dfs_HLD(it,cur_path++); } } ll count_inv(vector<pii>& T) { ll ans = 0; forall(it,T) { ans += get_sum(1,0,tree_siz/2,it.ff+1,tree_siz/2)*(ll)it.ss; change(it.ff,it.ss); } forall(it,T) change(it.ff,-it.ss); return ans; } vector<pii> pom; void query(int v) { int v2 = v; v = P[v]; vector<pii> ans; int cnt = 0; while(v != 0) { int pop = P[path_end[path[v]]]; while(!path_events[path[v]].empty()) { if(depth[v] > depth[path_events[path[v]].top().ss]) { pom.pb({path_events[path[v]].top().ff,depth[path_events[path[v]].top().ss]-depth[pop]}); pop = path_events[path[v]].top().ss; path_events[path[v]].pop(); } else { pom.pb({path_events[path[v]].top().ff,depth[v]-depth[pop]}); if(depth[v] == depth[path_events[path[v]].top().ss]) path_events[path[v]].pop(); break; } } if(cnt != 0) path_events[path[v]].push({val[v2],v}); else { if(path[v] == path[v2]) path_events[path[v]].push({val[v2],v2}); else { path_events[path[v]].push({val[v2],v}); path_events[path[v2]].push({val[v2],v2}); } } reverse(all(pom)); forall(it,pom) ans.pb(it); pom = {}; v = P[path_end[path[v]]]; cnt++; } reverse(all(ans)); cout << count_inv(ans) << "\n"; } int main() { //ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0); //random_start(); int n; cin >> n; rep2(i,1,n) cin >> val[i]; vi q = {}; P[1] = 0; rep(i,n-1) { int a,b; cin >> a >> b; graph[a].pb(b); P[b] = a; q.pb(b); } depth[0] = -1; dfs_siz(1); cur_path = 1; path_end[0] = 1; dfs_HLD(1,0); path_events[0].push({val[1],1}); forall(it,q) query(it); }
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...