結果

問題 No.1530 Permutation and Popcount
ユーザー tokusakurai
提出日時 2021-06-05 10:24:58
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 37 ms / 2,000 ms
コード長 1,971 bytes
コンパイル時間 2,005 ms
コンパイル使用メモリ 200,112 KB
最終ジャッジ日時 2025-01-22 03:29:35
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
#define rep(i, n) for(int i = 0; i < n; i++)
#define rep2(i, x, n) for(int i = x; i <= n; i++)
#define rep3(i, x, n) for(int i = x; i >= n; i--)
#define each(e, v) for(auto &e: v)
#define pb push_back
#define eb emplace_back
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
#define sz(x) (int)x.size()
using ll = long long;
using pii = pair<int, int>;
using pil = pair<int, ll>;
using pli = pair<ll, int>;
using pll = pair<ll, ll>;
const int MOD = 1000000007;
//const int MOD = 998244353;
const int inf = (1<<30)-1;
const ll INF = (1LL<<60)-1;
template<typename T> bool chmax(T &x, const T &y) {return (x < y)? (x = y, true) : false;};
template<typename T> bool chmin(T &x, const T &y) {return (x > y)? (x = y, true) : false;};

struct io_setup{
    io_setup(){
        ios_base::sync_with_stdio(false);
        cin.tie(NULL);
        cout << fixed << setprecision(15);
    }
} io_setup;

int main(){
    int N, M; cin >> N >> M;

    vector<int> a(N);

    rep2(i, 1, N){
        rep2(j, 1, N){
            a[i-1] += 1-(__builtin_popcount(i*j)%2);
        }
    }

    int K = accumulate(all(a), 0)-M;

    if(K < 0 || K%2 == 1) {cout << "-1\n"; return 0;}

    K /= 2;

    vector<vector<int>> dp(N+1, vector<int>(K+1, -1));
    dp[0][0] = 2;

    rep(i, N){
        rep(j, K+1){
            if(dp[i][j] == -1) continue;

            if(j+a[i] <= K) dp[i+1][j+a[i]] = 1;
            dp[i+1][j] = 0;
        }
    }

    if(dp[N][K] == -1) {cout << "-1\n"; return 0;}

    int pi = N, pj = K;
    vector<int> X, Y;

    while(true){
        if(dp[pi][pj] == 2) break;

        if(dp[pi][pj] == 0){
            X.eb(pi);
            --pi;
        }
        else{
            Y.eb(pi);
            pj -= a[--pi];
        }
    }

    int P = sz(X), Q = sz(Y);
    cout << P << ' ' << Q << '\n';
    rep(i, P) cout << X[i] << (i == P-1? '\n' : ' ');
    rep(i, Q) cout << Y[i] << (i == Q-1? '\n' : ' ');
}
0