結果
問題 | No.119 旅行のツアーの問題 |
ユーザー | shobonvip |
提出日時 | 2024-04-02 08:25:58 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 3 ms / 5,000 ms |
コード長 | 3,320 bytes |
コンパイル時間 | 2,548 ms |
コンパイル使用メモリ | 227,684 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-09-30 23:04:29 |
合計ジャッジ時間 | 3,246 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,248 KB |
testcase_02 | AC | 2 ms
5,248 KB |
testcase_03 | AC | 1 ms
5,248 KB |
testcase_04 | AC | 2 ms
5,248 KB |
testcase_05 | AC | 1 ms
5,248 KB |
testcase_06 | AC | 2 ms
5,248 KB |
testcase_07 | AC | 2 ms
5,248 KB |
testcase_08 | AC | 2 ms
5,248 KB |
testcase_09 | AC | 2 ms
5,248 KB |
testcase_10 | AC | 1 ms
5,248 KB |
testcase_11 | AC | 1 ms
5,248 KB |
testcase_12 | AC | 1 ms
5,248 KB |
testcase_13 | AC | 2 ms
5,248 KB |
testcase_14 | AC | 2 ms
5,248 KB |
testcase_15 | AC | 1 ms
5,248 KB |
testcase_16 | AC | 2 ms
5,248 KB |
testcase_17 | AC | 2 ms
5,248 KB |
testcase_18 | AC | 2 ms
5,248 KB |
testcase_19 | AC | 2 ms
5,248 KB |
testcase_20 | AC | 2 ms
5,248 KB |
testcase_21 | AC | 3 ms
5,248 KB |
testcase_22 | AC | 2 ms
5,248 KB |
ソースコード
#include<bits/stdc++.h> using namespace std; #include<atcoder/maxflow> template <typename T> struct MoyasuUmeru { int n, s, t; T geta; vector<vector<T>> cost; atcoder::mf_graph<T> graph; MoyasuUmeru(int n): n(n), s(n), t(n+1), geta(0), cost(2), graph(0) { graph = atcoder::mf_graph<T>(n+2); cost[0].resize(n); cost[1].resize(n); } void add_cost(int i, int state, T val){ cost[state][i] += val; } void add_constraint(int x, int y, const vector<vector<T>> &cs){ T a = cs[0][0]; T b = cs[0][1]; T c = cs[1][0]; T d = cs[1][1]; geta += a; add_cost(x, 1, c - a); add_cost(y, 1, d - c); assert(b + c - a - d >= 0); graph.add_edge(x, y, b + c - a - d); } pair<T, vector<int>> solve(){ for (int i=0; i<n; i++){ if (cost[0][i] <= cost[1][i]){ graph.add_edge(s, i, cost[1][i] - cost[0][i]); geta += cost[0][i]; }else{ graph.add_edge(i, t, cost[0][i] - cost[1][i]); geta += cost[1][i]; } } T ret1 = geta + graph.flow(s, t); vector<bool> result = graph.min_cut(s); vector<int> ret2(n); for (int i=0; i<n; i++){ if (result[i]){ ret2[i] = 1; } } return pair(ret1, ret2); }; }; template<typename T> struct KvaluedMoyasuUmeru { int n; int alls; vector<int> k; MoyasuUmeru<T> mf; vector<int> offset; T geta; KvaluedMoyasuUmeru(const vector<int> &kk, T INF): k(kk), n((int)kk.size()), alls(0), geta(0), mf(0), offset(n) { for (int i=0; i<n; i++){ assert(k[i] >= 1); } for (int i=0; i<n-1; i++){ offset[i+1] = offset[i] + k[i] - 1; } alls = offset[n-1] + k[n-1] - 1; mf = MoyasuUmeru<T>(alls); for (int i=0; i<n; i++){ for (int j=0; j<k[i]-2; j++){ mf.graph.add_edge(offset[i] + j + 1, offset[i] + j, INF); } } } void add_1v(int i, const vector<T> &val){ assert((int)val.size() == k[i]); geta += val[k[i]-1]; for (int j=0; j<k[i]-1; j++){ mf.add_cost(offset[i] + j, 1, val[j] - val[j+1]); } } void add_2v(int x, int y, const vector<vector<T>> &val){ assert((int)val.size() == k[x]); assert((int)val[0].size() == k[y]); vector<T> add_to_x(k[x]); vector<T> add_to_y(k[y]); for (int i=0; i<k[x]; i++){ add_to_x[i] = val[i][0]; } for (int i=0; i<k[y]; i++){ add_to_y[i] = val[0][i]; } add_1v(x, add_to_x); add_1v(y, add_to_y); geta -= val[0][0]; for (int i=0; i<k[x]-1; i++){ for (int j=0; j<k[y]-1; j++){ mf.add_constraint( offset[x]+i, offset[y]+j, { {val[i+1][j+1] - val[i][j+1] - val[i+1][j] + val[i][j], 0}, {0, 0} } ); } } } pair<T, vector<int>> solve() { auto [old1, old2] = mf.solve(); T ret1 = old1 + geta; vector<int> ret2(n); for (int i=0; i<n; i++){ ret2[i] = k[i] - 1; for (int j=0; j<k[i]-1; j++){ if (old2[offset[i]+j] == 1){ ret2[i] = j; break; } } } return pair(ret1, ret2); } }; typedef long long ll; int main(){ int n; cin >> n; KvaluedMoyasuUmeru<ll> mf(vector<int>(n, 3), (ll)1e9); vector<ll> b(n), c(n); for (int i=0; i<n; i++){ cin >> b[i] >> c[i]; mf.add_1v(i, {-c[i], 0, -b[i]}); } int m; cin >> m; for (int i=0; i<m; i++){ int d, e; cin >> d >> e; mf.add_2v(d, e, { {0, 0, 0}, {0, 0, 0}, {(ll)1e9, 0, 0} } ); } auto [x1, x2] = mf.solve(); cout << - x1 << '\n'; }