結果
問題 |
No.3214 small square
|
ユーザー |
![]() |
提出日時 | 2025-07-25 22:48:20 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 292 ms / 3,000 ms |
コード長 | 7,193 bytes |
コンパイル時間 | 2,774 ms |
コンパイル使用メモリ | 216,624 KB |
実行使用メモリ | 28,080 KB |
最終ジャッジ日時 | 2025-07-25 22:48:33 |
合計ジャッジ時間 | 12,524 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 40 |
ソースコード
//#pragma GCC target("avx2") //#pragma GCC optimize("O3") //#pragma GCC optimize("unroll-loops") #include <bits/stdc++.h> using namespace std; using ll = long long; using pii = pair<int,int>; using pll = pair<ll,ll>; using pli = pair<ll,int>; #define AMARI 998244353 //#define AMARI 1000000007 #define el '\n' #define El '\n' #define YESNO(x) ((x) ? "Yes" : "No") #define YES YESNO(true) #define NO YESNO(false) #define EXIT_ANS(x) {cout << (x) << '\n'; return;} template <typename T> void inline SORT(vector<T> &v){sort(v.begin(),v.end()); return;} template <typename T> void inline REV(vector<T> &v){reverse(v.begin(),v.end()); return;} template <typename T> void inline VEC_UNIQ(vector<T> &v){sort(v.begin(),v.end()); v.erase(unique(v.begin(),v.end()),v.end()); return;} template <typename T> T inline MAX(vector<T> &v){return *max_element(v.begin(),v.end());} template <typename T> T inline MIN(vector<T> &v){return *min_element(v.begin(),v.end());} template <typename T> T inline SUM(vector<T> &v){T ans = 0; for(int i = 0; i < (int)v.size(); i++)ans += v[i]; return ans;} template <typename T> void inline DEC(vector<T> &v){for(int i = 0; i < (int)v.size(); i++)v[i]--; return;} template <typename T> void inline INC(vector<T> &v){for(int i = 0; i < (int)v.size(); i++)v[i]++; return;} void inline TEST(void){cerr << "TEST" << endl; return;} template <typename T> bool inline chmin(T &x,T y){ if(x > y){ x = y; return true; } return false; } template <typename T> bool inline chmax(T &x,T y){ if(x < y){ x = y; return true; } return false; } template <typename T = long long> vector<T> inline get_vec(int n){ vector<T> ans(n); for(int i = 0; i < n; i++)cin >> ans[i]; return ans; } template <typename T> void inline print_vec(vector<T> &vec,bool kaigyou = false){ int n = (int)vec.size(); for(int i = 0; i < n; i++){ cout << vec[i]; if(kaigyou || i == n - 1)cout << '\n'; else cout << ' '; } if(!n)cout << '\n'; return; } template <typename T> void inline debug_vec(vector<T> &vec,bool kaigyou = false){ int n = (int)vec.size(); for(int i = 0; i < n; i++){ cerr << vec[i]; if(kaigyou || i == n - 1)cerr << '\n'; else cerr << ' '; } if(!n)cerr << '\n'; return; } vector<vector<int>> inline get_graph(int n,int m = -1,bool direct = false){ if(m == -1)m = n - 1; vector<vector<int>> g(n); while(m--){ int u,v; cin >> u >> v; u--; v--; g[u].push_back(v); if(!direct)g[v].push_back(u); } return g; } template <typename T> vector<vector<T>> inline get_weighted_graph(int n,int m = -1,bool direct = false){ if(m == -1)m = n - 1; vector<vector<pair<T,int>>> g(n); while(m--){ int u,v; cin >> u >> v; u--; v--; ll w; cin >> w; g[u].push_back(pair(w,v)); if(!direct)g[v].push_back(pair(w,u)); } return g; } vector<int> make_inv(vector<int> p){ int n = (int)p.size(); vector<int> ans(n); for(int i = 0; i < n; i++)ans[p[i]] = i; return ans; } // 抽象再帰セグ木?(抽象セグ木の定義分からんな)(遅延ではない) template <typename T> class ococo_segtree { T e = {0LL,LLONG_MIN / 2LL}; T func(T a, T b) { T ans; ans[0] = a[0] + b[0]; ans[1] = max(a[1],a[0] + b[1]); return ans; } public: int n; T tmax; // range minimum query vector<T> rmq; //(データの大きさ,単位元) ococo_segtree(int N) { syokica(N); } // 配列の初期化 O(N) void syokica(int a) { n = 1; tmax = e; while(n < a) n *= 2; rmq.resize(2 * n - 1); for(int i = 0; i < 2 * n - 1; i++) { rmq[i] = tmax; } } // a[i]をxにする O(logN) void update_num(int i, T x) { i += (n - 1); rmq[i] = x; while(i) { i = (i - 1) / 2; rmq[i] = func(rmq[i * 2 + 1], rmq[i * 2 + 2]); } } // a[i]にxを加える O(logN) void add_num(int i, T x) { i += (n - 1); rmq[i] += x; while(i) { i = (i - 1) / 2; rmq[i] = func(rmq[i * 2 + 1], rmq[i * 2 + 2]); } } T get_minimum2(int a, int b, int k, int l, int r) { if(a <= l && r <= b) return rmq[k]; else if(r <= a || b <= l) return tmax; else return func(get_minimum2(a, b, k * 2 + 1, l, (l + r) / 2), get_minimum2(a, b, k * 2 + 2, (l + r) / 2, r)); } //[l,r]の区間演算の取得 O(logN) T get_val(int l, int r) { return get_minimum2(l, r + 1, 0, 0, n); } }; #define MULTI_TEST_CASE false void solve(void){ //問題を見たらまず「この問題設定から言えること」をいっぱい言う //よりシンプルな問題に言い換えられたら、言い換えた先の問題を自然言語ではっきりと書く //複数の解法のアイデアを思いついた時は全部メモしておく //g++ -D_GLIBCXX_DEBUG -Wall -O2 f.cpp -o o int n; cin >> n; ll a; cin >> a; vector<ll> x(n),y(n),v(n); vector<ll> xa,ya; a *= 2LL; for(int i = 0; i < n; i++){ cin >> x[i] >> y[i] >> v[i]; x[i] *= 2LL; y[i] *= 2LL; xa.push_back(x[i]); xa.push_back(x[i] + a + 1); ya.push_back(y[i]); ya.push_back(y[i] + a + 1); } VEC_UNIQ(xa); VEC_UNIQ(ya); //{x,y,val} vector<array<ll,3>> va(4 * n); for(int i = 0; i < n; i++){ ll xpidx = distance(xa.begin(),lower_bound(xa.begin(),xa.end(),x[i])); ll ypidx = distance(ya.begin(),lower_bound(ya.begin(),ya.end(),y[i])); ll xmidx = distance(xa.begin(),lower_bound(xa.begin(),xa.end(),x[i] + a + 1)); ll ymidx = distance(ya.begin(),lower_bound(ya.begin(),ya.end(),y[i] + a + 1)); va[4 * i + 0] = {xpidx,ypidx,v[i]}; va[4 * i + 1] = {xpidx,ymidx,-1LL * v[i]}; va[4 * i + 2] = {xmidx,ypidx,-1LL * v[i]}; va[4 * i + 3] = {xmidx,ymidx,v[i]}; } auto comp = [](array<ll,3> l,array<ll,3> r){ if(l[0] == r[0])return (l[2] < r[2]); return (l[0] < r[0]); }; sort(va.begin(),va.end(),comp); //ここで、セグ木内の prefix_sum の最大値が答えに関わってくる ococo_segtree<array<ll,2>> sgt(2 * n); //↓ vec に入れる用の値 vector<ll> raw_vec(2 * n); ll ans = 0LL; int idx = 0; for(int i = 0; i <4 *n ;i++){ ll xidx,yidx,val; xidx = va[i][0]; yidx = va[i][1]; val = va[i][2]; //cerr << xidx << ' ' << yidx << ' ' << val << el; raw_vec[yidx] += val; sgt.update_num(yidx,{raw_vec[yidx],max(0LL,raw_vec[yidx])}); //debug_vec(raw_vec); chmax(ans,sgt.get_val(0,2 * n - 1)[1]); } cout << ans << el; return; } void calc(void){ return; } signed main(void){ cin.tie(nullptr); ios::sync_with_stdio(false); calc(); int t = 1; if(MULTI_TEST_CASE)cin >> t; while(t--){ solve(); } return 0; }