結果

問題 No.778 クリスマスツリー
ユーザー fumiphys
提出日時 2019-03-09 20:55:49
言語 C++11
(gcc 13.3.0)
結果
AC  
実行時間 72 ms / 2,000 ms
コード長 2,195 bytes
コンパイル時間 849 ms
コンパイル使用メモリ 105,356 KB
実行使用メモリ 29,860 KB
最終ジャッジ日時 2024-10-14 01:58:35
合計ジャッジ時間 2,028 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

// includes
#include <cstdio>
#include <cstdint>
#include <iostream>
#include <iomanip>
#include <string>
#include <queue>
#include <stack>
#include <vector>
#include <set>
#include <map>
#include <unordered_map>
#include <algorithm>
#include <utility>
#include <functional>
#include <cmath>
#include <climits>
#include <bitset>
#include <list>
#include <random>
// macros
#define ll long long int
#define pb emplace_back
#define mk make_pair
#define pq priority_queue
#define FOR(i, a, b) for(int i=(a);i<(b);++i)
#define rep(i, n) FOR(i, 0, n)
#define rrep(i, n) for(int i=((int)(n)-1);i>=0;i--)
#define all(x) (x).begin(),(x).end()
#define sz(x) ((int)(x).size())
#define UNIQUE(v) v.erase(unique(v.begin(), v.end()), v.end())
using namespace std;
// types
typedef pair<int, int> P;
typedef pair<ll, int> Pl;
typedef pair<ll, ll> Pll;
typedef pair<double, double> Pd;
// constants
const int inf = 1e9;
const ll linf = 1LL << 50;
const double EPS = 1e-10;
const int mod = 1e9 + 7;
// solve
template <class T>bool chmax(T &a, const T &b){if(a < b){a = b; return 1;} return 0;}
template <class T>bool chmin(T &a, const T &b){if(a > b){a = b; return 1;} return 0;}
vector<int> child[200001];
vector<int> eu;
int cs[200001];
template<typename T>
struct BIT_{
int n;
vector<T> bit;
BIT_(int n_){
n = n_;
bit = vector<T>(n+1, 0);
}
void add(int a, T w){
for(int x = a; x < n; x |= (x + 1)){
bit[x] += w;
}
}
T sum(int a){
T res = 0;
for(int x = a; x >= 0; x = (x & (x + 1)) - 1){
res += bit[x];
}
return res;
}
};
typedef struct BIT_<int> BITI;
typedef struct BIT_<long long> BITL;
int dfs(int i){
int res = 0;
eu.pb(i);
for(auto c: child[i]){
res += dfs(c);
}
cs[i] = res;
return res + 1;
}
int main(int argc, char const* argv[])
{
ios_base::sync_with_stdio(false);
cin.tie(0);
int n;
cin >> n;
rep(i, n - 1){
int x;
cin >> x;
child[x].pb(i+1);
}
dfs(0);
vector<int> tr(n, 0);
rep(i, n)tr[eu[i]] = i;
BITI b(n);
ll res = 0;
for(int i = n - 1; i >= 0; i--){
res += b.sum(tr[i]+cs[i]);
res -= b.sum(tr[i]);
b.add(tr[i], 1);
}
cout << res << endl;
return 0;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0