Submission #50415768

---

Source Code Expand

#include <bits/stdc++.h>
#include <atcoder/modint>

namespace {
    using ModInt [[maybe_unused]] = atcoder::modint998244353;
    using Num [[maybe_unused]] = long long int;
    using Vec [[maybe_unused]] = std::vector<Num>;
    using Set [[maybe_unused]] = std::set<Num>;
    using Mset [[maybe_unused]] = std::multiset<Num>;
    using Edges [[maybe_unused]] = std::vector<std::vector<Num>>;

    template<typename T>
    using Q [[maybe_unused]] = std::queue<T>;

    template<typename T>
    using PQ [[maybe_unused]] = std::priority_queue<T, std::vector<T>, std::greater<T>>;
}

void solve(std::istream& is, std::ostream& os) {
    Num n, m;
    is >> n >> m;

    Edges edges_fwd(n);
    for(Num i{0}; i<m; ++i) {
        Num u, v;
        is >> u >> v;
        --u;
        --v;
        edges_fwd[u].push_back(v);
        edges_fwd[v].push_back(u);
    }

    Vec ws(n);
    for(auto&& w : ws) {
        is >> w;
    }

    Vec aset(n);
    for(auto&& a : aset) {
        is >> a;
    }

    std::vector<std::pair<Num, Num>> ordered;
    for(Num from{0}; from<n; ++from) {
        ordered.push_back(std::make_pair(ws.at(from), from));
    }
    std::ranges::sort(ordered);

    Num total {0};
    Vec degs(n, 0);
    for(const auto& [nw, from] : ordered) {
        const auto from_w = ws.at(from);
        Vec dp(from_w, -1000000000000000000LL);
        dp.at(0) = 0;

        for(const auto& to : edges_fwd[from]) {
            const auto to_w = ws.at(to);
            auto next = dp;

            for(Num i{0}; i<from_w; ++i) {
                Num wsum = i + to_w;
                if (wsum >= from_w) {
                    break;
                }
                next.at(wsum) = std::max(next.at(wsum), dp.at(i) + degs.at(to));
            }

            std::swap(dp, next);
        }

        const auto deg = 1 + (*std::ranges::max_element(dp));
        total += aset.at(from) * deg;
        degs.at(from) = deg;
    }

    os << total << "\n";
}

int main(void) {
    solve(std::cin, std::cout);
    return 0;
}

Submission Info

Judge Result