Submission #1511732

---

Source Code Expand

#include <bits/stdc++.h>
using namespace std;
const int MAX = 501;

int n, cnt[MAX], res[MAX*MAX];
pair<int,int> x[MAX];
set<int> pos;
int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	cin >> n;
	for(int i = 1; i <= n*n; i++) pos.insert(i);
	for(int i = 1; i <= n; i++) cnt[i] = n;
	for(int i = 1; i <= n; i++) {
		cin >> x[i].first;
		x[i].second = i;
	}
	sort(x+1, x+n+1);
	for(int i = 1; i <= n; i++) {
		if(x[i].first == x[i-1].first) cout << "No", exit(0);
		for(int j = 0; j < x[i].second-1; j++) {
			if(pos.empty() || *pos.begin() >= x[i].first) cout << "No", exit(0);
			res[*pos.begin()] = x[i].second;
			--cnt[x[i].second];
			pos.erase(pos.begin());
		}
		res[x[i].first] = x[i].second;
		--cnt[x[i].second];
		pos.erase(x[i].first);
	}
	for(int i = 1, j = 1; i <= n*n; i++) if(!res[i]) {
		while(j <= n && !cnt[x[j].second]) ++j;
		if(i <= x[j].first) cout << "No", exit(0);
		res[i] = x[j].second;
		--cnt[x[j].second];
	}
	cout << "Yes\n";
	for(int i = 1; i <= n*n; i++) cout << res[i] << ' ';
}

Submission Info

Judge Result