oj.uz

Submission #402034

# Submission time Handle Problem Language Result Execution time Memory
402034 2021-05-11T08:27:35 Z Collypso Painting Walls (APIO20_paint) C++17
0 / 100
 1 ms 204 KB 
#include <bits/stdc++.h>
#define ll long long
#define ld long double
#define vt vector
#define pb push_back
#define all(x) (x).begin(), (x).end()
#define sz(x) (int) (x).size()
#pragma GCC optimize ("O3")
#pragma GCC optimize ("O2")
//#define endl '\n'
//#define int long long

using namespace std;

bool exists(vt<int>& master, int m)
{
    auto it = lower_bound(all(master), m);
    if (it == master.end()) return 0;
    return (*it == m);
}

int minimumInstructions(int N, int M, int K, vt<int> C, vt<int> A, vt<vt<int>> B)
{
    vt<vt<int>> master(K);
    for(int i = 0; i < M; i++)
        for(int j = 0; j < A[i]; j++) master[B[i][j]].pb(i);
    vt<vt<pair<int, int>>> dp(N);
    for(int m : master[C[0]]) dp[0].pb({m, 1});
    for(int i = 1; i < N; i++)
    {
        for(int m : master[C[i]])
        {
            int preM = (m + M - 1) % M, len = 1;
            auto it = lower_bound(all(dp[i - 1]), make_pair(preM, 0));
            if (it != dp[i - 1].end())
                if ((*it).first == preM) len = min(M, (*it).second + 1);
            dp[i].pb({m, len});
        }
    }
    int ans = 0, pos = -1;
    for(int i = M - 1, d = 1, flag = 0; i < N && d > 0; i++, d--)
    {
        for(auto m : dp[i]) m.second == M ? pos = i, flag = 1 : 0;
        if ((d == 1 || i == N - 1) && flag) ans++, d = M + 1;
    }
    if (pos == N - 1) return ans;
    return -1;
}

/**
main()
{
	ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    freopen("input.txt", "r", stdin);
    int N, M, K;
    cin >> N >> M >> K;
    vt<int> C(N), A(M);
    vt<vt<int>> B(M);
    for(int i = 0; i < N; i++) cin >> C[i];
    for(int i = 0; i < M; i++)
    {
        cin >> A[i];
        B[i].resize(A[i]);
        for(int j = 0; j < A[i]; j++) cin >> B[i][j];
    }

    //cout << M << endl;
    int ans = minimumInstructions(N, M, K, C, A, B);
    cout << ans << endl;

    //int ans = minimumInstructions(6, 4, 4, {2, 1, 0, 3, 2, 1}, {3}, {{2}, {1}, {0}, {3}});
    //cout << ans << endl;
}
/**/

Compilation message

paint.cpp:74:1: warning: "/*" within comment [-Wcomment]
   74 | /**/
      |

Subtask #1
0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 204 KB Output is correct
2 Correct 1 ms 204 KB Output is correct
3 Correct 1 ms 204 KB Output is correct
4 Correct 1 ms 204 KB Output is correct
5 Correct 1 ms 204 KB Output is correct
6 Correct 1 ms 204 KB Output is correct
7 Correct 1 ms 204 KB Output is correct
8 Correct 1 ms 204 KB Output is correct
9 Incorrect 1 ms 204 KB Output isn't correct
10 Halted 0 ms 0 KB -

Subtask #2
0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 204 KB Output is correct
2 Correct 1 ms 204 KB Output is correct
3 Correct 1 ms 204 KB Output is correct
4 Correct 1 ms 204 KB Output is correct
5 Correct 1 ms 204 KB Output is correct
6 Correct 1 ms 204 KB Output is correct
7 Correct 1 ms 204 KB Output is correct
8 Correct 1 ms 204 KB Output is correct
9 Incorrect 1 ms 204 KB Output isn't correct
10 Halted 0 ms 0 KB -

Subtask #3
0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 204 KB Output is correct
2 Correct 1 ms 204 KB Output is correct
3 Correct 1 ms 204 KB Output is correct
4 Correct 1 ms 204 KB Output is correct
5 Correct 1 ms 204 KB Output is correct
6 Correct 1 ms 204 KB Output is correct
7 Correct 1 ms 204 KB Output is correct
8 Correct 1 ms 204 KB Output is correct
9 Incorrect 1 ms 204 KB Output isn't correct
10 Halted 0 ms 0 KB -

Subtask #4
0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 204 KB Output is correct
2 Correct 1 ms 204 KB Output is correct
3 Correct 1 ms 204 KB Output is correct
4 Correct 1 ms 204 KB Output is correct
5 Correct 1 ms 204 KB Output is correct
6 Correct 1 ms 204 KB Output is correct
7 Correct 1 ms 204 KB Output is correct
8 Correct 1 ms 204 KB Output is correct
9 Incorrect 1 ms 204 KB Output isn't correct
10 Halted 0 ms 0 KB -

Subtask #5
0 / 37.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 204 KB Output is correct
2 Correct 1 ms 204 KB Output is correct
3 Correct 1 ms 204 KB Output is correct
4 Correct 1 ms 204 KB Output is correct
5 Correct 1 ms 204 KB Output is correct
6 Correct 1 ms 204 KB Output is correct
7 Correct 1 ms 204 KB Output is correct
8 Correct 1 ms 204 KB Output is correct
9 Incorrect 1 ms 204 KB Output isn't correct
10 Halted 0 ms 0 KB -