#962146 (C++17) No.2680 研究室配属

• 問題: No.2680 研究室配属
• ユーザー: cho435
• 提出日時: 2024-03-20 21:08:01
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 240 ms / 2,000 ms
• コード長: 773 bytes
• コンパイル時間: 3,877 ms
• コンパイル使用メモリ: 271,212 KB
• 実行使用メモリ: 10,112 KB
• 最終ジャッジ日時: 2024-09-30 06:53:53
• 合計ジャッジ時間: 6,039 ms
• ジャッジサーバーID （参考情報）: judge3 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 6,820 KB
testcase_01	AC	1 ms; 6,816 KB
testcase_02	AC	1 ms; 6,816 KB
testcase_03	AC	2 ms; 6,820 KB
testcase_04	AC	2 ms; 6,820 KB
testcase_05	AC	6 ms; 6,820 KB
testcase_06	AC	7 ms; 6,820 KB
testcase_07	AC	6 ms; 6,816 KB
testcase_08	AC	6 ms; 6,816 KB
testcase_09	AC	3 ms; 6,816 KB
testcase_10	AC	27 ms; 6,820 KB
testcase_11	AC	26 ms; 6,816 KB
testcase_12	AC	36 ms; 6,816 KB
testcase_13	AC	25 ms; 6,820 KB
testcase_14	AC	39 ms; 6,816 KB
testcase_15	AC	6 ms; 6,816 KB
testcase_16	AC	88 ms; 6,816 KB
testcase_17	AC	16 ms; 6,816 KB
testcase_18	AC	53 ms; 6,816 KB
testcase_19	AC	115 ms; 6,816 KB
testcase_20	AC	91 ms; 6,820 KB
testcase_21	AC	10 ms; 6,816 KB
testcase_22	AC	66 ms; 6,820 KB
testcase_23	AC	187 ms; 7,552 KB
testcase_24	AC	187 ms; 7,552 KB
testcase_25	AC	240 ms; 10,112 KB

ソースコード

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using ll = long long;
#define rep(i,n) for(ll i=0;i<(ll)(n);i++)
using mint = atcoder::modint998244353;

int main(){
	int n,m;
	cin>>n>>m;
	vector<int> a(m);
	rep(i,m) cin>>a.at(i);
	vector<vector<int>> t(n,vector<int>(m,0));
	rep(i,n) rep(j,m) cin>>t.at(i).at(j);
	vector<vector<int>> ans(m);
	set<int> st;
	rep(i,n) st.insert(i);
	rep(c,m){
		if(st.empty()) break;
		for(int x:st){
			ans.at(t.at(x).at(c)).push_back(x);
		}
		set<int> nst;
		rep(i,m){
			while((int)ans.at(i).size()>a.at(i)){
				nst.insert(ans.at(i).back());
				ans.at(i).pop_back();
			}
		}
		swap(st,nst);
	}
	vector<int> v(n);
	rep(i,m){
		for(int x:ans.at(i)){
			v.at(x)=i;
		}
	}
	rep(i,n) cout<<v.at(i)<<" ";
	cout<<endl;
}