結果
問題 | No.778 クリスマスツリー |
ユーザー | 37zigen |
提出日時 | 2020-03-25 15:46:13 |
言語 | Java21 (openjdk 21) |
結果 |
AC
|
実行時間 | 612 ms / 2,000 ms |
コード長 | 3,787 bytes |
コンパイル時間 | 2,640 ms |
コンパイル使用メモリ | 76,908 KB |
実行使用メモリ | 125,232 KB |
最終ジャッジ日時 | 2023-08-30 11:41:22 |
合計ジャッジ時間 | 9,342 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 45 ms
49,672 KB |
testcase_01 | AC | 45 ms
49,448 KB |
testcase_02 | AC | 45 ms
49,340 KB |
testcase_03 | AC | 45 ms
49,360 KB |
testcase_04 | AC | 45 ms
49,312 KB |
testcase_05 | AC | 45 ms
49,340 KB |
testcase_06 | AC | 398 ms
124,936 KB |
testcase_07 | AC | 342 ms
84,408 KB |
testcase_08 | AC | 612 ms
96,312 KB |
testcase_09 | AC | 593 ms
88,160 KB |
testcase_10 | AC | 586 ms
87,684 KB |
testcase_11 | AC | 591 ms
87,804 KB |
testcase_12 | AC | 545 ms
87,880 KB |
testcase_13 | AC | 359 ms
88,148 KB |
testcase_14 | AC | 377 ms
125,232 KB |
ソースコード
import java.io.FileNotFoundException; import java.io.IOException; import java.io.InputStream; import java.util.ArrayList; import java.util.Arrays; import java.util.NoSuchElementException; public class Main { public static void main(String[] args) throws FileNotFoundException { long t = System.currentTimeMillis(); new Main().run(); System.err.println(System.currentTimeMillis() - t); } class SegTree{ int n=1; int[] v; public SegTree(int n) { while(this.n<n)this.n*=2; v=new int[2*this.n-1]; } void add(int k,int val) { k+=n-1; v[k]+=val; while(k>0) { k=(k-1)/2; v[k]=v[2*k+1]+v[2*k+2]; } } int query(int a,int b) { return query(0,n,a,b,0); } int query(int l,int r,int a,int b,int k) { if(a<=l&&r<=b)return v[k]; else if(r<=a||b<=l)return 0; else { int vl=query(l,(l+r)/2,a,b,2*k+1); int vr=query((l+r)/2,r,a,b,2*k+2); return vl+vr; } } } int pointer=0; void dfs(int cur,int par,int[] left,int[] right,ArrayList<Integer>[] g) { left[cur]=pointer++; for(int dst:g[cur]) { if(dst==par)continue; dfs(dst,cur,left,right,g); } right[cur]=pointer++; } void run() { FastScanner sc = new FastScanner(); int N=sc.nextInt(); SegTree seg=new SegTree(2*N); ArrayList<Integer>[] g=new ArrayList[N]; for(int i=0;i<N;++i)g[i]=new ArrayList<>(); for(int i=0;i<N-1;++i) { int parent=sc.nextInt(); g[parent].add(i+1); g[i+1].add(parent); } int[] l=new int[N]; int[] r=new int[N]; Arrays.fill(l, -1); Arrays.fill(r, -1); dfs(0,-1,l,r,g); long ans=0; for(int i=N-1;i>=0;--i) { ans+=seg.query(l[i], r[i]); seg.add(l[i], 1); seg.add(r[i], 1); } System.out.println(ans/2); } static void tr(Object... objects) { System.out.println(Arrays.deepToString(objects)); } } class FastScanner { private final InputStream in = System.in; private final byte[] buffer = new byte[1024]; private int ptr = 0; private int buflen = 0; private boolean hasNextByte() { if (ptr < buflen) { return true; }else{ ptr = 0; try { buflen = in.read(buffer); } catch (IOException e) { e.printStackTrace(); } if (buflen <= 0) { return false; } } return true; } private int readByte() { if (hasNextByte()) return buffer[ptr++]; else return -1;} private static boolean isPrintableChar(int c) { return 33 <= c && c <= 126;} private void skipUnprintable() { while(hasNextByte() && !isPrintableChar(buffer[ptr])) ptr++;} public boolean hasNext() { skipUnprintable(); return hasNextByte();} public String next() { if (!hasNext()) throw new NoSuchElementException(); StringBuilder sb = new StringBuilder(); int b = readByte(); while(isPrintableChar(b)) { sb.appendCodePoint(b); b = readByte(); } return sb.toString(); } public long nextLong() { if (!hasNext()) throw new NoSuchElementException(); long n = 0; boolean minus = false; int b = readByte(); if (b == '-') { minus = true; b = readByte(); } if (b < '0' || '9' < b) { throw new NumberFormatException(); } while(true){ if ('0' <= b && b <= '9') { n *= 10; n += b - '0'; }else if(b == -1 || !isPrintableChar(b)){ return minus ? -n : n; }else{ throw new NumberFormatException(); } b = readByte(); } } public int nextInt() { return (int)nextLong(); } }