칵테일

https://www.acmicpc.net/problem/1033

1033번: 칵테일

일단 비율들을 가지고 최소공배수를 구한다.

그리고 그 최소공배수를 ans[0]에 박고,

DFS로 순회를 하면서 비율을 곱하면서 업데이트 해준다.

 

#include <iostream>
#include <vector>
#include <tuple>
 
using namespace std;
 
vector<tuple<int, int, int>> A[10];
long lcm;
bool visited[10];
long ans[10];
 
long gcd(long a, long b)
{
    if (b == 0)
    {
        return a;
    }
    else
    {
        return gcd(b, a % b);
    }
}
 
void DFS(int node)
{
    visited[node] = true;
 
    for (tuple<int, int, int> i : A[node])
    {
        int next = get<0>(i);
        if (!visited[next])
        {
            // 주어진 비율로 다음 노드값 업데이트
            ans[next] = ans[node] * get<2>(i) / get<1>(i);
            DFS(next);
        }
    }
}
 
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    
    int n;
    cin >> n;
    lcm = 1;
    
    for (int i = 0; i < n - 1; i++)
    {
        int a, b, p, q;
        cin >> a >> b >> p >> q;
 
        //순서 주의!!
        A[a].push_back(make_tuple(b, p, q));
        A[b].push_back(make_tuple(a, q, p));
        
        // 두 수의 최소 공배수는 두 수의 곱을 최대공약수로 나눈 것
        lcm *= (p * q / gcd(p, q));
    }
 
    ans[0] = lcm;
    DFS(0);
 
    // 여러 개의 최대 공약수 구하기
    long mgcd = ans[0];
    for (int i = 1; i < n; i++)
    {
        mgcd = gcd(mgcd, ans[i]);
    }
 
    for (int i = 0; i < n; i++)
    {
        cout << ans[i] / mgcd << ' ';
    }
    
    return 0;
}