Submission #566350

#TimeUsernameProblemLanguageResultExecution timeMemory
566350zaneyuPainting Walls (APIO20_paint)C++14
100 / 100
1477 ms296224 KiB
/*input
8 3 5
3 3 1 3 4 4 2 2
3 0 1 2
2 2 3
2 3 4
 
*/
#include "paint.h"
#include <bits/stdc++.h>
using namespace std;
#define REP(i,n) for(int i=0;i<n;i++)
#define REP1(i,n) for(int i=1;i<=n;i++)
#define ALL(x) x.begin(),x.end()
#define pii pair<int,int>
#define f first
#define s second
#define MXTO(a,b) a=max(a,b)
#define MNTO(a,b) a=min(a,b)
#define sz(x) (int)x.size()
#define pb push_back
const int maxn=1e5+5;
int n,m,k;
int dp[maxn];
vector<int> ok[maxn];
const int INF=0x3f3f3f3f;
vector<int> v[maxn];
int lp[maxn],rp[maxn];
mt19937 rng(69);
int md[maxn];
int minimumInstructions(
    int N, int M, int K, std::vector<int> C,
    std::vector<int> A, std::vector<std::vector<int>> B) {
    C.emplace(C.begin(),0);
    n=N,m=M,k=K;
    multiset<int> ms;
    ms.insert(0);
    REP(i,m){
        for(int x:B[i]) ok[x].pb(i);
    } 
    REP1(i,k) shuffle(ALL(ok[i]),rng);
	REP1(i,n) md[i]=(i%m);
    REP1(i,n){
        for(int x:ok[C[i]]){
			int z=(x-md[i]+m);
			if(z>=m) z-=m;
			v[z].pb(i);
		}
    }
	
    deque<int> dq;
    dq.pb(0);
    REP1(i,m-1) dp[i]=INF,dq.pb(i);
    for(int i=m;i<=n;i++){
        while(sz(dq) and dq.front()<i-m) dq.pop_front();
        int mn=dp[dq.front()];
        dp[i]=INF;
        vector<int> ans;
        for(int a:ok[C[i]]){
            int z=(a-md[i]+m);
			if(z>=m) z-=m;
            while(rp[z]<sz(v[z]) and v[z][rp[z]]<=i) ++rp[z];
            while(lp[z]<sz(v[z]) and v[z][lp[z]]<i-m+1) ++lp[z];
            if(rp[z]-lp[z]==m){
                dp[i]=mn+1;
                break;
            }
        }
        while(sz(dq) and dp[dq.back()]>dp[i]) dq.pop_back();
        dq.pb(i);
    }
    if(dp[n]>=INF) dp[n]=-1;
    return dp[n];
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...