結果

問題 No.1333 Squared Sum
ユーザー kotatsugame
提出日時 2021-01-08 22:22:47
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 362 ms / 2,000 ms
コード長 2,875 bytes
コンパイル時間 1,050 ms
コンパイル使用メモリ 75,576 KB
実行使用メモリ 40,780 KB
最終ジャッジ日時 2024-11-16 13:03:27
合計ジャッジ時間 11,208 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 44
権限があれば一括ダウンロードができます
コンパイルメッセージ
a.cpp:31:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]

ソースコード

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

#line 1 "a.cpp"
#include<iostream>
#include<vector>
using namespace std;
#line 3 "/home/kotatsugame/library/math/modint.cpp"
#include<utility>
template<int m>
struct modint{
    unsigned int x;
    constexpr modint()noexcept:x(){}
    template<typename T>
    constexpr modint(T x_)noexcept:x((x_%=m)<0?x_+m:x_){}
    constexpr unsigned int val()const noexcept{return x;}
    constexpr modint&operator++()noexcept{if(++x==m)x=0;return*this;}
    constexpr modint&operator--()noexcept{if(x==0)x=m;--x;return*this;}
    constexpr modint operator++(int)noexcept{modint res=*this;++*this;return res;}
    constexpr modint operator--(int)noexcept{modint res=*this;--*this;return res;}
    constexpr modint&operator+=(const modint&a)noexcept{x+=a.x;if(x>=m)x-=m;return*this;}
    constexpr modint&operator-=(const modint&a)noexcept{if(x<a.x)x+=m;x-=a.x;return*this;}
    constexpr modint&operator*=(const modint&a)noexcept{x=(unsigned long long)x*a.x%m;return*this;}
    constexpr modint&operator/=(const modint&a)noexcept{return*this*=a.inv();}
    constexpr modint operator+()const noexcept{return*this;}
    constexpr modint operator-()const noexcept{return modint()-*this;}
    constexpr modint pow(long long n)const noexcept
    {
        if(n<0)return pow(-n).inv();
        modint x=*this,r=1;
        for(;n;x*=x,n>>=1)if(n&1)r*=x;
        return r;
    }
    constexpr modint inv()const noexcept
    {
        int s=x,t=m,x=1,u=0;
        while(t)
        {
            int k=s/t;
            s-=k*t;
            swap(s,t);
            x-=k*u;
            swap(x,u);
        }
        return modint(x);
    }
    friend constexpr modint operator+(const modint&a,const modint&b){return modint(a)+=b;}
    friend constexpr modint operator-(const modint&a,const modint&b){return modint(a)-=b;}
    friend constexpr modint operator*(const modint&a,const modint&b){return modint(a)*=b;}
    friend constexpr modint operator/(const modint&a,const modint&b){return modint(a)/=b;}
    friend constexpr bool operator==(const modint&a,const modint&b){return a.x==b.x;}
    friend constexpr bool operator!=(const modint&a,const modint&b){return a.x!=b.x;}
    friend ostream&operator<<(ostream&os,const modint&a){return os<<a.x;}
    friend istream&operator>>(istream&is,modint&a){long long v;is>>v;a=modint(v);return is;}
};
#line 5 "a.cpp"
using mint=modint<(int)1e9+7>;
int N;
vector<pair<int,mint> >G[2<<17];
mint ans=0;
pair<mint,mint>dfs(int u,int p,mint d)
{
    vector<pair<mint,mint> >A;
    mint ch=1,ds=0;
    for(pair<int,mint>q:G[u])if(q.first!=p)
    {
        pair<mint,mint>r=dfs(q.first,u,d+q.second);
        A.push_back(r);
        ch+=r.first;
        ds+=r.second;
    }
    ans+=d*d*(N-2);
    for(pair<mint,mint>q:A)
    {
        ans+=2*(ch-q.first)*q.first*d*d;
        ans-=4*(ch-q.first)*d*q.second;
    }
    ans+=2*(ch-1)*d*d;
    ans-=4*(ch-1)*d*d;
    ds+=d;
    return make_pair(ch,ds);
}
main()
{
    cin>>N;
    for(int i=1;i<N;i++)
    {
        int u,v;mint w;cin>>u>>v>>w;
        u--,v--;
        G[u].push_back(make_pair(v,w));
        G[v].push_back(make_pair(u,w));
    }
    mint ad=dfs(0,0,mint(0)).second;
    ans+=ad*ad;
    cout<<ans<<endl;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0