結果
問題 | No.1418 Sum of Sum of Subtree Size |
ユーザー |
![]() |
提出日時 | 2021-03-05 22:57:52 |
言語 | C#(csc) (csc 3.9.0) |
結果 |
AC
|
実行時間 | 246 ms / 2,000 ms |
コード長 | 3,060 bytes |
コンパイル時間 | 2,739 ms |
コンパイル使用メモリ | 111,228 KB |
実行使用メモリ | 47,220 KB |
最終ジャッジ日時 | 2024-10-07 04:20:16 |
合計ジャッジ時間 | 7,216 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
コンパイルメッセージ
Microsoft (R) Visual C# Compiler version 3.9.0-6.21124.20 (db94f4cc) Copyright (C) Microsoft Corporation. All rights reserved.
ソースコード
using System;using System.Collections.Generic;using System.Collections;using System.Collections.Specialized;using System.Linq;using System.Text;using System.IO;using System.Reflection;using static System.Math;using System.Numerics;using System.Threading;using System.Runtime.CompilerServices;static class Program{const int mod=(int)1e9+7;static List<int>[] li;static bool[] b;static int[] pa;static int n;static long ans=0;static long[] dp1,dp2,dp3;static void Main(){Sc sc=new Sc();var s=sc.Ia;n=s[0];int m=n-1;li=new List<int>[n+1];b=new bool[n+1];pa=new int[n+1];dp1=new long[n+1];dp2=new long[n+1];dp3=new long[n+1];for(int i=1;i<=n;i++){li[i]=new List<int>();}for(int i=0;i<m;i++){var e=sc.Ia;li[e[0]].Add(e[1]);li[e[1]].Add(e[0]);}Fu(1,0);Fu2(1);Console.WriteLine("{0}",n*(long)n+ans);}static void Fu2(int a) {//Console.WriteLine((a,n-dp1[a]));for(int i=0;i<li[a].Count;i++){if(li[a][i]!=pa[a]){Fu2(li[a][i]);ans+=(n-dp1[li[a][i]])*dp1[li[a][i]];}}}static void Fu(int a,int g){b[a]=true;pa[a]=g;for(int i=0;i<li[a].Count;i++){if(!b[li[a][i]]){Fu(li[a][i],a);dp1[a]+=dp1[li[a][i]];//dp2[a]+=dp2[li[a][i]];}}dp1[a]++;//dp2[a]+=dp1[a];ans+=(n-dp1[a])*dp1[a];}}public class Sc{public int I{get{return int.Parse(Console.ReadLine());}}public long L{get{return long.Parse(Console.ReadLine());}}public double D{get{return double.Parse(Console.ReadLine());}}public string S{get{return Console.ReadLine();}}public int[] Ia{get{return Array.ConvertAll(Console.ReadLine().Split(),int.Parse);}}public long[] La{get{return Array.ConvertAll(Console.ReadLine().Split(),long.Parse);}}public double[] Da{get{return Array.ConvertAll(Console.ReadLine().Split(),double.Parse);}}public string[] Sa{get{return Console.ReadLine().Split();}}public object[] Oa{get{return Console.ReadLine().Split();}}public int[] Ia2{get{return Array.ConvertAll(("0 "+Console.ReadLine()+" 0").Split(),int.Parse);}}public int[] Ia3(string a,string b){return Array.ConvertAll((a+Console.ReadLine()+b).Split(),int.Parse);}public long[] La2{get{return Array.ConvertAll(("0 "+Console.ReadLine()+" 0").Split(),long.Parse);}}public long[] La3(string a,string b){return Array.ConvertAll((a+Console.ReadLine()+b).Split(),long.Parse);}public double[] Da2{get{return Array.ConvertAll(("0 "+Console.ReadLine()+" 0").Split(),double.Parse);}}public double[] Da3(string a,string b){return Array.ConvertAll((a+Console.ReadLine()+b).Split(),double.Parse);}public T[] Arr<T>(int n,Func<T> f){var a=new T[n];for(int i=0;i<n;i++){a[i]=f();}return a;}public T[] Arr<T>(int n,Func<int,T> f){var a=new T[n];for(int i=0;i<n;i++){a[i]=f(i);}return a;}public T[] Arr<T>(int n,Func<string[],T> f){var a=new T[n];for(int i=0;i<n;i++){a[i]=f(Console.ReadLine().Split());}return a;}public T[] Arr<T>(int n,Func<int,string[],T> f){var a=new T[n];for(int i=0;i<n;i++){a[i]=f(i,Console.ReadLine().Split());}return a;}}