PTA L1-049 天梯赛座位分配

#include <iostream>
using namespace std;
int ans[101][11][11],a[101];
int maxx,n,m,k,num,pre;
int main() {
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		maxx=max(a[i],maxx);
	}
	for(int i=1;i<=maxx;i++){
		for(int j=1;j<=10;j++){
			for(int k=1;k<=n;k++){
				if(i<=a[k]){
					if(pre==k) num+=2;
					else num++;
					ans[k][i][j]=num;
					pre=k;
				}
			}
		}
	}
	for(int i=1;i<=n;i++){
		cout<<"#"<<i<<endl;
		for(int j=1;j<=a[i];j++){
			for(int k=1;k<=10;k++){
				if(k!=1) cout<<" ";
				cout<<ans[i][j][k];
			}
			cout<<endl;
		}
	}
	return 0;
}