結果

問題 No.118 門松列(2)
ユーザー ngtkana
提出日時 2020-03-25 10:57:41
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 15 ms / 5,000 ms
コード長 629 bytes
コンパイル時間 2,174 ms
コンパイル使用メモリ 198,748 KB
最終ジャッジ日時 2025-01-09 10:07:21
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
using lint=long long;
int main(){
    std::cin.tie(nullptr);std::ios_base::sync_with_stdio(false);
    std::cout.setf(std::ios_base::fixed);std::cout.precision(15);
    lint n;std::cin>>n;
    std::vector<lint>a(n);
    for(lint&x:a)std::cin>>x;
    std::sort(a.begin(),a.end());
    std::vector<lint>b={0};
    for(lint i=0,j=1;j<=n;j++){
        if(j<n&&a.at(j-1)==a.at(j))continue;
        b.push_back(b.back()+j-i);
        i=j;
    }
    lint ans=0,m=b.size()-1;
    for(lint i=0;i<m;i++){
        ans+=b.at(i)*(b.at(i+1)-b.at(i))*(b.at(m)-b.at(i+1));
    }
    std::cout<<ans%1'000'000'007<<'\n';
}
0