내 뒤에 나와 다른 수 (백준 24523)

안녕하세요. 오늘은 내 뒤에 나와 다른 수를 찾아볼 거예요.

문제

https://www.acmicpc.net/problem/24523

아이디어

ans[N]=-1입니다.
ans[N-1]은 무엇일까요? arr[N-1]==arr[N]이면 ans[N-1]=-1이 되고 아니면 N이 됩니다. 이를 통해서 알 수 있는 사실이 있습니다.
ans[i]는 arr[i]==arr[i+1]이면 ans[i+1], 아니면 i+1입니다.

소스코드

#include <iostream>
#define ll long long
using namespace std;


int main(void)
{
    ios_base::sync_with_stdio(false); cin.tie(NULL);
    ll arr[1010101] = { 0 }, ans[1010101] = { 0 }, N, i;

    cin >> N;
    for (i = 1; i <= N; i++) cin >> arr[i];
    ans[N] = -1;
    for (i = N - 1; i >= 1; i--)
    {
        if (arr[i] == arr[i + 1]) ans[i] = ans[i + 1];
        else ans[i] = i + 1;
    }

    for (i = 1; i <= N; i++) cout << ans[i] << ' ';
}

감사합니다.