Submission #262875

#	Time	Username	Problem	Language	Result	Execution time	Memory
262875		SorahISA	Land of the Rainbow Gold (APIO17_rainbow)	C++11	12 / 100	127 ms	4860 KiB

rainbow

#include "rainbow.h"
#include <bits/stdc++.h>
using namespace std;

// #define int long long
#define double long double
using pii = pair<int, int>;
template<typename T>
using prior = std::priority_queue<T, vector<T>, greater<T>>;
template<typename T>
using Prior = std::priority_queue<T>;

#define X first
#define Y second
#define ALL(x) (x).begin(), (x).end()
#define eb emplace_back
#define pb push_back
#define fastIO() ios_base::sync_with_stdio(false), cin.tie(0)

const int maxn = 1 << 18;

int arr[5][maxn];
vector<pii> vec[5];

void init(int R, int C, int sr, int sc, int M, char *S) {
    arr[sr][sc] = 1;
    for (int i = 0; i < M; ++i) {
        if (S[i] == 'N') --sr;
        if (S[i] == 'S') ++sr;
        if (S[i] == 'W') --sc;
        if (S[i] == 'E') ++sc;
        arr[sr][sc] = 1;
    }
    
    int lst1 = 0, lst2 = 0;
    for (int i = 0; i <= C; ++i) {
        if ( arr[1][i] and !arr[1][i + 1]) lst1 = i + 1;
        if (!arr[1][i] and  arr[1][i + 1]) vec[1].eb(lst1, i);
        if ( arr[2][i] and !arr[2][i + 1]) lst2 = i + 1;
        if (!arr[2][i] and  arr[2][i + 1]) vec[2].eb(lst2, i);
    }
    vec[1].eb(lst1, C + 1);
    vec[2].eb(lst2, C + 1);
}

int colour(int ar, int ac, int br, int bc) {
    int cnt1L, cnt1R, cnt2L, cnt2R;
    
    if (lower_bound(ALL(vec[1]), pii{ac, 0}) != vec[1].begin() and
        (*prev(lower_bound(ALL(vec[1]), pii{ac, 0}))).Y >= ac) {
        cnt1L = lower_bound(ALL(vec[1]), pii{ac, 0}) - vec[1].begin() - 1;
    }
    else {
        cnt1L = lower_bound(ALL(vec[1]), pii{ac, 0}) - vec[1].begin();
    }
    cnt1R = upper_bound(ALL(vec[1]), pii{bc, maxn}) - vec[1].begin() - 1;
    
    if (lower_bound(ALL(vec[2]), pii{ac, 0}) != vec[2].begin() and
        (*prev(lower_bound(ALL(vec[2]), pii{ac, 0}))).Y >= ac) {
        cnt2L = lower_bound(ALL(vec[2]), pii{ac, 0}) - vec[2].begin() - 1;
    }
    else {
        cnt2L = lower_bound(ALL(vec[2]), pii{ac, 0}) - vec[2].begin();
    }
    cnt2R = upper_bound(ALL(vec[2]), pii{bc, maxn}) - vec[2].begin() - 1;
    
    // cout << cnt1L << " " << cnt1R << " " << cnt2L << " " << cnt2R << " ";
    
    int ans = (ar == 1 ? cnt1R - cnt1L + 1 : 0LL) + (br == 2 ? cnt2R - cnt2L + 1 : 0LL);
    if (ar == 1 and br == 2) {
        ans -= cnt1L == 0 and cnt2L == 0;
        ans -= cnt1R == vec[1].size() - 1 and cnt2R == vec[2].size() - 1;
    }
    
    return ans;
}

Compilation message (stderr)

rainbow.cpp: In function 'int colour(int, int, int, int)':
rainbow.cpp:72:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   72 |         ans -= cnt1R == vec[1].size() - 1 and cnt2R == vec[2].size() - 1;
      |                ~~~~~~^~~~~~~~~~~~~~~~~~~~
rainbow.cpp:72:53: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   72 |         ans -= cnt1R == vec[1].size() - 1 and cnt2R == vec[2].size() - 1;
      |                                               ~~~~~~^~~~~~~~~~~~~~~~~~~~

• Subtask 1: 0 / 11
• Subtask 2: 12 / 12
• Subtask 3: 0 / 24
• Subtask 4: 0 / 27
• Subtask 5: 0 / 26