結果

問題 No.118 門松列(2)
ユーザー nenuon
提出日時 2017-07-28 01:16:47
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 20 ms / 5,000 ms
コード長 859 bytes
コンパイル時間 749 ms
コンパイル使用メモリ 88,804 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-10 02:29:54
合計ジャッジ時間 1,857 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <cstdio>
#include <iostream>
#include <map>
#include <cmath>
#include <queue>
#include <set>
#include <sstream>
#include <stack>
#include <string>
#include <vector>
#include <stdlib.h>
#include <stdio.h>
#include <bitset>
using namespace std;
#define FOR(I,A,B) for(int I = (A); I < (B); ++I)
typedef long long ll;

const int MOD = 1e9 + 7;
ll num[100005];
int main()
{
  FOR(i,0,100005) num[i] = 0;
  ll N;
  cin>>N;
  ll cnt = 0;
  FOR(i,0,N) {
    int in;
    cin >> in;
    if(num[in]==0) cnt++;
    num[in]++;
  }
  ll ans = N*(N-1)/2*(N-2)/3;
  ans %= MOD;
  FOR(i,0,100005) {
    ll n = num[i];
    if(n>1) {
      ans -= (n*(n-1)/2*(N-n)) % MOD;
      ans = (ans + MOD) % MOD;
    }
    if(n>2) {
      ans -= (n*(n-1)/2*(n-2)/3) % MOD;
      ans = (ans + MOD) % MOD;
    }
  }
  cout << ans % MOD << endl;
  return 0;
}
0