結果

問題 No.1333 Squared Sum
ユーザー 👑 NachiaNachia
提出日時 2021-01-09 12:32:14
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 172 ms / 2,000 ms
コード長 1,396 bytes
コンパイル時間 1,905 ms
コンパイル使用メモリ 207,912 KB
実行使用メモリ 27,488 KB
最終ジャッジ日時 2024-11-17 15:08:01
合計ジャッジ時間 7,163 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 ms
16,156 KB
testcase_01 AC 4 ms
16,156 KB
testcase_02 AC 4 ms
16,152 KB
testcase_03 AC 126 ms
27,088 KB
testcase_04 AC 123 ms
27,092 KB
testcase_05 AC 123 ms
27,092 KB
testcase_06 AC 125 ms
27,092 KB
testcase_07 AC 125 ms
27,092 KB
testcase_08 AC 123 ms
27,220 KB
testcase_09 AC 124 ms
27,216 KB
testcase_10 AC 123 ms
27,212 KB
testcase_11 AC 125 ms
27,088 KB
testcase_12 AC 124 ms
27,088 KB
testcase_13 AC 108 ms
27,480 KB
testcase_14 AC 132 ms
27,356 KB
testcase_15 AC 132 ms
27,488 KB
testcase_16 AC 5 ms
16,156 KB
testcase_17 AC 4 ms
16,148 KB
testcase_18 AC 4 ms
16,152 KB
testcase_19 AC 4 ms
16,152 KB
testcase_20 AC 4 ms
16,152 KB
testcase_21 AC 3 ms
16,148 KB
testcase_22 AC 4 ms
16,152 KB
testcase_23 AC 3 ms
16,152 KB
testcase_24 AC 4 ms
16,156 KB
testcase_25 AC 4 ms
16,280 KB
testcase_26 AC 172 ms
27,368 KB
testcase_27 AC 132 ms
27,480 KB
testcase_28 AC 132 ms
27,376 KB
testcase_29 AC 110 ms
27,360 KB
testcase_30 AC 46 ms
20,628 KB
testcase_31 AC 27 ms
18,824 KB
testcase_32 AC 68 ms
22,564 KB
testcase_33 AC 51 ms
21,400 KB
testcase_34 AC 95 ms
25,116 KB
testcase_35 AC 68 ms
22,700 KB
testcase_36 AC 40 ms
20,100 KB
testcase_37 AC 40 ms
20,244 KB
testcase_38 AC 46 ms
20,760 KB
testcase_39 AC 79 ms
23,936 KB
testcase_40 AC 100 ms
26,968 KB
testcase_41 AC 95 ms
27,124 KB
testcase_42 AC 96 ms
26,968 KB
testcase_43 AC 89 ms
27,080 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using LL=long long;
using ULL=unsigned long long;
#define rep(i,n) for(int i=0; i<(n); i++)

const ULL M=1000000007;

struct MLL{
	ULL v;
	MLL(ULL a=0){ v=a; }
	void operator+=(MLL r){ v+=r.v; if(v>=M) v-=M; }
	MLL operator+(MLL r) const { r+=*this; return r; }
	void operator*=(MLL r){ v=v*r.v%M; }
	MLL operator*(MLL r) const { r*=*this; return r; }
};

struct Mat{
  MLL S=0,L=0,C=0;
  Mat add_D0() const { return Mat{S,L,C+1}; }
  Mat extend(MLL d) const { return Mat{S+L*2*d+d*d*C,L+C*d,C}; }
};

MLL getSqSum(Mat l,Mat r){ return l.S*r.C + l.L*2*r.L + r.S*l.C; }
Mat operator+(Mat l,Mat r){ return { l.S+r.S, l.L+r.L, l.C+r.C }; }

struct Edge{ int to; MLL d; };

int N;
vector<Edge> E[200000];
Edge P[200000]={};
vector<int> I;
Mat V[200000];
MLL ans=0;

int main(){
  scanf("%d",&N);
  rep(i,N-1){
    int u,v,d; scanf("%d%d%d",&u,&v,&d); u--; v--;
    E[u].push_back({v,ULL(d)});
    E[v].push_back({u,ULL(d)});
  }
  int pI=0;
  P[1]={-1,0}; I.push_back(1);
  while(I.size()>pI){
    int p=I[pI++];
    for(Edge e:E[p]) if(P[p].to!=e.to){ P[e.to]={p,e.d}; I.push_back(e.to); }
  }
  reverse(I.begin(),I.end());
  I.pop_back();

  for(int p:I){
    V[p] = V[p].add_D0();
    V[p] = V[p].extend(P[p].d);
    ans += getSqSum(V[P[p].to],V[p]);
    ans += V[p].S;
    V[P[p].to] = V[P[p].to] + V[p];
  }

  printf("%llu\n",ans.v);

  return 0;
}
0