結果
問題 | No.1364 [Renaming] Road to Cherry from Zelkova |
ユーザー | kaikey |
提出日時 | 2021-01-23 00:22:57 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 8,847 bytes |
コンパイル時間 | 3,281 ms |
コンパイル使用メモリ | 246,536 KB |
実行使用メモリ | 34,252 KB |
最終ジャッジ日時 | 2024-06-09 05:28:18 |
合計ジャッジ時間 | 11,167 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,376 KB |
testcase_02 | AC | 2 ms
5,376 KB |
testcase_03 | AC | 2 ms
5,376 KB |
testcase_04 | AC | 2 ms
5,376 KB |
testcase_05 | AC | 2 ms
5,376 KB |
testcase_06 | AC | 2 ms
5,376 KB |
testcase_07 | AC | 2 ms
5,376 KB |
testcase_08 | AC | 12 ms
5,504 KB |
testcase_09 | AC | 5 ms
5,376 KB |
testcase_10 | AC | 10 ms
5,376 KB |
testcase_11 | AC | 9 ms
5,376 KB |
testcase_12 | AC | 11 ms
5,376 KB |
testcase_13 | AC | 204 ms
25,268 KB |
testcase_14 | AC | 235 ms
25,924 KB |
testcase_15 | AC | 245 ms
27,752 KB |
testcase_16 | AC | 150 ms
20,324 KB |
testcase_17 | AC | 75 ms
17,808 KB |
testcase_18 | AC | 300 ms
34,216 KB |
testcase_19 | AC | 298 ms
34,252 KB |
testcase_20 | AC | 304 ms
34,228 KB |
testcase_21 | AC | 317 ms
34,220 KB |
testcase_22 | AC | 307 ms
34,228 KB |
testcase_23 | AC | 50 ms
11,812 KB |
testcase_24 | AC | 31 ms
7,644 KB |
testcase_25 | AC | 133 ms
17,688 KB |
testcase_26 | AC | 209 ms
23,896 KB |
testcase_27 | AC | 123 ms
18,196 KB |
testcase_28 | AC | 90 ms
14,064 KB |
testcase_29 | AC | 111 ms
17,668 KB |
testcase_30 | AC | 86 ms
14,212 KB |
testcase_31 | AC | 63 ms
12,844 KB |
testcase_32 | AC | 84 ms
15,984 KB |
testcase_33 | AC | 189 ms
21,928 KB |
testcase_34 | AC | 192 ms
23,032 KB |
testcase_35 | WA | - |
testcase_36 | AC | 187 ms
25,172 KB |
testcase_37 | AC | 75 ms
14,180 KB |
testcase_38 | AC | 99 ms
17,752 KB |
testcase_39 | AC | 97 ms
17,604 KB |
testcase_40 | AC | 96 ms
17,472 KB |
testcase_41 | AC | 103 ms
17,608 KB |
testcase_42 | AC | 101 ms
17,624 KB |
testcase_43 | AC | 73 ms
25,208 KB |
testcase_44 | AC | 58 ms
24,060 KB |
testcase_45 | AC | 69 ms
23,416 KB |
testcase_46 | AC | 12 ms
13,936 KB |
testcase_47 | AC | 2 ms
5,376 KB |
ソースコード
#include <bits/stdc++.h> #include <random> using namespace std; typedef unsigned long long _ulong; typedef long long int lint; typedef pair<lint, lint> plint; typedef pair<double long, double long> pld; #define ALL(x) (x).begin(), (x).end() #define SZ(x) ((lint)(x).size()) #define FOR(i, begin, end) for(lint i=(begin),i##_end_=(end);i<i##_end_;++i) #define IFOR(i, begin, end) for(lint i=(end)-1,i##_begin_=(begin);i>=i##_begin_;--i) #define REP(i, n) FOR(i,0,n) #define IREP(i, n) IFOR(i,0,n) #define endk '\n' #define fi first #define se second struct fast_ios { fast_ios() { cin.tie(nullptr), ios::sync_with_stdio(false), cout << fixed << setprecision(20); }; } fast_ios_; template<class T> auto add = [](T a, T b) -> T { return a + b; }; template<class T> auto f_max = [](T a, T b) -> T { return max(a, b); }; template<class T> auto f_min = [](T a, T b) -> T { return min(a, b); }; template<class T> using V = vector<T>; using Vl = V<lint>; using VVl = V<Vl>; template< typename T > ostream& operator<<(ostream& os, const vector< T >& v) { for (int i = 0; i < (int)v.size(); i++) os << v[i] << (i + 1 != v.size() ? " " : ""); return os; } template< typename T >istream& operator>>(istream& is, vector< T >& v) { for (T& in : v) is >> in; return is; } template<class T> bool chmax(T& a, const T& b) { if (a < b) { a = b; return 1; } return 0; } template<class T> bool chmin(T& a, const T& b) { if (b < a) { a = b; return 1; } return 0; } lint gcd(lint a, lint b) { if (b == 0) return a; else return gcd(b, a % b); } lint ceil(lint a, lint b) { return (a + b - 1) / b; } lint digit(lint a) { return (lint)log10(a); } lint e_dist(plint a, plint b) { return abs(a.fi - b.fi) * abs(a.fi - b.fi) + abs(a.se - b.se) * abs(a.se - b.se); } lint m_dist(plint a, plint b) { return abs(a.fi - b.fi) + abs(a.se - b.se); } void Worshall_Floyd(VVl& g) { REP(k, SZ(g)) REP(i, SZ(g)) REP(j, SZ(g)) chmin(g[i][j], g[i][k] + g[k][j]); } const lint MOD = 1e9 + 7, INF = 1e9 + 1; lint dx[8] = { 1, 0, -1, 0, 1, -1, 1, -1 }, dy[8] = { 0, 1, 0, -1, -1, -1, 1, 1 }; bool YN(bool flag) { cout << (flag ? "YES" : "NO") << endk; return flag; } bool yn(bool flag) { cout << (flag ? "Yes" : "No") << endk; return flag; } struct Edge { lint from, to; lint cost; Edge(lint u, lint v, lint c) { cost = c; from = u; to = v; } bool operator<(const Edge& e) const { return cost < e.cost; } }; struct WeightedEdge { lint to; lint cost; WeightedEdge(lint v, lint c = 1) { to = v; cost = c; } bool operator<(const WeightedEdge& e) const { return cost < e.cost; } }; using WeightedGraph = V<V<WeightedEdge>>; typedef pair<lint, plint> tlint; typedef pair<plint, plint> qlint; typedef pair<string, lint> valstring; template <std::int_fast64_t Modulus> class modint { using u64 = std::int_fast64_t; public: u64 a; constexpr modint(const u64 x = 0) noexcept : a(x% Modulus) {} constexpr u64& value() noexcept { return a; } constexpr const u64& value() const noexcept { return a; } constexpr modint operator+(const modint rhs) const noexcept { return modint(*this) += rhs; } constexpr modint operator-(const modint rhs) const noexcept { return modint(*this) -= rhs; } constexpr modint operator*(const modint rhs) const noexcept { return modint(*this) *= rhs; } constexpr modint operator/(const modint rhs) const noexcept { return modint(*this) /= rhs; } constexpr modint& operator+=(const modint rhs) noexcept { a += rhs.a; if (a >= Modulus) { a -= Modulus; } return *this; } constexpr modint& operator-=(const modint rhs) noexcept { if (a < rhs.a) { a += Modulus; } a -= rhs.a; return *this; } constexpr modint& operator*=(const modint rhs) noexcept { a = a * rhs.a % Modulus; return *this; } constexpr modint& operator/=(modint rhs) noexcept { u64 exp = Modulus - 2; while (exp) { if (exp % 2) { *this *= rhs; } rhs *= rhs; exp /= 2; } return *this; } }; typedef modint<MOD> ModInt; ModInt mod_pow(ModInt x, lint n) { ModInt ret = 1; while (n > 0) { if (n & 1) (ret *= x); (x *= x); n >>= 1; } return ret; } ModInt func[200000]; void funcinit(int N) { func[0] = 1; for (int i = 1; i <= N; i++) { func[i] = func[i - 1] * i; } } ModInt comb(ModInt n, ModInt r) { if (n.a <= 0 || n.a < r.a) { return 1; } return func[n.a] / (func[r.a] * func[(n - r).a]); } struct Topological_Sort { public: Topological_Sort(int _n) : G(_n, Vl()), indegree(_n, 0), N(_n) {} void add_edge(int u, int v) { G[u].push_back(v); indegree[v]++; } Vl get() { Vl sorted_vertices; Vl tmp_indegree = indegree; queue<int> que; REP(i, N) { if (tmp_indegree[i] == 0) que.push(i); } while (!que.empty()) { int v = que.front(); que.pop(); REP(i, SZ(G[v])) { int u = G[v][i]; tmp_indegree[u]--; if (tmp_indegree[u] == 0) que.push(u); } sorted_vertices.push_back(v); } return sorted_vertices; } private: VVl G; Vl indegree; int N; }; struct UnionFind { public: UnionFind() : _n(0) {} UnionFind(int n) : _n(n), parent_or_size(n, -1) {} int merge(int a, int b) { assert(0 <= a && a < _n); assert(0 <= b && b < _n); int x = leader(a), y = leader(b); if (x == y) return x; if (-parent_or_size[x] < -parent_or_size[y]) std::swap(x, y); if (used_count) { if (count_in_set[x].size() < count_in_set[y].size()) { std::swap(count_in_set[x], count_in_set[y]); } for (auto p : count_in_set[y]) { count_in_set[x][p.first] += p.second; } } if (set_operate) { root_values[x] = f(root_values[y], root_values[x]); } parent_or_size[x] += parent_or_size[y]; parent_or_size[y] = x; return x; } bool same(int a, int b) { assert(0 <= a && a < _n); assert(0 <= b && b < _n); return leader(a) == leader(b); } int leader(int a) { assert(0 <= a && a < _n); if (parent_or_size[a] < 0) return a; return parent_or_size[a] = leader(parent_or_size[a]); } int size(int a) { assert(0 <= a && a < _n); return -parent_or_size[leader(a)]; } std::vector<std::vector<int>> groups() { std::vector<int> leader_buf(_n), group_size(_n); for (int i = 0; i < _n; i++) { leader_buf[i] = leader(i); group_size[leader_buf[i]]++; } std::vector<std::vector<int>> result(_n); for (int i = 0; i < _n; i++) { result[i].reserve(group_size[i]); } for (int i = 0; i < _n; i++) { result[leader_buf[i]].push_back(i); } result.erase( std::remove_if(result.begin(), result.end(), [&](const std::vector<int>& v) { return v.empty(); }), result.end()); return result; } //update root calc //set by set operations void set_operate_and_value(std::vector<lint> array, function<lint(lint, lint)> _f) { f = _f; root_values = array; set_operate = true; } lint get_set_value(int a) { return root_values[leader(a)]; } //regist count void regist_count(int a, int label) { if (!used_count) { used_count = true; count_in_set.assign(_n, std::map<int, int>()); } count_in_set[leader(a)][label]++; } int get_count(int a, int label) { if (!used_count) return -1; return count_in_set[leader(a)][label]; } private: int _n; std::vector<int> parent_or_size; std::vector<std::map<int, int>> count_in_set; bool used_count = false; std::vector<lint> root_values; function<lint(lint, lint)> f; bool set_operate = false; }; lint N, M, u, v, l, a; int main() { cin >> N >> M; V<V<tlint>> rev(N + 1, V<tlint>()); VVl to(N + 1, Vl()); Topological_Sort topo(N + 1); V<plint> edges; UnionFind tree(N + 1); REP(i, M) { cin >> u >> v >> l >> a; rev[v].push_back({ u, {l, a} }); edges.push_back({ u, v }); tree.merge(u, v); to[u].push_back(v); to[v].push_back(u); } queue<lint> que; deque<bool> visited(N + 1, false), _visited(N + 1, false); que.push(0); visited[0] = true; while (!que.empty()) { lint curr = que.front(); que.pop(); for (lint nxt : to[curr]) { if (visited[nxt]) continue; visited[nxt] = true; que.push(nxt); } } que.push(N); _visited[N] = true; while (!que.empty()) { lint curr = que.front(); que.pop(); for (auto v : rev[curr]) { lint nxt = v.first; if (_visited[nxt]) continue; _visited[nxt] = true; que.push(nxt); } } REP(i, M) { if (visited[edges[i].first] && _visited[edges[i].second]) { topo.add_edge(edges[i].first, edges[i].second); } } auto vec = topo.get(); if (!tree.same(0, N) || !visited[N]) { cout << 0 << endk; }else if (SZ(vec) != N + 1) { cout << "INF" << endk; } else { reverse(ALL(vec)); V<ModInt> dp(N + 1); V<ModInt> cnt(N + 1, 0); cnt[N] = 1; for (lint v : vec) { for (auto nxt : rev[v]) { dp[nxt.first] += dp[v] * nxt.second.second + (cnt[v] * nxt.second.second) * nxt.second.first; cnt[nxt.first] += cnt[v] * nxt.second.second; } } ModInt ans = 0; cout << dp[0].a << endk; } }