oj.uz

답안 #557922

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
557922 2022-05-06T10:08:31 Z FatihSolak W (RMI18_w) C++17
50 / 100
 319 ms 21180 KB 
#include <bits/stdc++.h>
#define N 300005
using namespace std;
const int mod = 1e9 + 7;
long long fact[N];
long long ifact[N];
long long binpow(long long a,long long b){
	long long ret = 1;
	while(b){
		if(b & 1)
			ret = ret*a%mod;
		a = a*a%mod;
		b >>= 1;
	}
	return ret;
}
long long C(int n,int r){
	return fact[n] * ifact[r]%mod*ifact[n-r]%mod;
}
long long val[N][5];
void solve(){
	int n;
	cin >> n;
	map<int,int> cnt;
	for(int i = 1;i<=n;i++){
		int x;
		cin >> x;
		cnt[x]++;
	}
	vector<int> v;
	for(auto u:cnt)
		v.push_back(u.second);
	if(cnt.size() == 2){
		if(v[0] < 2 || v[1] < 3){
			cout << 0;
			return;
		}
		cout << C(v[0]-2+1,1) *C(v[1]-3+2,2) %mod;
		return;
	}
	/*
	for(int i = 2;i<=n;i++){
		val[i][1] = 1;
	}
	for(int i = 3;i<=n;i++){
		val[i][2] = (binpow(2,i-1) - 2 + mod)%mod;
	}
	for(int i = 4;i<=n;i++){
		//val[n][3] = (3^n - 2^(n+2) - 2n + 11)/4
		val[i][3] = (binpow(3,i) - binpow(2,i+2) - 2*i + 11 + 2*mod)%mod*binpow(4,mod-2)%mod;
		//cout << i << " " << val[i][3] << endl;
	}*/
	for(int i = n;i<=n;i++){
		for(int j = 2;j<=i-3;j++){
			val[i][4] = (val[i][4] + C(i-1,j-1) * 
			(   (binpow(3,i-j) - binpow(2,i-j+1) - 2 * (i-j) + 3 + 5ll*mod  )%mod       )%mod)%mod;
		}

		/*
		long long add = 0;

		add = 0;
		add = (binpow(4,n-1));
		for(int j = 0;j<1;j++){
			add = (add - C(i-1,j)*binpow(3,i-j) + mod)%mod;
		}
		for(int j = i-4;j<i;j++){
			add = (add - C(i-1,j)*binpow(3,i-j) + mod)%mod;
		}
		add = (add * binpow(3,mod-2))%mod;
		val[i][4] = (val[i][4] + add)%mod;


		add = 0;
		add = (binpow(3,n-1));
		for(int j = 0;j<1;j++){
			add = (add - C(i-1,j)*binpow(2,i-j) + mod)%mod;
		}
		for(int j = i-4;j<i;j++){
			add = (add - C(i-1,j)*binpow(2,i-j) + mod)%mod;
		}
		val[i][4] = (val[i][4] + add)%mod;

		add = 0;
		add = (binpow(3,n-1));
		for(int j = 0;j<1;j++){
			add = (add - C(i-1,j)*binpow(2,i-j) + mod)%mod;
		}
		for(int j = i-4;j<i;j++){
			add = (add - C(i-1,j)*binpow(2,i-j) + mod)%mod;
		}
		val[i][4] = (val[i][4] + add)%mod;
		*/
		val[i][4] = val[i][4]*binpow(2,mod-2) %mod;
		//cout << i << " " << val[i][4] << endl;
	}
	cout << val[n][4] << endl;

}
int main(){
	ios_base::sync_with_stdio(false);
	cin.tie(nullptr);
	#ifdef Local
		freopen("in.txt","r",stdin);
		freopen("out.txt","w",stdout);
	#endif
	fact[0] = ifact[0]= 1;
	for(int i = 1;i<N;i++){
		fact[i] = fact[i-1] * i %mod;
		ifact[i] = binpow(fact[i],mod-2);
	}
	int t = 1;
	//cin >> t;
	while(t--){
		solve();
	}
	#ifdef Local
		cout << endl << fixed << setprecision(2) << 1000.0*clock()/CLOCKS_PER_SEC << " milliseconds.";
	#endif
}

Subtask #1
20.0 / 20.0

# 결과 실행 시간 메모리 Grader output
1 Correct 42 ms 4908 KB Output is correct
2 Correct 42 ms 4944 KB Output is correct
3 Correct 51 ms 4992 KB Output is correct
4 Correct 64 ms 4988 KB Output is correct

Subtask #2
30.0 / 30.0

# 결과 실행 시간 메모리 Grader output
1 Correct 44 ms 4980 KB Output is correct
2 Correct 46 ms 5044 KB Output is correct
3 Correct 59 ms 6664 KB Output is correct
4 Correct 114 ms 10312 KB Output is correct
5 Correct 225 ms 15552 KB Output is correct
6 Correct 319 ms 21180 KB Output is correct

Subtask #3
0 / 50.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 45 ms 4948 KB Output isn't correct
2 Incorrect 46 ms 4936 KB Output isn't correct
3 Incorrect 45 ms 5016 KB Output isn't correct
4 Incorrect 46 ms 5008 KB Output isn't correct
5 Incorrect 44 ms 5008 KB Output isn't correct
6 Incorrect 46 ms 5212 KB Output isn't correct
7 Incorrect 49 ms 5224 KB Output isn't correct
8 Incorrect 78 ms 4944 KB Output isn't correct
9 Incorrect 102 ms 5052 KB Output isn't correct
10 Incorrect 307 ms 20804 KB Output isn't correct