2019 가축 오프 더 학교 (두 번째 필드) D.Kth 최소 파벌

주제 링크 : https://ac.nowcoder.com/acm/contest/882/D

그래서 우리는 시작 열거를 고려 플러스 포인트는 빈 세트이며, 다음 작은 첫 $의 케이 $를 찾을 수 있습니다, 우리는 올바른 지점의 현재 상태를 유지하는 우선 순위 큐가 필요합니다, 우리가 발견하는 $ n 개의 \의 당량 $ 100 $ K \ 당량 10 ^ 6 $ 우리는 단지 $ 10 ^ 6 $ 상태까지 열거 할 필요가있다, 우리는 국가가, 우리가 열거를 시작하기 위해 현재 상태에서 가장 큰 지점을 표시 할 때마다 열거 반복되지 않도록하고 싶습니다. 세트의 타입을 처리하는 방법은, 비트 세트에 의해 최적화 될 수있다

#include <bits/stdc++.h>
#define pii pair<int, int>
#define pil pair<int, long long>
#define pll pair<long long, long long>
#define lowbit(x) ((x)&(-x))
#define mem(i, a) memset(i, a, sizeof(i))
#define sqr(x) ((x)*(x))
#define all(x) x.begin(),x.end()
#define ls (k << 1)
#define rs (k << 1 | 1)
using namespace std;
typedef long long ll;
template <typename T>
inline void read(T &X) {
    X = 0; char ch = 0; T op = 1;
    for(; ch > '9' || ch < '0'; ch = getchar())
        if(ch == '-') op = -1;
    for(; ch >= '0' && ch <= '9'; ch = getchar())
        X = (X << 3) + (X << 1) + ch - 48;
    X *= op;
}

const int INF = 0x3f3f3f3f;
const int N = 105 + 5;
struct node {
    ll w,s;
    bitset<105> cli;
    bool operator <(const node &a) const {
        return w > a.w;
    }
};
int n,k,val[N];
bitset<105> e[N];
void bfs() {
    priority_queue<node> q;
    bitset<105> fir;
    fir.reset();
    q.push({0, 0, fir});
    int cnt = 0;
    while(!q.empty()) {
        node u = q.top(); q.pop();
       // cout << u.w << "\n";
        cnt++; if(cnt == k) {cout << u.w << "\n"; exit(0);}
        for(int i = u.s + 1; i <= n; i++) {
            if((e[i] & u.cli) == u.cli) {
                bitset<105> b(u.cli);
                b.set(i, 1);
                q.push({u.w + val[i], i, b});
            }
        }
    }
    cout << -1;
}
int main() {
#ifdef INCTRY
    freopen("input.txt", "rt", stdin);
#endif
    read(n); read(k);
    for(int i = 1; i <= n; i++) read(val[i]);
    for(int i = 1; i <= n; i++) {
        string s; cin >> s;
       // cout << s;
        for(int j = 1; j <= n; j++) {
           // cout << s[j - 1];
            if(s[j - 1] == '1') e[i].set(j, 1);
            else e[i].set(j, 0);
        }
    }
    bfs();


#ifdef INCTRY
    cerr << "\nTime elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << " s.\n";
#endif
    return 0;
}