Submission #395229

#	Time	Username	Problem	Language	Result	Execution time	Memory
395229		abc864197532	Painting Walls (APIO20_paint)	C++17	0 / 100	1 ms	204 KiB

paint

#include <bits/stdc++.h>
// #include "grader_paint.cpp"
using namespace std;
#define lli long long int
#define mp make_pair
#define pb push_back
#define eb emplace_back
#define test(x) cout << "Line(" << __LINE__ << ") " #x << ' ' << x << endl
#define printv(x) {\
	for (auto i : x) cout << i << ' ';\
	cout << endl;\
}
#define pii pair <int, int>
#define pll pair <lli, lli>
#define X first
#define Y second
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
template<typename A, typename B>
ostream& operator << (ostream& o, pair<A, B> a){
    return o << a.X << ' ' << a.Y;
}
template<typename A, typename B>
istream& operator >> (istream& o, pair<A, B> &a){
    return o >> a.X >> a.Y;
}

int minimumInstructions(int n, int m, int k,vector <int> c, vector <int> a, vector <vector <int>> b) {
    vector <int> tmp[n], pos[k];
    int cha[n];
    for (int i = 0; i < n; ++i) cha[i] = i < m ? i : cha[i - m];
    for (int i = 0; i < n; ++i) pos[c[i]].pb(i);
    for (int i = 0; i < m; ++i) {
        for (int j : b[i]) {
            for (int ii : pos[j]) {
                int t = ii - i;
                t = t >= 0 ? cha[t] : t + m;
                tmp[ii].pb(t);
            }
        }
    }
    int cnt[m]{}, num[m + 1]{}, dp[n + 1] = {-1};
    num[0] = m;
    dp[0] = 0;
    for (int i = 0; i < m; ++i) {
        for (int j : tmp[i]) {
            num[cnt[j]]--;
            cnt[j]++;
            num[cnt[j]]++;
        }
    }
    vector <int> dq;
    int now = 0;
    if (num[m] > 0) dp[m] = 1, dq.push_back(m);
    for (int i = m; i < n; ++i) {
        for (int j : tmp[i - m]) {
            num[cnt[j]]--;
            cnt[j]--;
            num[cnt[j]]++;
        }
        for (int j : tmp[i]) {
            num[cnt[j]]--;
            cnt[j]++;
            num[cnt[j]]++;
        }
        if (num[m] > 0) {
            if (now < dq.size() && dq[now] <= i - m) now++;
            if (now < dq.size()) dp[i + 1] = dp[dq[now]] + 1;
            if (dp[i + 1] != -1) {
                while (now < dq.size() && dp[dq.back()] >= dp[i + 1]) dq.pop_back();
                dq.push_back(i + 1);
            }
        }
    }
    return dp[n];
}

/*
 8 3 5
 3 3 1 3 4 4 2 2
 3 0 1 2
 2 2 3
 2 3 4

 5 4 4
 1 0 1 2 2
 2 0 1
 1 1
 1 2
 1 3
 */

Compilation message (stderr)

paint.cpp: In function 'int minimumInstructions(int, int, int, std::vector<int>, std::vector<int>, std::vector<std::vector<int> >)':
paint.cpp:67:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   67 |             if (now < dq.size() && dq[now] <= i - m) now++;
      |                 ~~~~^~~~~~~~~~~
paint.cpp:68:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   68 |             if (now < dq.size()) dp[i + 1] = dp[dq[now]] + 1;
      |                 ~~~~^~~~~~~~~~~
paint.cpp:70:28: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   70 |                 while (now < dq.size() && dp[dq.back()] >= dp[i + 1]) dq.pop_back();
      |                        ~~~~^~~~~~~~~~~

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