結果
問題 |
No.3201 Corporate Synergy
|
ユーザー |
![]() |
提出日時 | 2025-07-11 21:55:02 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 2 ms / 2,000 ms |
コード長 | 7,792 bytes |
コンパイル時間 | 2,802 ms |
コンパイル使用メモリ | 215,428 KB |
実行使用メモリ | 6,272 KB |
最終ジャッジ日時 | 2025-07-11 21:55:07 |
合計ジャッジ時間 | 3,354 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 20 |
ソースコード
//#pragma GCC target("avx2") //#pragma GCC optimize("O3") //#pragma GCC optimize("unroll-loops") #include <bits/stdc++.h> using namespace std; using ll = long long; using pii = pair<int,int>; using pll = pair<ll,ll>; using pli = pair<ll,int>; #define AMARI 998244353 //#define AMARI 1000000007 #define el '\n' #define El '\n' #define YESNO(x) ((x) ? "Yes" : "No") #define YES YESNO(true) #define NO YESNO(false) #define REV_PRIORITY_QUEUE(tp) priority_queue<tp,vector<tp>,greater<tp>> #define EXIT_ANS(x) {cout << (x) << '\n'; return;} template <typename T> void inline SORT(vector<T> &v){sort(v.begin(),v.end()); return;} template <typename T> void inline VEC_UNIQ(vector<T> &v){sort(v.begin(),v.end()); v.erase(unique(v.begin(),v.end()),v.end()); return;} template <typename T> T inline MAX(vector<T> &v){return *max_element(v.begin(),v.end());} template <typename T> T inline MIN(vector<T> &v){return *min_element(v.begin(),v.end());} template <typename T> T inline SUM(vector<T> &v){T ans = 0; for(int i = 0; i < (int)v.size(); i++)ans += v[i]; return ans;} template <typename T> void inline DEC(vector<T> &v){for(int i = 0; i < (int)v.size(); i++)v[i]--; return;} template <typename T> void inline INC(vector<T> &v){for(int i = 0; i < (int)v.size(); i++)v[i]++; return;} void inline TEST(void){cerr << "TEST" << endl; return;} template <typename T> bool inline chmin(T &x,T y){ if(x > y){ x = y; return true; } return false; } template <typename T> bool inline chmax(T &x,T y){ if(x < y){ x = y; return true; } return false; } template <typename T = long long> vector<T> inline get_vec(int n){ vector<T> ans(n); for(int i = 0; i < n; i++)cin >> ans[i]; return ans; } template <typename T> void inline print_vec(vector<T> &vec,bool kaigyou = false){ int n = (int)vec.size(); for(int i = 0; i < n; i++){ cout << vec[i]; if(kaigyou || i == n - 1)cout << '\n'; else cout << ' '; } if(!n)cout << '\n'; return; } template <typename T> void inline debug_vec(vector<T> &vec,bool kaigyou = false){ int n = (int)vec.size(); for(int i = 0; i < n; i++){ cerr << vec[i]; if(kaigyou || i == n - 1)cerr << '\n'; else cerr << ' '; } if(!n)cerr << '\n'; return; } vector<vector<int>> inline get_graph(int n,int m = -1,bool direct = false){ if(m == -1)m = n - 1; vector<vector<int>> g(n); while(m--){ int u,v; cin >> u >> v; u--; v--; g[u].push_back(v); if(!direct)g[v].push_back(u); } return g; } vector<int> make_inv(vector<int> p){ int n = (int)p.size(); vector<int> ans(n); for(int i = 0; i < n; i++)ans[p[i]] = i; return ans; } // フロー周りを扱う色々(になる予定) template <typename T> class ococo_flow { private: class Edge { public: int to; T cap; int gyakuhen; // 逆辺のindex Edge(int xxx, T yyy, int zzz) { to = xxx; cap = yyy; gyakuhen = zzz; } }; int n; vector<vector<Edge>> g; vector<int> dist; vector<int> visited; T inft; // BFS O(|V|) void dinic_bfs(int s) { queue<int> que; que.push(s); dist[s] = 0; while(!que.empty()) { int temp = que.front(); que.pop(); for(int i = 0; i < g[temp].size(); i++) { if(g[temp][i].cap > 0 && dist[g[temp][i].to] < 0) { dist[g[temp][i].to] = dist[temp] + 1; que.push(g[temp][i].to); } } } } // 再帰DFS 増加パスを取得する 最悪O(|V||E|) T dinic_dfs(int point, int goal, T f) { if(point == goal) { return f; } for(int i = visited[point]; i < g[point].size(); i++) { visited[point] = i; if(g[point][i].cap > 0 && dist[point] < dist[g[point][i].to]) { T temp = dinic_dfs(g[point][i].to, goal, min(f, g[point][i].cap)); if(temp > 0) { g[point][i].cap -= temp; g[g[point][i].to][g[point][i].gyakuhen].cap += temp; return temp; } } } return 0; } // デバッグ用 void print_flow(void) { for(int i = 0; i < n; i++) { cerr << "i = " << i; cerr << " dist[i] = " << dist[i] << el; for(int j = 0; j < g[i].size(); j++) { cerr << "to = " << g[i][j].to; cerr << " cap = " << g[i][j].cap; cerr << el; } cerr << el; } cerr << el; return; } public: ococo_flow(int N = 0, T Tinf = INT_MAX) { inft = Tinf; n = N; g.resize(n); dist.resize(n); visited.resize(n); } // s→tの大きさflの辺を追加する void einsert(int s, int t, T fl) { assert(s < n && t < n); g[s].push_back(Edge(t, fl, g[t].size())); // 逆辺も張る g[t].push_back(Edge(s, 0, g[s].size() - 1)); } // 点sから点tへの最大流をdinicで求める 最悪O(|V|**2 |E|)だが実際にそこまでいくことは少ない // BFSで距離が増加する向きの変のみで構成されたグラフを取得する // 増加パスをDFSで求めてフローを流す // ↑このループを増加パスがなくなるまでやる T dinic_max_flow(int s, int t) { T ans = 0; while(1) { for(int i = 0; i < n; i++) { dist[i] = -1; } dinic_bfs(s); if(dist[t] < 0) return ans; T temp = 0; for(int i = 0; i < n; i++) visited[i] = 0; while(1) { temp = dinic_dfs(s, t, inft); ans += temp; if(temp == 0) break; } } return 0; } }; #define MULTI_TEST_CASE false void solve(void){ //問題を見たらまず「この問題設定から言えること」をいっぱい言う //よりシンプルな問題に言い換えられたら、言い換えた先の問題を自然言語ではっきりと書く //複数の解法のアイデアを思いついた時は全部メモしておく //g++ -D_GLIBCXX_DEBUG -Wall -O2 g.cpp -o o int n; cin >> n; vector<ll> p = get_vec(n); int m; cin >> m; vector<int> u(m),v(m); for(int i = 0; i < m; i++){ cin >> u[i] >> v[i]; } DEC(u); DEC(v); int k; cin >> k; vector<int> a(k),b(k); vector<ll> s(k); for(int i = 0; i < k; i++){ cin >> a[i] >> b[i] >> s[i]; } DEC(a); DEC(b); ococo_flow<ll> fl(n + k + 2); ll ans = 0; for(int i = 0; i < n; i++){ if(p[i] < 0){ fl.einsert(n + k,i,-1LL * p[i]); fl.einsert(i,n + k + 1,0LL); } if(p[i] >= 0){ fl.einsert(n + k,i,0LL); fl.einsert(i,n + k + 1,p[i]); ans += p[i]; } } for(int i = 0; i < m; i++){ fl.einsert(u[i],v[i],LLONG_MAX / 1000LL); } for(int i = 0; i < k; i++){ fl.einsert(n + k,i + n,0LL); fl.einsert(i + n,n + k + 1,s[i]); ans += s[i]; fl.einsert(a[i],i + n,LLONG_MAX / 1000LL); fl.einsert(b[i],i + n,LLONG_MAX / 1000LL); } ans -= fl.dinic_max_flow(n + k,n + k + 1); cout << ans << el; return; } void calc(void){ return; } signed main(void){ cin.tie(nullptr); ios::sync_with_stdio(false); calc(); int t = 1; if(MULTI_TEST_CASE)cin >> t; while(t--){ solve(); } return 0; }