Submission #924344

#	Time	Username	Problem	Language	Result	Execution time	Memory
924344		JakobZorz	Popeala (CEOI16_popeala)	C++17	17 / 100	2041 ms	24412 KiB

popeala

#include<iostream>
#include<vector>
#include<queue>
#include<stack>
#include<algorithm>
#include<limits.h>
#include<math.h>
#include<map>
#include<set>
#include<unordered_map>
#include<unordered_set>
#include<iomanip>
#include<cstring>
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
using namespace std;
 
int n,t,s;
int pts[20000];
int done[50][20000];
ll dp[51][20001];
const int TREE_SIZE=1<<15;

struct Tree{
    vector<ll>arr;
    
    void build(vector<ll>init){
        arr=vector<ll>(TREE_SIZE);
        for(int i=0;i<int(init.size());i++)
            arr[i]=init[i];
    }
    
    void add(int l,int r,ll x){
        for(int i=l;i<r;i++)
            arr[i]+=x;
    }
    
    ll get_min(int l,int r){
        ll res=1e18;
        for(int i=l;i<r;i++)
            res=min(res,arr[i]);
        return res;
    }
};

Tree cost[51];

void solve(){
    cin>>n>>t>>s;
    for(int i=0;i<t;i++)
        cin>>pts[i];
    for(int i=0;i<n;i++){
        string str;
        cin>>str;
        for(int j=0;j<t;j++){
            if(str[j]=='1'){
                if(j)
                    done[i][j]=done[i][j-1]+1;
                else
                    done[i][j]=1;
            }
        }
    }
    
    for(int i=0;i<51;i++)
        for(int j=0;j<20001;j++)
            dp[i][j]=1e18;
    
    dp[0][0]=0;
    
    for(int num=1;num<=s;num++){
        for(int p=0;p<=n;p++){
            vector<ll>init;
            for(int i=0;i<=t;i++)
                init.push_back(dp[num-1][i]);
            cost[p].build(init);
        }
        
        for(int i=0;i<=t;i++){
            if(i){
                vector<int>vec;
                vec.push_back(i);
                vec.push_back(0);
                for(int p=0;p<n;p++)
                    vec.push_back(done[p][i-1]);
                sort(vec.begin(),vec.end());
                reverse(vec.begin(),vec.end());
                for(int j=0;j<=n;j++){
                    dp[num][i]=min(dp[num][i],cost[j].get_min(i-vec[j],i-vec[j+1]));
                }
            }
            
            for(int p=0;p<=n;p++){
                cost[p].add(0,i+1,pts[i]*p);
            }
        }
    }
    
    for(int i=0;i<s;i++)
        cout<<dp[i+1][t]<<"\n";
}
 
int main(){
    ios::sync_with_stdio(false);cout.tie(NULL);cin.tie(NULL);
    //freopen("input.in","r",stdin);freopen("output.out","w",stdout);
    int t=1;//cin>>t;
    while(t--)solve();
    return 0;
}
 
/*
 
2 3 3
4 3 5
101
110
 
 
0
8
16
 
 */

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