# | Submission time | Handle | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
934301 | 2024-02-27T06:40:29 Z | irmuun | Painting Walls (APIO20_paint) | C++17 | 0 ms | 0 KB |
#include<bits/stdc++.h> using namespace std; #define ll long long #define pb push_back #define ff first #define ss second #define all(s) s.begin(),s.end() #define rall(s) s.rbegin(),s.rend() int minimumInstructions(int n,int m,int k,vector<int>C,vector<int>A,vector<vector<int>>B){ vector<int>cnt(k,0); int mx=0; for(int i=0;i<m;i++){ for(auto x:B[i]){ cnt[x]++; mx=max(mx,cnt[x]); } } if(mx<=1){ vector<int>can(k,-1); for(int i=0;i<m;i++){ for(int j=0;j<A[i];j++){ can[B[i][j]]=i; } } for(int i=0;i<n;i++){ if(can[C[i]]==-1) return -1; } vector<int>count(n,0); int cur=1; count[n-1]=1; for(int i=n-2;i>=0;i--){ if(can[C[i]]==can[C[i+1]]-1||(can[C[i+1]]==0&&can[C[i]]==m-1)){ cur++; } else{ cur=1; } count[i]=cur; } int ans=0; bool ok=true; if(count[0]<m) return -1; ans++; int r=m-1,nxt=0; for(int i=1;i<n;i++){ if(count[i]>=m){ nxt=i; } if(i>r){ r=nxt+m-1; if(i>r) return -1; ans++; } } return ans; } vector<vector<int>>q(n,vector<int>(m)); vector<vector<bool>>have(k,vector<bool>(m,0)); vector<int>cnt(k,0); int mx=0; for(int i=0;i<m;i++){ for(auto x:B[i]){ have[x][i]=true; cnt[x]++; mx=max(mx,cnt[x]); } } for(int i=n-1;i>=0;i--){ for(int j=0;j<m;j++){ if(have[C[i]][j]==false){ q[i][j]=-1e9; } else if(i==n-1||have[C[i+1]][(j+1)%m]==false){ q[i][j]=1; } else{ q[i][j]=q[i+1][(j+1)%m]+1; } } } vector<int>mn(n+1,1e9); mn[n]=0; multiset<int>st; for(int i=n-m+1;i<=n;i++){ st.insert(mn[i]); } for(int i=n-m;i>=0;i--){ bool ok=false; for(int j=0;j<m;j++){ if(q[i][j]>=m){ ok=true; break; } } if(ok){ mn[i]=*st.begin()+1; } st.erase(st.find(mn[i+m])); st.insert(mn[i]); } if(mn[0]>=1e9) return -1; return mn[0]; }