結果

問題 No.2235 Line Up Colored Balls
コンテスト
ユーザー Nzt3
提出日時 2023-03-03 22:09:41
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 28 ms / 2,000 ms
コード長 613 bytes
コンパイル時間 2,521 ms
コンパイル使用メモリ 195,412 KB
最終ジャッジ日時 2025-02-11 02:49:10
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 55
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7;
long long modinv(long long n){
  long long ret=1,t=n;
  for(int i=0;i<30;i++){
    if((mod-2)>>i&1)ret=ret*t%mod;
    t=t*t%mod;
  }
  return ret;
}

int main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  int N;
  cin>>N;
  if(N==1){
    cout<<"1\n";
    return 0;
  }
  vector<long long>A(N);
  for(auto &i:A)cin>>i;
  long long s=accumulate(A.begin(),A.end(),0ll);
  long long ans=0;
  for(int i=0;i<N;i++){
    ans=(ans+A[i]*(s-A[i]))%mod;
  }
  ans=ans*modinv(s)%mod*modinv(s-1)%mod*(s-1)%mod;
  ans=(ans+1)%mod;
  cout<<ans<<'\n';
}
0