ソートしてからの出力がわからない

Question

AtCoderのABCコンテスト142 C問題 Go to Schoolについて質問がございます。
#問題文
高橋くんは
N人の生徒たちのいるクラスの担当教師です。
生徒たちには1からNまでの出席番号が重複なく割り当てられています。
今日は全ての生徒たちが相異なるタイミングで登校しました。
高橋くんは、出席番号 iの生徒が登校した時点で、教室に
Ai人の生徒たちがいたことを記録しています(出席番号iの生徒を含む)。

記録された情報を元に、生徒たちの登校した順番を復元してください。
#制約
・1≤N≤10^5
・1≤Ai≤N
・Ai≠Aj(i≠j)
入力はすべて整数

となっている中で解説ではソートを使っていなかったのですが自分は以下のようにソートを使おうとしました。

#include <bits/stdc++.h>
using std::cout;
using std::cin;
using std::endl;
using std::vector;

int main(){
  int n;
  cin >> n;
  vector<int> a(n);
  for (int i=0;i<n;i++) cin >> a[i];
  
  for (int i=0;i<n;i++){
   sort(a.begin(),a.end());
   cout << i << endl;
　　return 0;
  }
}

これでコンパイルを試みようとするとWA(Wrong Answer)となってしまいます。なんとなくこれではダメなのはわかるのですがどう改善したらいいのかわからないので改善方法がわかる方いらっしゃいましたら教えていただけると嬉しいです。よろしくお願い致します。

Answer 1

for文の中でreturnしてしまっているので、出力が一回しかないのが問題かと。
あと、sortもfor文の前にやるほうが圧倒的に計算量が短縮されます。

Comments

[Luke02561]

もう一つ、以上のことを訂正しても(0 1 2 3 ... n となるので)WAになるかと。
ちなみに、僕はCで通しました
https://atcoder.jp/contests/abc142/submissions/7760921

[cunwe]

初心者なのでいろいろ勉強になります、ありがとうございます。