寝癖頭の解法

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

AtCoder Problems in C++ #flu - インフルエンザ (Flu)

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

#flu - インフルエンザ (Flu)

https://www.ioi-jp.org/camp/2008/2008-sp-tasks/2008-sp_tr-day1_20.pdf

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

/*
AtCoder Problems in C++
#flu - インフルエンザ (Flu)
https://www.ioi-jp.org/camp/2008/2008-sp-tasks/2008-sp_tr-day1_20.pdf
提出コードの解答例
https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
using ll=long long;
pair<ll,ll> p[100001];
ll c[1001][1001],f[100001];
int main(void){
  ll n,m,d,k;
  cin>>n>>m>>d>>k;
  for(ll i=1;i<=n;i++){
    ll x,y;
    cin>>x>>y;
    p[i]={x,y};
    c[x][y]=i;
  }
  stack<ll> s,t;
  s.push(1);
  f[1]=m;
  for(ll i=1;i<=k;i++){
    while(!(s.empty())){
      ll v=s.top();
      s.pop();
      auto[x,y]=p[v];
      for(ll j=x-d;j<=x+d;j++){
        for(ll l=y-d;l<=y+d;l++){
          if(j<0 || j>=1000 || l<0 || l>=1000){
            continue;
          }
          if((j-x)*(j-x)+(l-y)*(l-y)>d*d){
            continue;
          }
          if(!c[j][l]){
            continue;
          }
          if(f[c[j][l]]){
            continue;
          }
          t.push(c[j][l]);
          f[c[j][l]]=i+m;
        }
      }
    }
    swap(s,t);
  }
  ll ans=0;
  for(ll i=1;i<=n;i++){
    if(f[i] && f[i]>k){
      ans++;
    }
  }
  cout<<ans<<endl;
  return 0;
}