寝癖頭の解法

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

AtCoder Problems in C++ #E - 最軽量のモビール

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

#E - 最軽量のモビール

https://www.ioi-jp.org/joi/2006/2007-ho-prob_and_sol/2007-ho.pdf#page=12

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

/*
AtCoder Problems in C++
#E - 最軽量のモビール
https://www.ioi-jp.org/joi/2006/2007-ho-prob_and_sol/2007-ho.pdf#page=12
提出コードの解答例
https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
tuple<int,int,int,int> t[101];
vector<int> c(101);
int main(void){
  int n;
  cin>>n;
  for(int i=0;i<n;i++){
    int p,q,r,b;
    cin>>p>>q>>r>>b;
    int g=gcd(p,q);
    p/=g;
    q/=g;
    r--;
    b--;
    t[i]={p,q,r,b};
  }
  auto f=[&](auto f,int i){
    if(i==-1){
      return 1;
    }
    if(c[i]){
      return c[i];
    }
    auto[p,q,r,b]=t[i];
    long long x=f(f,r),y=f(f,b);
    return c[i]=x*y/gcd(x*p,y*q)*(p+q);
  };
  int ans=0;
  for(int i=0;i<n;i++){
    ans=max(ans,f(f,i));
  }
  cout<<ans<<endl;
  return 0;
}