# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
386331 | leinad2 | 벽 칠하기 (APIO20_paint) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<bits/stdc++.h>
using namespace std;
vector<vector<int> >B;
int f(int a, int m)
{
if(a<0)while(a<0)a+=m;
else while(a>=m)a-=m;
return a;
}
int minimumInstructions(int N, int M, int K, vector<int>C, vector<int>A, vector<vector<int> >bB)
{
B=bB;
int i, j, k;
vector<int>X;X.resize(N);
vector<vector<int> >S;S.resize(K);
vector<int>cnt;cnt.resize(M);
vector<int>chk;chk.resize(M+1);
for(i=0;i<M;i++)
{
for(j=0;j<A[i];j++)
{
S[B[i][j]].push_back(i);
}
}
for(i=0;i<M;i++)
{
for(j=0;j<S[C[i]].size();j++)
{
int a=f(i-S[C[i]][j], M);
chk[cnt[a]]--;
cnt[a]++;
chk[cnt[a]]++;
}
}
if(chk[M])X[0]=1;
for(i=1;i<=N-M;i++)
{
for(j=0;j<S[C[i-1]].size();j++)
{
int a=f(i-1-S[C[i-1]][j], M);
chk[cnt[a]]--;
cnt[a]--;
chk[cnt[a]]++;
}
for(j=0;j<S[C[i+M-1]].size();j++)
{
int a=f(i+M-1-S[C[i+M-1]][j], M);
chk[cnt[a]]--;
cnt[a]++;
chk[cnt[a]]++;
}
if(chk[M])X[i]=1;
}
i=0;
if(X[0]==0||X[N-M]==0)return -1;
int ans=0;
while(1)
{
if(i==N-M)return ans+1;
k=-1;
for(j=i+1;j<=i+M;j++)
{
if(X[j])k=j;
}
if(k==-1)return -1;
i=k;ans++;
}
}
int main() {
int N, M, K;
assert(3 == scanf("%d %d %d", &N, &M, &K));
vector<int> C(N);
for (int i = 0; i < N; ++i) {
assert(1 == scanf("%d", &C[i]));
}
vector<int> A(M);
vector<vector<int> > B(M);
for (int i = 0; i < M; ++i) {
assert(1 == scanf("%d", &A[i]));
B[i].resize(A[i]);
for (int j = 0; j < A[i]; ++j) {
assert(1 == scanf("%d", &B[i][j]));
}
}
int minimum_instructions = minimumInstructions(N, M, K, C, A, B);
printf("%d\n", minimum_instructions);
return 0;
}