結果

問題 No.2900 Star Divine
コンテスト
ユーザー 獅子座じゃない人
提出日時 2024-09-19 22:34:40
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 99 ms / 2,000 ms
コード長 1,746 bytes
コンパイル時間 1,303 ms
コンパイル使用メモリ 113,624 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-09-20 00:28:12
合計ジャッジ時間 2,968 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <iostream>
#include <utility>
#include <vector>
using namespace std;

void solve() {
    int x,y,m;
    cin >> x >> y >> m;
    vector<pair<int, int>> uv(m);
    for(int i=0;i<m;++i){
        cin >> uv[i].first >> uv[i].second;
        --uv[i].first; --uv[i].second;
    }

    sort(uv.begin(), uv.end());
    uv.erase(unique(uv.begin(), uv.end()), uv.end());

    int m_=uv.size();
    vector<int> u(m_);
    vector<int> v(m_);
    vector g(y, vector<int>());
    vector<int> largest_star(x);
    for(int i=0;i<m_;++i){
        u[i]=uv[i].first;
        v[i]=uv[i].second;
        g[v[i]].emplace_back(u[i]);
        if(v[i]>largest_star[u[i]]){
            largest_star[u[i]]=v[i];
        }
    }

    vector<int> blue;
    vector<int> red;
    vector<bool> is_odd(x);
    for(int r=0;r<y;++r){
        vector<int> odd, even;
        for(int b: g[r]){
            if(largest_star[b]==r){
                if(is_odd[b]){
                    odd.emplace_back(b);
                } else {
                    even.emplace_back(b);
                }
            }
        }
        if(odd.size()>even.size()){
            for(int b: odd){
                blue.emplace_back(b);
            }
        } else {
            red.emplace_back(r);
            for(int b: even){
                blue.emplace_back(b);
            }
            for(int b: g[r]){
                is_odd[b].flip();
            }
        }
    }

    cout << blue.size() << " " << red.size() << endl;
    for(int b: blue){
        cout << b+1 << " ";
    }
    cout << endl;
    for(int r: red){
        cout << r+1 << " ";
    }
    cout << endl;
}

int main(void) {
    int t;
    cin >> t;
    while(t--){
        solve();
    }
    return 0;
}
0