結果
問題 |
No.1530 Permutation and Popcount
|
ユーザー |
|
提出日時 | 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 |
ソースコード
#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' : ' '); }