結果

問題 No.2869 yuusaan's Knapsacks
ユーザー nana
提出日時 2024-09-19 00:59:11
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,296 ms / 4,500 ms
コード長 4,214 bytes
コンパイル時間 3,613 ms
コンパイル使用メモリ 189,000 KB
最終ジャッジ日時 2025-02-24 09:31:43
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<sstream>
#include<algorithm>
#include<deque>
#include<list>
#include<map>
#include<memory>
#include<queue>
#include<set>
#include<stack>
#include<utility>
#include<string.h>
#include<string>
#include<math.h>
#include<float.h>
#include<stdio.h>
#include<vector>
#include<iomanip>
#include<bitset>
#include<random>
#include<complex>
#include <cassert>
#include <utility>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; }
template<typename T> inline bool chmin(T &a, T b) { return ((a>b) ? (a = b, true) : (false));}
#define rep(i,s,n) for(long long i=s;i<(long long)(n);i++)
#define rrep(i,s,n) for(long long i=n-1;i>=s;i--)
const long long inf = 1LL<<60;
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
//pairのsecondでソートsort(p.begin(),p.end(),cmp)
#define cmp [](pair<ll,ll> a, pair<ll,ll> b){return a.second<b.second;}
//cmpArg := 偏角ソート, atan2l(y, x); atan2l(y, x) = [-pi, pi]
#define cmpArg [](const auto &p1, const auto &p2) { return atan2l(p1.second, p1.first) < atan2l(p2.second, p2.first); }
typedef pair<long long, long long> P;
typedef pair<ll, pair<ll,ll> > PP;
#define rll ll,vector<ll>,greater<ll>
#define rP P,vector<P>,greater<P>
const long double pi = 3.14159265358979;
typedef unsigned long long ull;
#define vll vector<ll>
#define vvll vector<vector<ll>>
#define vmint vector<mint>
#define vvmint vector<vector<mint>>
#define vvch vector<vector<char>>
#define vch vector<char>
#define vstring vector<string>
#define rPP PP,vector<PP>,greater<PP>
#define vP vector<P> 
#define vvP vector<vector<P>>
#define vPP vector<PP>
#define all(x) x.begin(), x.end()
//UNIQUE(x) xをソートして値の被りがないようにする
#define UNIQUE(x) sort(all(x)), x.erase(unique(all(x)), x.end())
int pc(ll x) { return __builtin_popcount(x); } //ビット列にどれだけ1がたっているかを求める pop count
//逆順のlower_bound(単調減少関数で自分以下を二分探索)するときは`ll index = lower_bound(all(inv), -a[i], greater<ll>()) - inv.begin();`, 逆からみたlis的なやつが作れる
//オバフロしない計算はa > inf - b および a > inf / bでとってね
//半分全列挙は前の方を(siz+1)/2ででかくする
//using mint = atcoder::modint, main関数でmint::set_mod(M)とすると任意modのmintにできる
//mapでも auto it = mp.lower_bound(key)としてlower_boundが使用できる while(it != mp.end())でループすることもできる

using mint = modint998244353;

int main()
{
    int n,m; cin >> n >> m;
    vector<int> e(n); rep(i,0,n) cin >> e[i];
    vector<int> v(m), w(m);
    rep(i,0,m) cin >> v[i] >> w[i];

    //各ナップサックに対して入れれる集合を列挙しておく
    vector<vector<long long>> bags(n,vector<long long>(1<<m,-inf));
    rep(i,0,n) {
        for(int S=0;S<1<<m;S++) {
            long long weight = 0, value = 0;
            rep(j,0,m) if((S>>j)&1) weight+=w[j], value+=v[j];
            if(weight <= e[i]) { bags[i][S] = value; }
        }
    }

    //ビットdp
    vector<vector<long long>> dp(n+1,vector<long long>(1<<m,0));
    dp[0][0] = 0;
    vector<vector<long long>> prev(n+1,vector<long long>(1<<m,-1));
    rep(i,0,n) {
        for(int S=0;S<1<<m;S++) {
            int cS = (1<<m)-1-S;
            if(chmax(dp[i+1][S], dp[i][S])) {
                prev[i+1][S] = 0;
            }
            for(int add=cS;add!=0;add=(add-1)&cS) {
                if(bags[i][add] < 0) continue;
                if(chmax(dp[i+1][S | add], dp[i][S] + bags[i][add])) {
                    prev[i+1][S|add] = add;
                }
            }
        }
    }

    long long ans = 0;
    long long posS = 0;
    rep(S,0,1<<m) if(chmax(ans, dp[n][S])) posS = S;
    cout << ans << endl;
    int pos = n;
    vector<vector<int>> vecs(n);
    while(posS) {
        int cS = prev[pos][posS];
        pos--;
        posS ^= cS;
        rep(i,0,m) if((cS>>i)&1) vecs[pos].push_back(i);
    }
    rep(i,0,n) {
        cout << vecs[i].size() << " "; for(auto v : vecs[i]) cout << v+1 << " "; cout << endl;
    }
}
0