天梯赛-L1-049 天梯赛座位分配 (20分)

#include<iostream>
#include<vector>
#include<set>
#include<algorithm>
using namespace std;
int main(){
	int N;
	cin>>N;
	vector<vector<int>>s;
	vector<int>aa;
	set<int>TJ; 
	s.resize(N);
	int sum=0;
	for(int i=1;i<=N;i++){
		int num;
		cin>>num;
		aa.push_back(num);
		sum+=num;
	} 
	for(int j=1;TJ.size()!=N;){//一直循环到set数组内都存放了队伍 
		for(int i=0;i<N;i++){
			if(s[i].size()==aa[i]*10){//代表学校队伍人数已经满了 
				TJ.insert(i);//利用set的元素不重复性 把已经满的学校记录下来 
				continue;
			}
			if(TJ.size()!=N-1)//还没有剩下一支队伍的情况 
				s[i].push_back(j);
			if(TJ.size()==N-1){//只剩下一支队伍
				if(!s[i].empty()){
					j=s[i].back(); 
					j+=2;	
				}
				s[i].push_back(j);
				if(s[i].empty())//用来处理1个学校1支队伍的情况 
					j+=2;
			}else{
				j++;
			}		
		}
	} 
	int row=0;
	for(int i=0;i<N;i++){
		cout<<"#"<<i+1<<endl;
		for(int j=0;j<s[i].size();j++){
			cout<<s[i][j];
			if((j+1)%10==0&&row+1!=sum){//用来控制换行 
				row++;
				cout<<endl;
			}
			if((j+1)%10!=0){
                cout<<" ";
            }
		}	
	}
	return 0;
}