oj.uz

Submission #335907

# Submission time Handle Problem Language Result Execution time Memory
335907 2020-12-14T08:22:39 Z IZhO_2021_I_want_Silver Beautiful row (IZhO12_beauty) C++14
100 / 100
 921 ms 172828 KB 
#include <iostream>
#include <algorithm>
#include <vector>
#include <cmath>
#include <set>
#include <map>
#include <iomanip>
#include <cassert>
#include <stack>
#include <queue>
#include <deque>
//#include <ext/pb_ds/assoc_container.hpp>
//#include <ext/pb_ds/tree_policy.hpp>

using namespace std;
//using namespace __gnu_pbds;

typedef long long ll;
typedef pair <int, int> pii;
typedef pair <ll, ll> pll;

// template<typename T> using ordered_set = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
//  order_of_key (k) : Number of items strictly smaller than k .
//  find_by_order(k) : K-th element in a set (counting from zero).
#define sz(a) (int)a.size()
#define all(a) a.begin(), a.end()
#define pb push_back
#define ppb pop_back
#define mkp make_pair
#define F first
#define S second
#define lb lower_bound
#define ub upper_bound
#define show(a) cerr << #a <<" -> "<< a <<" "
#define nl cerr <<"\n"
#define fo(a, b, c, d) for(int (a) = (b); (a) <= (c); (a) += (d))
#define foo(a, b, c ,d) for(int (a) = (b); (a) >= (c); (a) -= (d))
//#define int ll

const int N = (1 << 20);

int n, cnt2[21], cnt3[21];
ll dp[N][21];

void solve() {
    cin >> n;
    for (int i = 0; i < n; ++i) {
        int x;
        cin >> x;
        cnt2[i] = __builtin_popcount(x);
        while (x) {
            if (x % 3 == 1) { ++cnt3[i]; }
            x /= 3;
        }
    }

    for (int i = 0; i < n; ++i) { dp[(1 << i)][i] = 1; }

    for (int mask = 1; mask < (1 << n); ++mask) {
        for (int last = 0; last < n; ++last) {
            if (!((mask >> last) & 1)) { continue; }
            for (int i = 0; i < n; ++i) {
                if ((mask >> i) & 1) { continue; }
                if (cnt2[last] == cnt2[i] || cnt3[last] == cnt3[i]) {
                    dp[(mask ^ (1 << i))][i] += dp[mask][last];
                }
            }
        }
    }

    ll ans = 0;
    for (int i = 0; i < n; ++i) {
        ans += dp[(1 << n) - 1][i];
    }
    cout << ans;
}

 main () {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	int tests = 1;
	//cin >> tests;
	while (tests --) {
		solve();
	}
	return 0;
}
/*
    Just Chalish!
*/

Compilation message

beauty.cpp:78:8: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   78 |  main () {
      |        ^

Subtask #1
100.0 / 100.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 364 KB Output is correct
2 Correct 0 ms 364 KB Output is correct
3 Correct 1 ms 364 KB Output is correct
4 Correct 1 ms 364 KB Output is correct
5 Correct 1 ms 364 KB Output is correct
6 Correct 1 ms 492 KB Output is correct
7 Correct 1 ms 492 KB Output is correct
8 Correct 1 ms 492 KB Output is correct
9 Correct 1 ms 492 KB Output is correct
10 Correct 1 ms 492 KB Output is correct
11 Correct 9 ms 3052 KB Output is correct
12 Correct 9 ms 3052 KB Output is correct
13 Correct 39 ms 11116 KB Output is correct
14 Correct 217 ms 43372 KB Output is correct
15 Correct 166 ms 43372 KB Output is correct
16 Correct 202 ms 43372 KB Output is correct
17 Correct 216 ms 43372 KB Output is correct
18 Correct 207 ms 43628 KB Output is correct
19 Correct 921 ms 172828 KB Output is correct
20 Correct 780 ms 172780 KB Output is correct