Submission #893421

#	Time	Username	Problem	Language	Result	Execution time	Memory
893421		ksujay2	Council (JOI23_council)	C++17	78 / 100	4035 ms	13328 KiB

council

#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")
#include <bits/stdc++.h>
using namespace std;
const int BS = 5000;
const int INF = 1e9;
int main() {
    ios::sync_with_stdio(false), cin.tie(nullptr);
    int N, M; cin >> N >> M;
    vector<int> A(N);
    vector<int> cnt(M);
    for(int i = 0; i < N; i++) {
        for(int j = 0; j < M; j++) {
            int a; cin >> a;
            A[i] += a << j;
            cnt[j] += a;
        }
    }
    vector<int> D;
    auto dp = [&] () {
        for(int i = 0; i < (1 << M); i++) {
            for(int j = 0; j < M; j++) {
                if((i & (1 << j)) > 0) {
                    D[i] = min(D[i], D[i ^ (1 << j)]);
                }
            }
        }
        for(int i = (1 << M) - 1; i >= 0; i--) {
            for(int j = 0; j < M; j++) {
                if((i & (1 << j)) == 0) {
                    D[i] = min(D[i], D[i ^ (1 << j)] + 1);
                }
            }
        }
    };
    for(int i = 0; i < N; i++) {
        if(i % BS == 0) {
            D = vector<int>(1 << M, INF);
            for(int j = 0; j < N; j++) {
                if(j < i || i + BS <= j) {
                    D[A[j]] = 0;
                }
            }
            dp();
        }
        int v = (1 << M) - 1;
        int ans = 0;
        for(int j = 0; j < M; j++) {
            int c = (cnt[j] - ((A[i] & (1 << j)) > 0));
            if(c == N / 2) {
                v ^= 1 << j;
            }
            if(c >= N / 2) ans++;
        }
        int best = D[v];
        for(int j = i - (i % BS); j < (i - (i % BS) + BS) && j < N; j++) {
            if(j != i) {
                best = min(best, __popcount(A[j] ^ (A[j] & v)));
            }
        }
        cout << ans - best << '\n';
    }
}

• Subtask 1: 8 / 8
• Subtask 2: 8 / 8
• Subtask 3: 6 / 6
• Subtask 4: 19 / 19
• Subtask 5: 15 / 15
• Subtask 6: 22 / 22
• Subtask 7: 0 / 22