寝癖頭の解法

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

AtCoder Problems in C++ #D - 釘 (Nails)

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

#D - 釘 (Nails)

https://www.ioi-jp.org/joi/2011/2012-ho-prob_and_sol/2012-ho.pdf#page=10

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

/*
AtCoder Problems in C++
#D - 釘 (Nails)
https://www.ioi-jp.org/joi/2011/2012-ho-prob_and_sol/2012-ho.pdf#page=10
提出コードの解答例
https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
using ll=long long;
array<array<ll,5001>,5001> sum={0};
int main(void){
  ll n,m;
  cin>>n>>m;
  for(ll i=0;i<m;i++){
    ll a,b,x;
    cin>>a>>b>>x;
    sum.at(a).at(b)=max(sum.at(a).at(b),x+1);
  }
  ll ans=0;
  for(ll i=1;i<=n;i++){
    for(ll j=1;j<=i;j++){
      if(sum.at(i).at(j)>0){
        ans+=1;
      }
      if(i<n){
        sum.at(i+1).at(j)=max(sum.at(i+1).at(j),sum.at(i).at(j)-1);
        sum.at(i+1).at(j+1)=max(sum.at(i+1).at(j+1),sum.at(i).at(j)-1);
      }
    }
  }
  cout<<ans<<endl;
  return 0;
}