This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 2e4+1;
#define owo ios_base::sync_with_stdio(0);cin.tie(0);
typedef long long int ll;
int pt[MAXN],n,t,s;
string stu[50];
ll dp1[MAXN],dp2[MAXN],mn[2*MAXN],lazy[2*MAXN];
int last[50];
void build(int v,int l,int r){
lazy[v] = 0;
if(l==r){
mn[v] = dp1[l];
return;
}
int tm = (l+r)/2;
build(2*v,l,tm);
build(2*v+1,tm+1,r);
mn[v] = min(mn[2*v],mn[2*v+1]);
}
void upd(int v,int l,int r,int tl,int tr,int val){
if(l>r)return;
if(l==tl && r==tr){
mn[v]+=val;
lazy[v]+=val;
return;
}
int tm = (tl+tr)/2;
upd(2*v,l,min(tm,r),tl,tm,val);
upd(2*v+1,max(l,tm+1),r,tm+1,tr,val);
mn[v] = min(mn[2*v],mn[2*v+1]) + lazy[v];
}
ll query(int v,int l,int r,int tl,int tr){
if(l>r)return 1e15;
if(l==tl && r==tr)return mn[v];
int tm = (tl+tr)/2;
return min(query(2*v,l,min(tm,r),tl,tm),
query(2*v+1,max(l,tm+1),r,tm+1,tr)) + lazy[v];
}
void tran(){
for(int i=0;i<n;i++)last[i] = 0;
build(1,0,t);
for(int i=1;i<=t;i++){
for(int j=0;j<n;j++){
if(stu[j][i-1] == '1'){
upd(1,last[j],i-1,0,t,pt[i]);
//for(int k=last[j];k<i;k++)dp1[k]+=pt[i];
//[last[j],i) gets increased by pt[i]
}else{
//reverse the change did
if(i>1 && stu[j][i-2] == '1'){
//for(int x=last[j]+1;x<i;x++){
// for(int k=last[j];k<x;k++)dp1[k]-=pt[x];
//}
for(int x=last[j]+1;x<i;x++)upd(1,last[j],x-1,0,t,-pt[x]);
}
last[j] = i;
}
}
//for(int x=0;x<i;x++)dp2[i] = min(dp2[i],dp1[x]);
dp2[i] = query(1,0,i-1,0,t);
}
//for(int i=0;i<=t;i++)cout<<dp2[i]<<" ";
//cout<<'\n';
}
int main(){
owo
cin>>n>>t>>s;
for(int i=1;i<=t;i++)cin>>pt[i];
for(int i=0;i<n;i++)cin>>stu[i];
vector<int>ans(s+1,0);
for(int i=0;i<=t;i++)dp1[i] = dp2[i]= 1e15;
dp1[0] = 0;
for(int i=1;i<=s;i++){
tran();
ans[i] = dp2[t];
for(int j=0;j<=t;j++){
swap(dp1[j],dp2[j]);
dp2[j] = 1e15;
}
}
for(int i=1;i<=s;i++)cout<<ans[i]<<'\n';
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |