QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#310021 | #8134. LCA Counting | ucup-team197# | WA | 2ms | 12268kb | C++17 | 2.1kb | 2024-01-20 23:58:13 | 2024-01-20 23:58:13 |
Judging History
answer
#include<bits/stdc++.h>
using namespace std ;
typedef long long ll ;
typedef unsigned long long ull ;
typedef pair < int , int > pii ;
typedef vector < int > vi ;
#define fi first
#define se second
mt19937 rng(chrono::high_resolution_clock::now().time_since_epoch().count());
#define rep(i, a, b) for(int i = a; i < (b); ++i)
#define all(x) begin(x), end(x)
#define sz(x) (int)(x).size()
const int MAXN = 2e5 + 7 ;
const int LOG = 20 ;
int n , k ;
vector < int > v[ MAXN ] ;
int dp[ MAXN ] , nxt[ MAXN ][ 2 ] ;
int prv[ MAXN ] ;
void dfs ( int x ) {
vector < pii > srt ;
for ( auto y : v[ x ] ) {
dfs ( y ) ;
srt.push_back ( { dp[ y ] , y } ) ;
}
int sz = v[ x ].size ( ) ;
if ( sz == 0 ) {
++ k ;
return ;
}
if ( sz == 1 ) {
dp[ x ] = srt[ 0 ].fi ;
return ;
}
sort ( srt.begin ( ) , srt.end ( ) , greater < pii > ( ) ) ;
dp[ x ] = srt[ 0 ].fi + srt[ 1 ].fi ;
nxt[ x ][ 0 ] = srt[ 0 ].se , nxt[ x ][ 1 ] = srt[ 1 ].se ;
++ dp[ x ] ;
}
int ans[ MAXN ] ;
void solve ( ) {
cin >> n ;
for ( int i = 2 , x ; i <= n ; ++ i ) {
cin >> x ;
prv[ i ] = x ;
v[ x ].push_back ( i ) ;
}
dfs ( 1 ) ;
vector < int > srt ;
for ( int i = 1 ; i <= n ; ++ i ) {
if ( nxt[ prv[ i ] ][ 0 ] != i && nxt[ prv[ i ] ][ 1 ] != i ) {
srt.push_back ( dp[ i ] ) ;
}
}
sort ( srt.begin ( ) , srt.end ( ) ) ;
int tp = 1 ;
ans[ 1 ] = 0 ;
while ( srt.empty ( ) == false ) {
int aux = srt.back ( ) ;
srt.pop_back ( ) ;
while ( aux > 0 && tp < k ) {
++ tp , -- aux ;
ans[ tp ] = ans[ tp - 1 ] + 1 ;
}
++ tp ;
ans[ tp ] = ans[ tp - 1 ] ;
}
while ( tp < k ) {
++ tp ;
ans[ tp ] = ans[ tp - 1 ] ;
}
for ( int i = 1 ; i <= k ; ++ i ) {
ans[ i ] += i ;
printf ( "%d " , ans[ i ] ) ;
}
printf ( "\n" ) ;
}
int main ( ) {
ios_base :: sync_with_stdio ( false ) ;
cin.tie ( NULL ) ;
int t = 1 ; // cin >> t ;
while ( t -- ) { solve ( ) ; }
return 0 ;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 11608kb
input:
7 1 1 2 4 2 2
output:
1 3 5 6
result:
ok 4 number(s): "1 3 5 6"
Test #2:
score: 0
Accepted
time: 1ms
memory: 8684kb
input:
10 1 1 2 2 1 1 1 2 4
output:
1 3 5 6 7 8 9
result:
ok 7 numbers
Test #3:
score: 0
Accepted
time: 1ms
memory: 9072kb
input:
10 1 2 2 4 5 6 1 2 4
output:
1 3 5 7 8
result:
ok 5 number(s): "1 3 5 7 8"
Test #4:
score: 0
Accepted
time: 1ms
memory: 8848kb
input:
10 1 2 3 3 3 5 6 4 9
output:
1 3 4
result:
ok 3 number(s): "1 3 4"
Test #5:
score: 0
Accepted
time: 2ms
memory: 12196kb
input:
1000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 2 1 2 1 1 1 1 1 1 2 1 1 1 1 1 1 1 1 1 1 4 1 2 2 1 1 1 1 2 1 1 1 1 2 2 1 1 1 1 1 6 3 1 1 1 2 2 1 1 2 1 2 1 3 1 2 1 1 2 1 1 1 1 1 1 1 4 1 5 1 1 1 1 1 2 1 1 2 1 2 1 2 5 3 1 3 1 1 2 1 2 1 1 3 2 1 6 2 1 2 5 1 1 1 3 2 1 2 1 1 1 1 1...
output:
1 3 5 7 8 10 11 13 14 16 17 19 20 22 23 25 26 28 29 31 32 34 35 37 38 40 41 43 44 46 47 49 50 52 53 55 56 58 59 61 62 64 65 67 68 70 71 73 74 76 77 79 80 82 83 85 86 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 12...
result:
ok 962 numbers
Test #6:
score: 0
Accepted
time: 0ms
memory: 12268kb
input:
1000 1 1 1 1 1 1 1 2 2 1 2 3 1 1 1 2 2 1 1 1 2 1 2 3 8 2 2 2 8 1 6 3 1 1 1 4 2 14 8 2 1 1 1 3 1 4 1 1 3 11 1 1 2 2 3 2 5 5 6 5 3 3 10 5 3 1 9 14 3 19 4 15 4 3 3 10 3 10 5 1 2 12 7 13 3 9 6 19 9 6 4 3 5 28 10 16 8 12 39 1 11 12 2 2 10 15 5 28 10 4 4 5 5 18 15 15 11 5 25 7 2 10 1 24 8 15 14 11 38 9 9 ...
output:
1 3 5 7 9 11 13 15 17 19 21 23 25 27 28 30 32 34 36 37 39 41 43 44 46 48 50 51 53 55 57 58 60 62 64 65 67 69 71 72 74 76 78 79 81 83 85 86 88 90 92 93 95 97 99 100 102 104 105 107 109 110 112 114 115 117 119 120 122 124 125 127 129 130 132 133 135 136 138 139 141 142 144 145 147 148 150 151 153 154 ...
result:
ok 822 numbers
Test #7:
score: -100
Wrong Answer
time: 0ms
memory: 9236kb
input:
1000 1 2 2 2 1 4 3 6 7 5 8 8 1 12 12 16 4 16 18 3 17 22 7 22 9 17 21 16 16 18 29 31 12 29 32 36 22 31 32 14 15 13 8 1 36 19 33 2 10 28 29 1 53 27 15 56 34 57 21 15 58 36 57 57 16 45 16 51 44 18 31 52 53 34 65 30 51 8 59 72 53 49 49 35 31 23 3 19 69 5 16 11 45 12 62 23 46 5 24 61 63 81 1 85 73 53 84 ...
output:
1 3 5 7 9 11 13 15 17 19 21 23 25 27 29 31 33 35 37 39 41 43 45 47 49 51 53 55 57 59 61 63 65 67 69 71 73 75 77 79 81 83 85 87 89 91 93 95 97 99 101 103 105 107 109 111 113 115 117 119 121 123 125 127 129 131 133 135 137 139 141 143 145 147 149 151 153 155 157 159 161 163 165 167 169 171 173 175 177...
result:
wrong answer 189th numbers differ - expected: '371', found: '372'