yukicoder No.15 カタログショッピング

最近yukicoderやってなかった…

問題

No.15 カタログショッピング - yukicoder

解法

半分全列挙するだけ。

const int MAXN = 32;
int P[MAXN];
struct pll {
    ll first;
    ll second;
};

bool operator<(const pll& lhs, const pll& rhs) {
    return lhs.first < rhs.first;
}

int main() {
    cin.tie(0);
    ios::sync_with_stdio(false);
    int N, S;
    cin >> N >> S;
    for (int i = 0; i < N; i++) {
        cin >> P[i];
    }
    int n = N/2;
    vector<pll> first, second;
    for (ll s = 0; s < (1<<n); s++) {
        pll item;
        item.first = 0;
        item.second = s;
        for (int i = 0; i < n; i++) {
            if ((s>>i)&1) item.first += P[i];
        }
        first.push_back(item);
    }
    for (ll s = 0; s < (1<<(N-n)); s++) {
        pll item;
        item.first = 0;
        item.second = (s<<n);
        for (int i = 0; i < (N-n); i++) {
            if ((s>>i)&1) item.first += P[i+n];
        }
        second.push_back(item);
    }
    sort(first.begin(), first.end());
    sort(second.begin(), second.end());
    vector<ll> ans;
    for (pll p : first) {
        pll q;
        q.first = S-p.first;
        q.second = p.second;
        auto it1 = lower_bound(second.begin(), second.end(), q);
        auto it2 = upper_bound(second.begin(), second.end(), q);
        for (auto it = it1; it < it2; it++) {
            ans.push_back(p.second|(it->second));
        }
    }
    vector<vi> ANS;
    for (ll el : ans) {
        vi tmp;
        for (int i = 0; i < 32; i++) {
            if ((el>>i)&1) tmp.push_back(i+1);
        }
        sort(tmp.begin(), tmp.end());
        ANS.push_back(tmp);
    }
    sort(ANS.begin(), ANS.end());
    for (vi v : ANS) {
        int n = v.size();
        for (int i = 0; i < n; i++) {
            cout << v[i];
            if (i != n-1) cout << " ";
        }
        cout << endl;
    }
    return 0;
}