Yukicoder No.165 四角で囲え!

,

http://yukicoder.me/problems/no/165

solution

片側固定してしゃくとり法。$O(N^3)$。

単純に左上と右下について全列挙すると、どうやっても$O(N^4)$になる。 $3$つ決めてもう$1$つは一意に定まる、のようにして$O(N^3)$にしたい。

$y$軸について区間$[l_y, r_y)$を固定し、$x$軸について左端$l_x$をとれば、右端$r_x$は一意にさだまる。 このようにするしゃくとり法で、$O(N^3)$となる。

座標圧縮をしておくと楽だが必須ではない。

implementation

#include <iostream>
#include <vector>
#include <algorithm>
#include <numeric>
#include <map>
#define repeat(i,n) for (int i = 0; (i) < int(n); ++(i))
#define whole(f,x,...) ([&](decltype((x)) whole) { return (f)(begin(whole), end(whole), ## __VA_ARGS__); })(x)
typedef long long ll;
using namespace std;
template <class T> void setmax(T & a, T const & b) { if (a < b) a = b; }

template <typename T>
map<T,int> coordinate_compression_map(vector<T> const & xs) {
    int n = xs.size();
    vector<int> ys(n);
    whole(iota, ys, 0);
    whole(sort, ys, [&](int i, int j) { return xs[i] < xs[j]; });
    map<T,int> f;
    for (int i : ys) {
        if (not f.count(xs[i])) { // make unique
            int j = f.size();
            f[xs[i]] = j; // f[xs[i]] has a side effect, increasing the f.size()
        }
    }
    return f;
}
template <typename T>
vector<int> apply_compression(map<T,int> const & f, vector<T> const & xs) {
    int n = xs.size();
    vector<int> ys(n);
    repeat (i,n) ys[i] = f.at(xs[i]);
    return ys;
}

int main() {
    int n, b; cin >> n >> b;
    vector<int> x(n), y(n), p(n); repeat (i,n) cin >> x[i] >> y[i] >> p[i];
    x = apply_compression(coordinate_compression_map(x), x);
    y = apply_compression(coordinate_compression_map(y), y);
    vector<int> j(n);
    whole(iota, j, 0);
    whole(sort, j, [&](int i, int j) { return x[i] < x[j]; });
    int ans = 0;
    repeat (ry,n+1) repeat (ly,ry) {
        int li = 0, ri = 0;
        int lx = 0, rx = 0;
        int cnt = 0, sum_p = 0;
        while (rx < n) {
            ++ rx;
            while (ri < n and x[j[ri]] < rx) {
                if (ly <= y[j[ri]] and y[j[ri]] < ry) {
                    cnt += 1;
                    sum_p += p[j[ri]];
                }
                ++ ri;
            }
            while (b < sum_p) {
                ++ lx;
                while (li < n and x[j[li]] < lx) {
                    if (ly <= y[j[li]] and y[j[li]] < ry) {
                        cnt -= 1;
                        sum_p -= p[j[li]];
                    }
                    ++ li;
                }
            }
            setmax(ans, cnt);
        }
    }
    cout << ans << endl;
    return 0;
}