Submission #566276

#	Time	Username	Problem	Language	Result	Execution time	Memory
566276		zaneyu	Painting Walls (APIO20_paint)	C++14	0 / 100	4 ms	5012 KiB

paint

/*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];
set<int> ok[maxn];
const int INF=0x3f3f3f3f;
int minimumInstructions(
    int N, int M, int K, std::vector<int> C,
    std::vector<int> A, std::vector<std::vector<int>> B) {
    n=N,m=M,k=K;
    multiset<int> ms;
    ms.insert(0);
    REP1(i,m-1) dp[i]=INF,ms.insert(dp[i]);
    REP(i,m){
        for(int x:B[i]) ok[x].insert(i);
    } 
    for(int i=m;i<=n;i++){
        int mn=(*ms.begin());
        dp[i]=INF;
        vector<int> ans;
        for(int a:ok[C[i-m+1]]) ans.pb(a);
        for(int j=i-m+2;j<i;j++){
            vector<int> nw;
            for(int a:ans){
                if(ok[C[j]].count((a+j-(i-m+1))%m)) nw.pb(a);
            }
            ans=nw;
            if(!sz(nw)) break;
        }
        if(sz(ans)) dp[i]=mn+1;
        ms.insert(dp[i]);
        ms.erase(ms.find(dp[i-m]));
    }
    if(dp[n]>=INF) dp[n]=-1;
    return dp[n];
}

• Subtask 1: 0 / 12
• Subtask 2: 0 / 15
• Subtask 3: 0 / 13
• Subtask 4: 0 / 23
• Subtask 5: 0 / 37