Test/AtCoder/abc293_b.test.cpp

• View this file on GitHub
• Last update: 2023-08-08 12:55:07+09:00
• Problem: https://atcoder.jp/contests/abc293/tasks/abc293_b

Depends on

• std::arrayの入出力 (Src/Template/ArrayIO.hpp)
• 標準入力 (Src/Template/Input.hpp)
• 標準出力 (Src/Template/Output.hpp)
• std::pairの入出力 (Src/Template/PairIO.hpp)
• 標準データ型のエイリアス (Src/Template/TypeAlias.hpp)
• std::vectorの入出力 (Src/Template/VectorIO.hpp)

Code

#define PROBLEM "https://atcoder.jp/contests/abc293/tasks/abc293_b"

#include "../../Src/Template/TypeAlias.hpp"
#include "../../Src/Template/Input.hpp"
#include "../../Src/Template/Output.hpp"

using namespace zawa;

int main() {
    u32 N; input(N);
    std::vector<bool> A(N);
    for (u32 i{} ; i < N ; i++) {
        u32 a; input(a);
        if (not A[i]) A[a - 1] = true;
    }
    std::vector<u32> ans;
    for (u32 i{} ; i < N ; i++) {
        if (not A[i]) {
            ans.push_back(i + 1);
        }
    }
    out(ans.size());
    out(ans);
}

#line 1 "Test/AtCoder/abc293_b.test.cpp"
#define PROBLEM "https://atcoder.jp/contests/abc293/tasks/abc293_b"

#line 2 "Src/Template/TypeAlias.hpp"

#include <cstdint>
#include <cstddef>

namespace zawa {

using i16 = std::int16_t;
using i32 = std::int32_t;
using i64 = std::int64_t;
using i128 = __int128_t;

using u8 = std::uint8_t;
using u16 = std::uint16_t;
using u32 = std::uint32_t;
using u64 = std::uint64_t;

using usize = std::size_t;

} // namespace zawa
#line 2 "Src/Template/Input.hpp"

#line 2 "Src/Template/ArrayIO.hpp"

#line 4 "Src/Template/ArrayIO.hpp"

#include <iostream>
#include <array>

namespace zawa {

template <class T, usize N>
std::istream &operator>>(std::istream& is, std::array<T, N>& A) {
    for (T& a : A) {
        is >> a;
    }
    return is;
}

template <class T, usize N>
std::ostream &operator<<(std::ostream& os, const std::array<T, N>& A) {
    for (u32 i{} ; i < A.size() ; i++) {
        os << A[i] << (i + 1 == A.size() ? "" : " ");
    }
    return os;
}

} // namespace zawa
#line 2 "Src/Template/VectorIO.hpp"

#line 4 "Src/Template/VectorIO.hpp"

#line 6 "Src/Template/VectorIO.hpp"
#include <vector>

namespace zawa {

template <class T>
std::istream &operator>>(std::istream& is, std::vector<T>& A) {
    for (T& a : A) {
        is >> a;
    }
    return is;
}

template <class T>
std::ostream &operator<<(std::ostream& os, const std::vector<T>& A) {
    for (u32 i{} ; i < A.size() ; i++) {
        os << A[i] << (i + 1 == A.size() ? "" : " ");
    }
    return os;
}

} // namespace zawa
#line 2 "Src/Template/PairIO.hpp"

#line 4 "Src/Template/PairIO.hpp"

#line 6 "Src/Template/PairIO.hpp"
#include <utility>

namespace zawa {

template <class T1, class T2>
std::istream &operator>>(std::istream& is, std::pair<T1, T2>& P) {
    is >> P.first >> P.second;
    return is;
}

template <class T1, class T2>
std::ostream &operator<<(std::ostream& os, const std::pair<T1, T2>& P) {
    os << '(' << P.first << ',' << P.second << ')';
    return os;
}

} // namespace zawa
#line 6 "Src/Template/Input.hpp"

#line 8 "Src/Template/Input.hpp"

namespace zawa {

template <class T>
void input(T& value) {
    std::cin >> value;
}

template <class Head, class... Tail>
void input(Head& head, Tail&... tail) {
    input(head);
    input(tail...);
}

} // namespace zawa
#line 2 "Src/Template/Output.hpp"

#line 6 "Src/Template/Output.hpp"

#line 8 "Src/Template/Output.hpp"

namespace zawa {

void out() {
    std::cout << std::endl;
}

template <class T>
void out(const T& value) {
    std::cout << value << std::endl;
}

template <class Head, class... Tail>
void out(const Head& head, const Tail&... tail) {
    std::cout << head;
    if (sizeof...(tail)) {
        std::cout << ' ';
    }
    out(tail...);
}

void eout() {
    std::cerr << std::endl;
}

template <class T>
void eout(const T& value) {
    std::cerr << value << std::endl;
}

template <class Head, class... Tail>
void eout(const Head& head, const Tail&... tail) {
    std::cerr << head;
    if (sizeof...(tail)) {
        std::cerr << ' ';
    }
    eout(tail...);
}

} // namespace zawa
#line 6 "Test/AtCoder/abc293_b.test.cpp"

using namespace zawa;

int main() {
    u32 N; input(N);
    std::vector<bool> A(N);
    for (u32 i{} ; i < N ; i++) {
        u32 a; input(a);
        if (not A[i]) A[a - 1] = true;
    }
    std::vector<u32> ans;
    for (u32 i{} ; i < N ; i++) {
        if (not A[i]) {
            ans.push_back(i + 1);
        }
    }
    out(ans.size());
    out(ans);
}

Back to top page