結果
問題 |
No.2680 研究室配属
|
ユーザー |
|
提出日時 | 2024-03-20 21:18:32 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 95 ms / 2,000 ms |
コード長 | 1,501 bytes |
コンパイル時間 | 2,981 ms |
コンパイル使用メモリ | 273,196 KB |
実行使用メモリ | 12,672 KB |
最終ジャッジ日時 | 2024-09-30 07:04:56 |
合計ジャッジ時間 | 4,614 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 23 |
ソースコード
// #pragma GCC optimize("O3,unroll-loops") #include <bits/stdc++.h> // #include <x86intrin.h> using namespace std; #if __cplusplus >= 202002L using namespace numbers; #endif int main(){ cin.tie(0)->sync_with_stdio(0); cin.exceptions(ios::badbit | ios::failbit); int n, m; cin >> n >> m; vector<int> cap(m); copy_n(istream_iterator<int>(cin), m, cap.begin()); vector<vector<int>> prefer(n, vector<int>(m)); for(auto &x: prefer | ranges::views::join){ cin >> x; } vector<priority_queue<array<int, 2>>> cur(m); vector<int> exceeded_lab, active_lab; for(auto u = 0; u < n; ++ u){ int i = prefer[u][0]; if((int)cur[i].size() == cap[i]){ exceeded_lab.push_back(i); } cur[i].push({u, 0}); active_lab.push_back(i); } vector<int> res(n, -1); while(!active_lab.empty()){ vector<array<int, 2>> rem; for(auto i: exceeded_lab){ while((int)cur[i].size() > cap[i]){ auto [u, p] = cur[i].top(); cur[i].pop(); rem.push_back({u, p + 1}); } } for(auto i: active_lab){ assert((int)cur[i].size() <= cap[i]); while(!cur[i].empty()){ auto [u, p] = cur[i].top(); cur[i].pop(); res[u] = i; -- cap[i]; } } exceeded_lab.clear(); active_lab.clear(); for(auto [u, p]: rem){ assert(p < m); int i = prefer[u][p]; if((int)cur[i].size() == cap[i]){ exceeded_lab.push_back(i); } cur[i].push({u, p}); active_lab.push_back(i); } } ranges::copy(res, ostream_iterator<int>(cout, " ")); cout << "\n"; return 0; } /* */