Language: C++

#define MOD 1000000007
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

int main() {
    ll N, K;
    cin >> N >> K;
    int A[N];
    unordered_map<int, vector<int>> d;
    for (int i = 0; i < N; i++) {
        cin >> A[i];
        d[A[i]].push_back(i);
    }
    ll a = K - 1, b = 0;
    unordered_set<int> s;
    while (a) {
        if (b == N) {
            a--;
            b = 0;
            continue;
        }
        if (s.count(b)) {
            a %= (K - 1 - a);
            break;
        }
        else s.insert(b);
        vector<int> l = d[A[b]];
        int i = upper_bound(l.begin(), l.end(), b);
        if (i == l.end()) {
            a--;
            b = 1 + l[0];
        }
        else b = *i + 1
    }
//     while (a) {
//         if (b == N) {
//             a--;
//             b = 0;
//             continue;
//         }
//         vector<int> l = d[A[b]];
//         int i = *upper_bound(l.begin(), l.end(), b);
//         if (i) b = i + 1;
//         else {
//             a--;
//             b = 1 + l[0];
//         }
//     }
//     vector<int> ans;
//     while (b < N) {
//         vector<int> l = d[A[b]];
//         int i = *upper_bound(l.begin(), l.end(), b);
//         if (i) b = i + 1;
//         else ans.push_back(A[b++]);
//     }
//     if (ans.size()) {
//         cout << ans[0];
//         for (int i = 1; i < ans.size(); i++) cout << " " << ans[i];
//         cout << endl;
//     }
}


Compiler output:
a.cc: In function 'int main()':
a.cc:29:28: error: cannot convert '__gnu_cxx::__normal_iterator<int*, std::vector<int> >' to 'int' in initialization
   29 |         int i = upper_bound(l.begin(), l.end(), b);
      |                 ~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~
      |                            |
      |                            __gnu_cxx::__normal_iterator<int*, std::vector<int> >
a.cc:30:15: error: no match for 'operator==' (operand types are 'int' and 'std::vector<int>::iterator')
   30 |         if (i == l.end()) {
      |             ~ ^~ ~~~~~~~
      |             |         |
      |             int       std::vector<int>::iterator
In file included from /usr/include/c++/14/regex:68,
                 from /usr/include/x86_64-linux-gnu/c++/14/bits/stdc++.h:181,
                 from a.cc:2:
/usr/include/c++/14/bits/regex.h:1103:5: note: candidate: 'template<class _BiIter> bool std::__cxx11::operator==(const sub_match<_BiIter>&, const sub_match<_BiIter>&)'
 1103 |     operator==(const sub_match<_BiIter>& __lhs, const sub_match<_BiIter>& __rhs)
      |     ^~~~~~~~
/usr/include/c++/14/bits/regex.h:1103:5: note:   template argument deduction/substitution failed:
a.cc:30:24: note:   mismatched types 'const std::__cxx11::sub_match<_BiIter>' and 'int'
   30 |         if (i == l.end()) {
      |                        ^
/usr/include/c++/14/bits/regex.h:1199:5: note: candidate: 'template<class _Bi_iter, class _Ch_traits, class _Ch_alloc> bool std::__cxx11::operator==(__sub_match_string<_Bi_iter, _Ch_traits, _Ch_alloc>&, const sub_match<_BiIter>&)'
 1199 |     operator==(const __sub_match_string<_Bi_iter, _Ch_traits, _Ch_alloc>& __lhs,
      |     ^~~~~~~~
/usr/include/c++/14/bits/regex.h:1199:5: note:   template argument deduction/substitution failed:
a.cc:30:24: note:   mismatched types 'std::__cxx11::__sub_match_string<_Bi_iter, _Ch_traits, _Ch_alloc>' and 'int'
   30 |         if (i == l.end()) {
      |                        ^
/usr/include/c++/14/bits/regex.h:1274:5: note: candidate: 'template<class _Bi_iter, class _Ch_traits, class _Ch_alloc> bool std::__cxx11::operator==(const sub_match<_BiIter>&, __sub_match_string<_Bi_iter, _Ch_traits, _Ch_alloc>&)'
 1274 |     operator==(const sub_match<_Bi_iter>& __lhs,
      |     ^~~~~~~~
/usr/include/c++/14/bits/regex.h:1274:5: note:   template argument deduction/substitution failed:
a.cc:30:24: note:   mismatched types 'const std::__cxx11::sub_match<_BiIter>' and 'int'
   30 |         if (i == l.end()) {
      |                        ^
/usr/include/c++/14/bits/regex.h:1366:5: note: candidate: 'template<class _Bi_iter> bool std::__cxx11::operator==(const typename std::iterator_traits<_Iter>::value_type*, const sub_match<_BiIter>&)'
 1366 |     operator==(typename iterator_traits<_Bi_iter>::value_type const* __lhs,
      |     ^~~~~~~~
/usr/include/c++/14/bits/regex.h:1366:5: note:   template argument deduction/substitution failed:
a.cc:30:24: note:   'std::vector<int>::iterator' is not derived from 'const std::__cxx11::sub_match<_BiIter>'
   30 |         if (i == l.end()) {
      |                        ^
/usr/include/c++/14/bits/regex.h:1441:5: note: candidate: 'template<class _Bi_iter> bool std::__cxx11::operator==(const sub_match<_BiIter>&, const typename std::iterator_traits<_Iter>::value_type*)'
 1441 |     operator==(const sub_match<_Bi_iter>& __lhs,
      |     ^~~~~~~~
/usr/include/c++/14/bits/regex.h:1441:5: note:   template argument deduction/substitution failed:
a.cc:30:24: note:   mismatched types 'const std::__cxx11::sub_match<_BiIter>' and 'int'
   30 |         if (i == l.end()) {
      |                        ^
/usr/include/c++/14/bits/regex.h:1534:5: note: candidate: 'template<class _Bi_iter> bool std::__cxx11::operator==(const typename std::iterator_traits<_Iter>::value_type&, const sub_match<_BiIter>&)'
 1534 |     operator==(typename iterator_traits<_Bi_iter>::value_type const& __lhs,
      |     ^~~~~~~~
/usr/include/c++/14/bits/regex.h:1534:5: note:   template argument deduction/substitution failed:
a.cc:30:24: note:   'std::vector<int>::iterator' is not derived from 'const std::__cxx11::sub_match<_BiIter>'
   30 |         if (i == l.end()) {
      |                        ^
/usr/include/c++/14/bits/regex.h:1613:5: note: candidate: 'template<class _Bi_iter> bool std::__cxx11::operator==(const sub_match<_BiIter>&, const typename std::iterator_traits<_Iter>::value_type&)'
 1613 |     operator==(const sub_match<_Bi_iter>& __lhs,
      |     ^~~~~~~~
/usr/include/c++/14/bits/regex.h:1613:5: note:   template argument deduction/substitution failed:
a.cc:30:24: note:   mismatched types 'const std::__cxx11::sub_match<_BiIter>' and 'int'
   30 |         if (i == l.end()) {
      |                        ^
/usr/include/c++/14/bits/regex.h:2186:5: note: candidate: 'template<class _Bi_iter, class _Alloc> bool std::__cxx11::operator==(const match_results<_BiIter, _Alloc>&, const match_results<_BiIter, _Alloc>&)'
 2186 |     operator==(const match_results<_Bi_iter, _Alloc>& __m1,
      |     ^~~~~~~~
/usr/include/c++/14/bits/regex.h:2186:5: note:   template argument deduction/substitution failed:
a.cc:30:24: note:   mismatched types 'const std::__cxx11::match_results<_BiIter, _Alloc>' and 'int'
   30 |         if (i == l.end()) {
      |                        ^
In file included from /usr/include/c++/14/bits/stl_algobase.h:64,
                 from /usr/include/c++/14/algorithm:60,
                 from /usr/include/x86_64-linux-gnu/c++/14/bits/stdc++.h:51:
/usr/include/c++/14/bits/stl_pair.h:1033:5: note: candidate: 'template<class _T1, class _T2> constexpr bool std::operator==(const pair<_T1, _T2>&, const pair<_T1, _T2>&)'
 1033 |     operator==(const pair<_T1, _T2>& __x, const pair<_T1, _T2>& __y)
      |     ^~~~~~~~
/usr/include/c++/14/bits/stl_pair.h:1033:5: note:   template argument deduction/substitution failed:
a.cc:30:24: note:   mismatched types 'const std::pair<_T1, _T2>' and 'int'
   30 |         if (i == l.end()) {
      |                        ^
In file included from /usr/include/c++/14/bits/stl_algobase.h:67:
/usr/include/c++/14/bits/stl_iterator.h:441:5: note: candidate: 'template<class _Iterator> constexpr bool std::operator==(const reverse_iterator<_Iterator>&, const reverse_iterator<_Iterator>&)'
  441 |     operator==(const reverse_iterator<_Iterator>& __x,
      |     ^~~~~~~~
/usr/include/c++/14/bits/stl_iterator.h:441:5: note:   template argument deduction/substitution failed:
a.cc:30:24: note:   mismatched types 'const std::reverse_iterator<_Iterator>' and 'int'
   30 |         if (i == l.end()) {
      |                        ^
/usr/include/c++/14/bits/stl_iterator.h:486:5: note: candidate: 'template<class _IteratorL, class _IteratorR> constexpr bool std::operator==(const reverse_iterator<_Iterator>&, const reverse_iterator<_IteratorR>&)'
  486 |     operator==(const reverse_iterator<_IteratorL>& __x,
      |     ^~~~~~~~
/usr/include/c++/14/bits/stl_iterator.h:486:5: note:   template argument deduction/substitution failed:
a.cc:30:24: note:   mismatched types 'const std::reverse_iterator<_Iterator>' and 'int'
   30 |         if (i == l.end()) {
      |                        ^
/usr/include/c++/14/bits/stl_iterator.h:1667:5: note: candidate: 'template<class _IteratorL, class _IteratorR> constexpr bool std::operator==(const move_iterator<_IteratorL>&, const move_iterator<_IteratorR>&)'
 1667 |     operator==(const move_iterator<_IteratorL>& __x,
      |     ^~~~~~~~
/usr/include/c++/14/bits/stl_iterator.h:1667:5: note:   template argument deduction/substitution failed:
a.cc:30:24: note:   mismatched types 'const std::move_iterator<_IteratorL>' and 'int'
   30 |         if (i == l.end()) {
      |                        ^
/usr/include/c++/14/bits/stl_iterator.h:1737:5: note: candidate: 'template<class _Iterator> constexpr bool std::operator==(const move_iterator<_IteratorL>&, const move_iterator<_IteratorL>&)'
 1737 |     operator==(const move_iterator<_Iterator>& __x,
      |     ^~~~~~~~
/usr/include/c++/14/bits/stl_iterator.h:1737:5: note:   template argument deduction/substitution failed:
a.cc:30:24: note:   mismatched types 'const std::move_iterator<_IteratorL>' and 'int'
   30 |         if (i == l.end()) {
      |                        ^
In file included from /usr/include/c++/14/bits/char_traits.h:42,
                 from /usr/include/c++/14/string:42,
                 from /usr/include/c++/14/bitset:52,
                 from /usr/include/x86_64-linux-gnu/c++/14/bits/stdc++.h:52:
/usr/include/c++/14/bits/postypes.h:192:5: note: candidate: 'template<class _StateT> bool std::operator==(const fpos<_StateT>&, const fpos<_StateT>&)'
  192 |     operator==(const fpos<_StateT>& __lhs, const fpos<_StateT>& __rhs)
      |     ^~~~~~~~
/usr/include/c++/14/bits/postypes.h:192:5: note:   template argument deduction/substitution failed:
a.cc:30:24: note:   mismatched types 'const std::fpos<_StateT>' and 'int'
   30 |         if (i == l.end()) {
      |                        ^
In file included from /usr/include/c++/14/string:43:
/usr/include/c++/14/bits/allocator.h:235:5: note: candidate: 'template<class _T1, class _T2> bool std::operator==(const allocator<_CharT>&, const allocator<_T2>&)'
  235 |     operator==(const allocator<_T1>&, const allocator<_T2>&)
      |     ^~~~~~~~
/usr/include/c++/14/bits/allocator.h:235:5: note:   template argument deduction/substitution failed:
a.cc:30:24: note:   mismatched types 'const std::allocator<_CharT>' and 'int'
   30 |         if (i == l.end()) {
      |                        ^
In file included from /usr/include/c++/14/bits/basic_string.h:47,
                 from /usr/include/c++/14/string:54:
/usr/include/c++/14/string_view:629:5: note: candidate: 'template<class _CharT, class _Traits> constexpr bool std::operator==(basic_string_view<_CharT, _Traits>, __type_identity_t<basic_string_view<_CharT, _Traits> >)'
  629 |     operator==(basic_string_view<_CharT, _Traits> __x,
      |     ^~~~~~~~
/usr/include/c++/14/string_view:629:5: note:   template argument deduction/substitution failed:
a.cc:30:24: note:   mismatched types 'std::basic_string_view<_CharT, _Traits>' and 'int'
   30 |         if (i == l.end()) {
      |