Submission #567445

#	Time	Username	Problem	Language	Result	Execution time	Memory
567445		piOOE	Popeala (CEOI16_popeala)	C++17	0 / 100	125 ms	20140 KiB

popeala

#include <bits/stdc++.h>

using namespace std;

#define all(x) begin(x), end(x)
#define sz(x) ((int)size(x))
#define trace(x) cout << #x << ": " << (x) << endl;

typedef long long ll;

mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count());

const int S = 51, N = 51, T = 4001;
const ll infL = 3e18;

int s, n, t, A[T], ct[T][T], opt[T][S];
ll dp[T][S], pref[T];

bool solved[N][T];

template<typename T>
bool ckmn(T &a, T b) {
    if (a > b) {
        a = b;
        return true;
    }
    return false;
}

ll weight(int l, int r) {
    ll every = 0;
    for (int i = 1; i <= n; ++i) {
        ll sum = 0;
        for (int j = l; j <= r; ++j) {
            if (!solved[i][j]) {
                return 0;
            }
            sum += A[j];
        }
        every += sum;
    }
    return every;
}

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cin >> n >> t >> s;
    for (int i = 1; i <= t; ++i) {
        cin >> A[i];
        pref[i] = pref[i - 1] + A[i];
    }
    for (int i = 1; i <= n; ++i) {
        string ss;
        cin >> ss;
        for (int j = 1; j <= t; ++j) {
            solved[i][j] = (ss[j - 1] == '1');
        }
    }
    for (int k = 1; k <= n; k++) {
        for (int i = 1; i <= t; i++) {
            int cur = solved[k][i];
            ct[i][i] += cur;
            for (int j = i + 1; j <= t; j++) {
                if (solved[k][j] == 0) cur = 0;
                ct[i][j] += cur;
            }
        }
    }
    fill(dp[0] + 1, dp[0] + s + 1, infL);
    for (int i = 1; i <= t; ++i) {
        dp[i][1] = pref[i] * ct[1][i];
        for (int cnt = s; cnt > 1; --cnt) {
            dp[i][cnt] = infL;
            for (int j = max(i - 1, (cnt == s ? i - 1 : opt[i][cnt + 1])); j >= max(cnt - 1, opt[i - 1][cnt]); --j) {
                if (ckmn(dp[i][cnt], dp[j][cnt - 1] + (pref[i] - pref[j]) * ct[j + 1][i])) {
                    opt[i][cnt] = j;
                }
            }
        }
    }
    for (int i = 1; i <= s; ++i)
        cout << dp[t][i] << '\n';
    return 0;
}

• Subtask 1: 0 / 8
• Subtask 2: 0 / 9
• Subtask 3: 0 / 9
• Subtask 4: 0 / 74