結果
問題 | No.778 クリスマスツリー |
ユーザー |
![]() |
提出日時 | 2018-10-26 01:59:12 |
言語 | C++11 (gcc 13.3.0) |
結果 |
AC
|
実行時間 | 161 ms / 2,000 ms |
コード長 | 1,638 bytes |
コンパイル時間 | 1,164 ms |
コンパイル使用メモリ | 95,512 KB |
実行使用メモリ | 34,432 KB |
最終ジャッジ日時 | 2024-10-14 01:26:01 |
合計ジャッジ時間 | 4,039 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 12 |
ソースコード
#include <algorithm>#include <cfloat>#include <climits>#include <cmath>#include <complex>#include <cstdio>#include <cstdlib>#include <cstring>#include <functional>#include <iostream>#include <map>#include <memory>#include <queue>#include <set>#include <sstream>#include <stack>#include <string>#include <utility>#include <vector>using namespace std;typedef long long ll;#define sz size()#define pb push_back#define mp make_pair#define fi first#define se second#define all(c) (c).begin(), (c).end()#define rep(i,a,b) for(ll i=(a);i<(b);++i)#define per(i,a,b) for(ll i=b-1LL;i>=(a);--i)#define clr(a, b) memset((a), (b) ,sizeof(a))#define ctos(c) string(1,c)#define print(x) cout<<#x<<" = "<<x<<endl;#define MOD 1000000007#define N_MAX 400100struct bit{long long dat[N_MAX + 1];void init(){for(int i = 0; i < N_MAX; i++){dat[i] = 0;}}long long sum(int i){long long s = 0;while(i > 0){s += dat[i];i -= i & -i;}return s;}void add(int i, long long x){while(i <= N_MAX){dat[i] += x;i += i & -i;}}}bit1;ll index1;ll dl[200100];ll dr[200100];vector<vector<ll> > vv;void f(ll p){dl[p] = index1;index1++;rep(i,0,vv[p].sz){f(vv[p][i]);}dr[p] = index1;index1++;}int main() {ll n;cin>>n;bit1.init();vector<vector<ll> > vv_(n,vector<ll>());vv = vv_;rep(i,1,n){ll a;cin>>a;vv[a].pb(i);}index1 = 1;f(0);ll ans = 0;per(i,0,n){ans += bit1.sum(dr[i])-bit1.sum(dl[i]);bit1.add(dl[i],1);}cout << ans << endl;return 0;}