寝癖頭の解法

学習中の覚え書きを投稿、更新していきます。

AtCoder Problems in C++ #C - パーティー

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

#C - パーティー

atcoder.jp

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

/*
AtCoder Problems in C++
#C - パーティー
https://atcoder.jp/contests/joi2010yo/tasks/joi2010yo_c
提出コードの解答例
https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
int main(void){
  int n,m;
  cin>>n>>m;
  int cnt=0;
  vector<vector<int>> v(n+1);
  vector<int> a(m),b(m);
  for(int i=0;i<m;i++){
    cin>>a[i]>>b[i];
    v[a[i]].push_back(b[i]);
    v[b[i]].push_back(a[i]);
    if(a[i]==1 || b[i]==1){
      cnt++;
    }
  }
  set<int> F,FF;
  for(auto i : v[1]){
    F.insert(i);
  }
  for(auto i : v[1]){
    for(auto j : v[i]){
      if(j!=1 && F.count(j)!=1){
        FF.insert(j);
      }
    }
  }
  cout<<cnt+FF.size()<<endl;
  return 0;
}