Submission #364764

#	Time	Username	Problem	Language	Result	Execution time	Memory
364764		mathking1021	Painting Walls (APIO20_paint)	C++17	100 / 100	899 ms	18672 KiB

paint

#include "paint.h"
#include <vector>
#include <queue>
#define PLL pair<ll, ll>
#define F first
#define S second

using namespace std;

typedef int ll;

ll n, m, k;
vector<vector<ll>> ve;
vector<ll> ve2[100005], c;
ll lt[100005];
ll d[100005];
ll e[100005];
bool v[100005];
priority_queue<PLL, vector<PLL>, greater<PLL> > pq;

int minimumInstructions(
    int N, int M, int K, vector<int> C,
    vector<int> A, vector<vector<int>> B)
{
    n = N;
    m = M;
    k = K;
    ve = B;
    c = C;
    for(ll i = 0; i < ve.size(); i++)
    {
        for(ll j = 0; j < ve[i].size(); j++)
        {
            ve2[ve[i][j]].push_back(i);
        }
    }
    for(ll i = 0; i < 100000; i++) lt[i] = -1;
    for(ll i = 0; i < c.size(); i++)
    {
        for(ll j = 0; j < ve2[c[i]].size(); j++)
        {
            ll t = ve2[c[i]][j];
            if(lt[(t - i % m + m) % m] + 1 == i) d[(t - i % m + m) % m]++;
            else d[(t - i % m + m) % m] = 1;
            lt[(t - i % m + m) % m] = i;
            if(d[(t - i % m + m) % m] >= m) v[i] = true;
        }
    }
    if(!v[m - 1]) return -1;
    pq.push(make_pair(1, m - 1));
    e[m - 1] = 1;
    for(ll i = m; i < n; i++)
    {
        while(!pq.empty())
        {
            if(pq.top().S < i - m) pq.pop();
            else break;
        }
        if(pq.empty()) return -1;
        ll t1 = pq.top().S;
        ll t2 = pq.top().F;
        if(v[i]) e[i] = t2 + 1, pq.push(make_pair(e[i], i));
        else if(i == n - 1) return -1;
    }
    return e[n - 1];
}

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:30:21: warning: comparison of integer expressions of different signedness: 'll' {aka 'int'} and 'std::vector<std::vector<int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   30 |     for(ll i = 0; i < ve.size(); i++)
      |                   ~~^~~~~~~~~~~
paint.cpp:32:25: warning: comparison of integer expressions of different signedness: 'll' {aka 'int'} and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   32 |         for(ll j = 0; j < ve[i].size(); j++)
      |                       ~~^~~~~~~~~~~~~~
paint.cpp:38:21: warning: comparison of integer expressions of different signedness: 'll' {aka 'int'} and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   38 |     for(ll i = 0; i < c.size(); i++)
      |                   ~~^~~~~~~~~~
paint.cpp:40:25: warning: comparison of integer expressions of different signedness: 'll' {aka 'int'} and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   40 |         for(ll j = 0; j < ve2[c[i]].size(); j++)
      |                       ~~^~~~~~~~~~~~~~~~~~
paint.cpp:60:12: warning: unused variable 't1' [-Wunused-variable]
   60 |         ll t1 = pq.top().S;
      |            ^~

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