oj.uz

Submission #968126

# Submission time Handle Problem Language Result Execution time Memory
968126 2024-04-23T07:50:24 Z berr Popeala (CEOI16_popeala) C++17
0 / 100
 67 ms 5208 KB 
#include <bits/stdc++.h>
using namespace std;
const int N = 2e3+30;

int r[N][N];

void solve(){
    int n, t, s; cin >> n >> t >> s;
    vector<int> v(t);
    vector<vector<array<int, 2>>> g(t+1);
    vector dis(t+1, vector(s+2, (int)2e9));
    
    for(auto &i: v) cin >> i;

    vector<string> se(n);

    for(auto &i: se){
        cin >> i;
        int k=i[0]-'0';

        for(int l=1; l<t; l++){
            if(i[l]=='0'){
                if(k){
                    r[l-k][l-1]++;
                } 
                k=0;
            }
            else k++;
        }

        if(k){
            r[t-k][t-1]++;
        }
    } 

    for(int i=t; i>1; i--){
        for(int l=0; l+i-1<t; l++){
            r[l+1][l+i-1]+=r[l][l+i-1];
            r[l][l+i-2]+=r[l][l+i-1];
        }
    }

    for(int i=1; i<t; i++) v[i] += v[i-1];

    for(int i=0; i<t; i++){
        int ek=0;
        if(i>0) ek=v[i-1];

        for(int l=i+1; l<t; l++){

            g[i].push_back({l, (v[l-1]-ek)*r[i][l-1]});
        }
 
        g[i].push_back({t, (v[t-1]-ek)*r[i][t-1]});
    }

    priority_queue<array<int, 3>> pq;

    pq.push({0, 0, 0});
    dis[0][0]=0;
    while(pq.size()){

        auto [cos, ed, v]= pq.top();
        pq.pop();
        cos*=-1; 

        if(cos != dis[v][ed]) continue;

        for(auto [i, c]: g[v]){
            if(ed+1 <= s && cos+c < dis[i][ed+1]){
                dis[i][ed+1]=cos+c;
                pq.push({-(cos+c), (ed+1), i});
            }
        }
    }

    for(int i=1; i<=s; i++) cout<<dis[t][i]<<"\n";
}

signed main(){
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    /*freopen("popeala.in", "r", stdin);
    freopen("popeala.out", "w", stdout);
*/
    solve();
}

Subtask #1
0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Incorrect 1 ms 604 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #2
0 / 9.0

# Verdict Execution time Memory Grader output
1 Incorrect 67 ms 5208 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
0 / 9.0

# Verdict Execution time Memory Grader output
1 Runtime error 4 ms 3676 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #4
0 / 74.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Incorrect 1 ms 604 KB Output isn't correct
3 Halted 0 ms 0 KB -