結果

問題 No.2596 Christmas Eve (Heuristic ver.)
ユーザー FplusFplusF
提出日時 2023-12-24 13:52:22
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 5 ms / 1,224 ms
コード長 2,527 bytes
コンパイル時間 4,048 ms
コンパイル使用メモリ 275,204 KB
実行使用メモリ 6,676 KB
スコア 2,420,316
最終ジャッジ日時 2023-12-24 13:52:38
合計ジャッジ時間 15,003 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge11
純コード判定しない問題か言語
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 125
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using pii=pair<int,int>;
using tii=tuple<int,int,int>;
using qii=tuple<int,int,int,int>;
using ll=long long;
using ld=long double;
const int INF=1e9;
#define rep(i,n) for (int i=0;i<(int)(n);i++)
#define all(v) v.begin(),v.end()
template<class T> void chmin(T &a,T b){
    if(a>b){
        a=b;
    }
}
template<class T> void chmax(T &a,T b){
    if(a<b){
        a=b;
    }
}
struct Tree{
    int w,h;
};
int N,K;
vector<vector<Tree>> tree;  //0->back,1->front->2->center
struct Solver{
    void solve(){
        set<pii> st_back,st_center;
        rep(i,N) st_back.insert({tree[0][i].w,i});
        rep(i,2*N) st_center.insert({tree[2][i].w,i});
        vector<vector<int>> ans(K,vector<int>(4,-10));   //back->front->center1->center2
        int cnt_tree=0;
        vector<int> front_ord(N),center_ord(2*N);
        rep(i,N) front_ord[i]=i;
        rep(i,2*N) center_ord[i]=i;
        sort(all(front_ord),[&](int i,int j){return tree[1][i].w<tree[1][j].w;});
        sort(all(center_ord),[&](int i,int j){return tree[2][i].w<tree[2][j].w;});
        for(auto &i:front_ord){
            for(auto &j:center_ord){
                if(tree[2][j].w<=tree[1][i].w) continue;
                if(st_center.find({tree[2][j].w,j})==st_center.end()) continue;
                auto itr_back=st_back.lower_bound({tree[1][i].w,-INF});
                if(itr_back==st_back.begin()) continue;
                itr_back--;
                auto itr_center=st_center.upper_bound({tree[2][j].w,INF});
                if(itr_center==st_center.end()) continue;
                ans[cnt_tree]={(*itr_back).second,i,j,(*itr_center).second};
                st_back.erase(itr_back);
                st_center.erase(itr_center);
                st_center.erase({tree[2][j].w,j});
                cnt_tree++;
                break;
            }
            if(cnt_tree==K) break;
        }
        assert(cnt_tree==K);
        vector<set<int>> st(4);
        rep(i,K){
            cout << ans[i][1]+1 << ' ' << ans[i][2]+1 << ' ' << ans[i][3]+1 << ' ' << ans[i][0]+1 << '\n';
        }
    }
}Solver;
int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cin >> N >> K;
    tree.resize(3);
    tree[1].resize(N);
    tree[2].resize(2*N);
    tree[0].resize(N);
    rep(i,N) cin >> tree[1][i].w;
    rep(i,N) cin >> tree[1][i].h;
    rep(i,2*N) cin >> tree[2][i].w;
    rep(i,2*N) cin >> tree[2][i].h;
    rep(i,N) cin >> tree[0][i].w;
    rep(i,N) cin >> tree[0][i].h;
    Solver.solve();
}
0