結果
問題 |
No.3201 Corporate Synergy
|
ユーザー |
![]() |
提出日時 | 2025-07-11 22:20:00 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 2 ms / 2,000 ms |
コード長 | 4,738 bytes |
コンパイル時間 | 3,713 ms |
コンパイル使用メモリ | 290,716 KB |
実行使用メモリ | 7,844 KB |
最終ジャッジ日時 | 2025-07-11 22:20:04 |
合計ジャッジ時間 | 4,065 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 20 |
ソースコード
// #pragma GCC target("avx") // #pragma GCC optimize("O3") // #pragma GCC optimize("unroll-loops") #include <bits/stdc++.h> using namespace std; using std::cin; using std::cout; #define rep(i,n) for(int i = 0; i < (int)n; i++) #define FOR(n) for(int i = 0; i < (int)n; i++) #define repi(i,a,b) for(int i = (int)a; i < (int)b; i++) #define all(x) x.begin(),x.end() //#define mp make_pair #define vi vector<int> #define vvi vector<vi> #define vvvi vector<vvi> #define vvvvi vector<vvvi> #define pii pair<int,int> #define vpii vector<pair<int,int>> template<typename T> bool chmax(T &a, const T b) {if(a<b) {a=b; return true;} else {return false;}} template<typename T> bool chmin(T &a, const T b) {if(a>b) {a=b; return true;} else {return false;}} using ll = long long; using ld = long double; using ull = unsigned long long; const ll INF = numeric_limits<long long>::max() / 2; const ld pi = 3.1415926535897932384626433832795028; const ll mod = 998244353; int dx[] = {1, 0, -1, 0, -1, -1, 1, 1}; int dy[] = {0, 1, 0, -1, -1, 1, -1, 1}; #define int long long template<typename T> struct Edge { int to; T cap; int rev; Edge() = default; Edge(int to, T cap, int rev) : to(to), cap(cap), rev(rev) {} }; template<typename T> struct Dinic { const T e = numeric_limits<T>::max(); const T zero = 0; int n; vector<vector<Edge<T>>> g; vector<int> level, iter; vector<pair<int, int>> Edge_ID; Dinic(int n) : n(n), g(n) {} int add_edge(int from, int to, T cap) { Edge<T> f(to, cap, g[to].size()); Edge<T> t(from, zero, g[from].size()); int Now_Edge_ID = (int)Edge_ID.size(); Edge_ID.emplace_back(from, (int)g[from].size()); g[from].push_back(f); g[to].push_back(t); return Now_Edge_ID; } bool bfs(int s, int t) { level.assign(n, -1); level[s] = 0; queue<int> que; que.push(s); while(que.size()) { int v = que.front(); que.pop(); for(auto e : g[v]) { if(e.cap > zero && level[e.to] < 0) { level[e.to] = level[v] + 1; que.push(e.to); } } } return level[t] != -1; } T dfs(int v, int t, T f) { if(v == t) return f; for(int& i = iter[v]; i < g[v].size(); i++) { Edge<T>& e = g[v][i]; if(e.cap > zero && level[v] < level[e.to]) { T d = dfs(e.to, t, min(f, e.cap)); if(d > zero) { e.cap -= d; g[e.to][e.rev].cap += d; return d; } } } return zero; } T flow(int s, int t) { T f = zero; while(bfs(s, t)) { iter.assign(n, 0); T d; while((d = dfs(s, t, e)) > zero) f += d; } return f; } //sから見たときのmin_cutにおいてs側に所属するかどうかを全頂点に対して返すO(N) vector<bool> min_cut(int s) { vector<bool> visited(n, false); queue<int> que; visited[s] = true; que.push(s); while(!que.empty()) { int v = que.front(); que.pop(); for(auto e : g[v]) if(e.cap != 0 && !visited[e.to]) { visited[e.to] = true; que.push(e.to); } } return visited; } }; //bipartite matching //TODO: make faster Dinic using dynamic tree structure and current edge structure //燃やす埋めるref: https://qiita.com/ningenMe/items/69ed7ce43c9cd0a2de38 //add_edgeにintの返り値を持たせて、id = add_edge(), get_edge(id) = 元の辺 みたいにしたい。 //min_cut : s側ならばtrueのvector<bool>を返す。 //cut[from] = true && cut[to] = false --- from->toの辺が最小カットで切られる辺の一つ void solve() { int n; cin >> n; Dinic<int> dinic(n + 2); int src = n, dst = n + 1; vi p(n); int base = 0; FOR(n) { cin >> p[i]; if (p[i] > 0) base += p[i], dinic.add_edge(i, dst, p[i]); else dinic.add_edge(src, i, -p[i]); } int m; cin >> m; FOR(m) { int u, v; cin >> u >> v; u--; v--; dinic.add_edge(u, v, INF); } int k; cin >> k; FOR(k) { int a, b, s; cin >> a >> b >> s; a--; b--; base += s; dinic.add_edge(b, a, s); dinic.add_edge(a, dst, s); } cout << base - dinic.flow(src, dst) << endl; } signed main() { cin.tie(nullptr); ios::sync_with_stdio(false); solve(); return 0; }