寝癖頭の解法

小学生の目線から、勉強中の覚え書きを投稿、更新していきます。

AtCoder Problems in C++ #D - Play Train

AtCoder Beginner Contestの過去問から、その提出コードの解答例です。
AtCoderとは、コンテストを通じて、プログラミングやアルゴリズムを学習するサービスです。
atcoder.jp
プログラミングコンテストとは、「与えられた問題をいかに素早く、正確に」解くことができるかを競うものです。
競技プログラミング」を略して、「競プロ」などと呼ばれています。

#D - Play Train

atcoder.jp

僕が作成、提出したコードは、以下のとおりです。

/*
AtCoder Problems in C++
#D - Play Train
https://atcoder.jp/contests/abc225/tasks/abc225_d
提出コードの解答例
https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
using ll=long long;
int main(void){
  ll n,q;
  cin>>n>>q;
  vector<ll> next(n,-1),prev(n,-1);
  while(q--){
    ll t;
    cin>>t;
    if(t==1){
      ll x,y;
      cin>>x>>y;
      x--;y--;
      next[x]=y;
      prev[y]=x;
    }else if(t==2){
      ll x,y;
      cin>>x>>y;
      x--;y--;
      prev[y]=-1;
      next[x]=-1;
    }else{
      ll x;
      cin>>x;
      x--;
      while(prev[x]!=-1){
        x=prev[x];
      }
      vector<ll> v;
      v.push_back(x);
      while(next[x]!=-1){
        x=next[x];
        v.push_back(x);
      }
      ll vs=(ll)v.size();
      cout<<vs<<" ";
      for(ll i=0;i<vs;i++){
        cout<<v[i]+1<<((i==vs-1) ? "\n" : " ");
      }
    }
  }
  return 0;
}

AtCoder Beginner Contestは、オンラインジャッジによるプログラミングコンテストです。
日本語と英語に対応していて、週末ごとに実施されているみたいです。
https://practice.contest.atcoder.jp/tutorial
アカウントを登録すれば、誰でも参加できます。