#34759 (C++11) No.118 門松列(2)

• 問題: No.118 門松列(2)
• ユーザー: fiord
• 提出日時: 2015-06-29 23:32:06
• 言語: C++11 (gcc 11.4.0)
• 結果: AC
• 実行時間: 22 ms / 5,000 ms
• コード長: 545 bytes
• コンパイル時間: 625 ms
• コンパイル使用メモリ: 66,780 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-04-17 12:58:38
• 合計ジャッジ時間: 1,569 ms
• ジャッジサーバーID （参考情報）: judge3 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	3 ms; 5,248 KB
testcase_01	AC	22 ms; 5,376 KB
testcase_02	AC	21 ms; 5,376 KB
testcase_03	AC	21 ms; 5,376 KB
testcase_04	AC	21 ms; 5,376 KB
testcase_05	AC	21 ms; 5,376 KB
testcase_06	AC	2 ms; 5,376 KB
testcase_07	AC	1 ms; 5,376 KB
testcase_08	AC	1 ms; 5,376 KB
testcase_09	AC	20 ms; 5,376 KB
testcase_10	AC	5 ms; 5,376 KB
testcase_11	AC	7 ms; 5,376 KB
testcase_12	AC	5 ms; 5,376 KB
testcase_13	AC	3 ms; 5,376 KB
testcase_14	AC	19 ms; 5,376 KB
testcase_15	AC	2 ms; 5,376 KB
testcase_16	AC	11 ms; 5,376 KB
testcase_17	AC	3 ms; 5,376 KB
testcase_18	AC	7 ms; 5,376 KB
testcase_19	AC	10 ms; 5,376 KB
testcase_20	AC	10 ms; 5,376 KB
testcase_21	AC	17 ms; 5,376 KB
testcase_22	AC	6 ms; 5,376 KB
testcase_23	AC	15 ms; 5,376 KB
testcase_24	AC	7 ms; 5,376 KB
testcase_25	AC	9 ms; 5,376 KB

ソースコード

#include <iostream>
#include <map>
using namespace std;
typedef long long int ll;
#define E 1000000007

ll ncm(ll n,ll m){
	if(n<m)	return 0;
	if(n==m)	return 1;
	ll ans=1;
	for(int i=0;i<m;i++){
		ans*=n-i;
		ans/=i+1;
	}
	return ans;
}

int main(){
	int n;	cin>>n;
	map<int,int> lib;
	int a;
	for(int i=0;i<n;i++){
		cin>>a;
		lib[a]++;
	}
	ll ans=ncm(n,3);
	for(map<int,int>::iterator it=lib.begin();it!=lib.end();it++){
		ans-=(ncm(it->second,2)*(n-(it->second))+ncm(it->second,3))%E;
		if(ans<0)	ans+=E;
	}
	cout<<ans%E<<endl;
	return 0;
}