#include<bits/stdc++.h>
using namespace std;
#define fast ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
#define ll long long int
const ll N = 12 , INF = 1e18 , MOD = 1e9+7;
ll n;
vector<ll> a(N);
ll dp[(1ll<<N)][N];
ll log_a_to_base_b(ll a, ll b){
return log2(a) / log2(b);
}
ll expo(ll a , ll b){
if(b == 0) return 1;
if(b == 1) return a;
ll p = expo(a,b/2);
if(b%2 == 0) return (p*p);
return (p*p*a);
}
ll countone(ll x){
ll l = log_a_to_base_b(x,3);
ll cnt = 0;
ll s = expo(3,l);
while(x > 0){
ll p = x/s;
x -= s*p;
if(p == 1) cnt++;
s /= 3;
}
return cnt;
}
ll go(ll mask , ll last){
if(__builtin_popcountll(mask) == n) return 1;
if(dp[mask][last] != -1) return dp[mask][last];
ll ans = 0;
for(ll i = 1; i <= n; i++){
if(mask&(1ll<<i)) continue;
if(last != 0 && __builtin_popcountll(a[last]) == __builtin_popcountll(a[i])) ans += go(mask|(1ll<<i),i);
else if(last != 0 && countone(a[last]) == countone(a[i])) ans += go(mask|(1ll<<i),i);
else if(last == 0) ans += go(mask|(1ll<<i),i);
}
return dp[mask][last] = ans;
}
void solve(){
cin >> n;
for(ll i = 1; i <= n; i++) cin >> a[i];
memset(dp,-1,sizeof dp);
cout << go(0,0);
}
int main(){
fast;
ll tc = 1;
// cin >> tc;
while(tc--) solve();
return 0;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
596 KB |
Output is correct |
2 |
Correct |
1 ms |
584 KB |
Output is correct |
3 |
Correct |
1 ms |
596 KB |
Output is correct |
4 |
Correct |
1 ms |
596 KB |
Output is correct |
5 |
Correct |
0 ms |
724 KB |
Output is correct |
6 |
Correct |
2 ms |
724 KB |
Output is correct |
7 |
Correct |
1 ms |
596 KB |
Output is correct |
8 |
Correct |
1 ms |
596 KB |
Output is correct |
9 |
Correct |
1 ms |
596 KB |
Output is correct |
10 |
Correct |
2 ms |
596 KB |
Output is correct |
11 |
Runtime error |
1 ms |
1216 KB |
Execution killed with signal 11 |
12 |
Halted |
0 ms |
0 KB |
- |