結果

問題 No.3250 最小公倍数
ユーザー ooaiu
提出日時 2025-08-29 22:29:49
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 5,249 bytes
コンパイル時間 4,608 ms
コンパイル使用メモリ 319,816 KB
実行使用メモリ 417,392 KB
最終ジャッジ日時 2025-08-29 22:30:18
合計ジャッジ時間 27,963 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 18 TLE * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include <bits/stdc++.h>
using namespace std;
template <class F> class y_combinator {
    F f;

   public:
    y_combinator(F&& f) : f(std::forward<F>(f)) {}
    template <class... Args> auto operator()(Args&&... args) const { return f(*this, std::forward<Args>(args)...); }
};
using ll = long long;
using ld = long double;
using u8 = uint8_t;
using u32 = uint32_t;
using u64 = uint64_t;
using vi = vector<int>;
using vl = vector<ll>;
using pii = pair<int, int>;
using pll = pair<ll, ll>;
template <class T, class U = std::less<T>> using prique = std::priority_queue<T, std::vector<T>, U>;
inline constexpr int popcnt(u64 x) noexcept { return __builtin_popcountll(x); }
template <class T> T floor(T a, T b) noexcept { return a / b - (a % b && (a ^ b) < 0); }
template <class T> T ceil(T a, T b) noexcept { return floor(a + b - 1, b); }
template <class T> bool chmin(T& x, const T& y) noexcept { return (x > y ? x = y, true : false); }
template <class T> bool chmax(T& x, const T& y) noexcept { return (x < y ? x = y, true : false); }
template <class T> void UNIQUE(std::vector<T>& v) { std::sort(std::begin(v), std::end(v)), v.erase(std::unique(std::begin(v), std::end(v)), std::end(v)); }
void SCAN() {}
template <class H, class... T> void SCAN(H& h, T&... t) { std::cin >> h, SCAN(t...); }
#define INT(...)     \
    int __VA_ARGS__; \
    SCAN(__VA_ARGS__)
#define LL(...)            \
    long long __VA_ARGS__; \
    SCAN(__VA_ARGS__)
#define LD(...)              \
    long double __VA_ARGS__; \
    SCAN(__VA_ARGS__)
#define STR(...)             \
    std::string __VA_ARGS__; \
    SCAN(__VA_ARGS__)
#define CHAR(...)     \
    char __VA_ARGS__; \
    SCAN(__VA_ARGS__)
#define VEC(type, name, size)     \
    std::vector<type> name(size); \
    for (int i = 0; i < (int)size; i++) SCAN(name[i]);
#define VV(type, name, h, w)                                      \
    std::vector<std::vector<type>> name(h, std::vector<type>(w)); \
    for (int i = 0; i < (int)h; i++)                              \
        for (int j = 0; j < (int)w; j++) SCAN(name[i][j]);
#define overload4(a, b, c, d, e, ...) e
#define rep1(a) for (long long _i = 0; _i < (a); _i++)
#define rep2(i, a) for (long long i = 0; i < (a); i++)
#define rep3(i, a, b) for (long long i = (a); i < (b); i++)
#define rep4(i, a, b, c) for (long long i = (a); i < (b); i += (c))
#define rep(...) overload4(__VA_ARGS__, rep4, rep3, rep2, rep1)(__VA_ARGS__)
#define rrep(i, a, b, c) for (long long i = (a); i > (b); i += (c))
#define all(x) std::begin(x), std::end(x)
#define rall(x) std::rbegin(x), std::rend(x)
#define len(x) (long long)(size(x))
#define pb push_back
#ifndef LOCAL
#define debug(...)
#endif

template <int W = 1 << 20>
class Osa_k {
    std::array<int, W> lpf;

   public:
    constexpr Osa_k() {
        for (int x = 2; x < W; x++) {
            if (lpf[x] != 0) continue;
            lpf[x] = x;
            if (x >= (W + x - 1) / x) continue;
            for (int y = x * x; y < W; y += x) {
                if (lpf[y] == 0) lpf[y] = x;
            }
        }
    }

    std::vector<std::pair<int, int>> factor(int x) const {
        assert(1 <= x && x < W);
        std::vector<std::pair<int, int>> res;
        while (x > 1) {
            if (res.empty() || res.back().first != lpf[x]) {
                res.push_back(std::make_pair(lpf[x], 1));
            } else {
                res.back().second += 1;
            }
            x /= lpf[x];
        }
        return res;
    }

    bool is_prime(int x) const {
        assert(1 <= x && x < W);
        if (x == 1) return 0;
        return lpf[x] == x;
    }
};

#include <atcoder/modint>
using mint = atcoder::modint998244353;
Osa_k<1 << 20> F;
const int maxn = 5 << 17;
vector<int> g[5 << 17];
vector<pii> col[5 << 17];
int sz[5 << 17];
unordered_map<int, int> cnt[maxn];
mint ANS[5 << 17];
void dfs(int v, int p) {
    int mx = -1, bigChild = -1;
    for (auto u : g[v]) {
        if (u != p) {
            dfs(u, v);
            if (sz[u] > mx) {
                mx = sz[u];
                bigChild = u;
            }
        }
    }
    if (bigChild != -1) {
        cnt[v] = cnt[bigChild];
    } else {
        cnt[v].clear();
    }
    for (auto [p, e] : col[v]) chmax(cnt[v][p], e);
    for (auto u : g[v]) {
        if (u != p && u != bigChild) {
            for (auto x : cnt[u])
                chmax(cnt[v][x.first], x.second);
        }
    }
    mint t = 1;
    for (auto [p, e] : cnt[v]) {
        t *= mint(p).pow(e);
    }
    ANS[v] = t;
}
void run_case() {
    INT(N);
    VEC(int, A, N);
    rep(i, N) {
        col[i] = F.factor(A[i]);
    }
    rep(i, N - 1) {
        INT(a, b);
        a--, b--;
        g[a].pb(b);
        g[b].pb(a);
    }
    y_combinator([&](auto self, int v, int p) -> void {
        sz[v] = 1;
        for (int u : g[v]) {
            if (u != p) {
                self(u, v);
                sz[v] += sz[u];
            }
        }
    })(0, -1);
    dfs(0, -1);
    rep(i, N) cout << ANS[i].val() << "\n";
}
int main() {
    std::ios_base::sync_with_stdio(false);
    std::cin.tie(nullptr);
    std::fixed(std::cout).precision(16);
    int T = 1;
    while (T--) run_case();
    return 0;
}
0