結果

問題 No.1418 Sum of Sum of Subtree Size
ユーザー fgwiebfaoishfgwiebfaoish
提出日時 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.

ソースコード

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

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;}
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0