AtCoder AGC004F Namori (图论)
題目鏈接
https://atcoder.jp/contests/agc004/tasks/agc004_f
題解
神仙題。。
首先考慮樹的情況,樹是二分圖,因此假設我們對二分圖進行黑白染色,那么操作就變成了,每次選擇兩個不同色的點來取反。然后再把黑色視作標記,那么問題就變成了,初始一些點上有標記,每次可以把標記沿著邊移動到一個沒標記的點,要把標記全部移動到和原來不同的位置上,求最小代價!
然后這個問題的做法就是,首先如果兩種顏色個數(shù)不同就無解,否則考慮一個下界,對于每一條邊而言,它至少要運送標記的次數(shù)等于其一端子樹內(nèi)黑白點個數(shù)差的絕對值。對所有的邊求和就是答案的下界,而我們也能構(gòu)造出來一種達到這個下界的方案,構(gòu)造詳見官方題解。
然后考慮基環(huán)樹。當環(huán)是奇環(huán)和偶環(huán)時,其作用不同,因此需要分類討論。
當環(huán)是偶環(huán)時,非樹邊的作用是多了一條運送標記的邊。假設這條邊運送了\(x\)個標記(可正可負),那么其所影響的是環(huán)上的點,需要最小化的是一個\(\sum |x-a_i|\)的形式,直接取中位數(shù)即可。
當環(huán)是奇環(huán)時,非樹邊的作用是可以給兩個端點的標記同時\(+1\)或\(-1\). 顯然\(+1\)和\(-1\)都出現(xiàn)是不優(yōu)的,由于操作可逆可以假設是\(+1\) (否則交換初始狀態(tài)和終止狀態(tài))。在這種情況下,若兩種顏色個數(shù)奇偶性不同就無解,否則執(zhí)行這種操作的次數(shù)\(x\)是確定的(因為初始和終止時兩種顏色點數(shù)確定)。那么就可以認為給這兩個端點分別加了\(x\)個標記,然后再執(zhí)行樹的算法即可。
時間復雜度\(O(N)\)或\(O(N\log N)\).
代碼
#include<bits/stdc++.h> #define llong long long using namespace std;const int N = 1e5; struct Edge {int nxt,v; } e[(N<<1)+3]; int fe[N+3]; int fa[N+3]; bool vis[N+3]; int dep[N+3]; int a[N+3]; int s[N+3]; vector<int> vec; int n,m,en,au,av,sum; llong ans;int absl(int x) {return x<0?-x:x;}void addedge(int u,int v) {en++; e[en].v = v;e[en].nxt = fe[u]; fe[u] = en; }void dfs(int u,int prv) {a[u] = dep[u]&1?-1:1; sum += a[u]; s[u] = a[u];vis[u] = true;for(int i=fe[u]; i; i=e[i].nxt){int v = e[i].v;if(v==prv) continue;if(vis[v]){au = u,av = v;}else{dep[v] = dep[u]+1;fa[v] = u;dfs(v,u);s[u] += s[v];}} }void dfs2(int u,int prv) {s[u] = a[u];vis[u] = true;for(int i=fe[u]; i; i=e[i].nxt){int v = e[i].v;if(v==prv) continue;if(vis[v]){au = u,av = v;}else{dep[v] = dep[u]+1;dfs2(v,u);s[u] += s[v];}} }int main() {scanf("%d%d",&n,&m);for(int i=1; i<=m; i++){int u,v; scanf("%d%d",&u,&v);addedge(u,v); addedge(v,u);}sum = 0; dep[1] = 0; dfs(1,0);if(m==n-1){if(sum) {puts("-1");}else{ans = 0ll;for(int i=1; i<=n; i++){ans += absl(s[i]);}printf("%lld\n",ans);}}else if(m==n){if(dep[au]>dep[av]) swap(au,av);if((dep[au]^dep[av])&1){if(sum) {puts("-1");}else{vec.clear();int v = av;while(dep[v]>dep[au]){vec.push_back(-s[v]);v = fa[v];}sort(vec.begin(),vec.end());int x = vec[vec.size()>>1];a[au] -= x; a[av] += x;for(int i=1; i<=n; i++) vis[i] = 0;dfs2(1,0);ans = absl(x);for(int i=1; i<=n; i++){ans += absl(s[i]);}printf("%lld\n",ans);}}else{if(absl(sum)&1) {puts("-1");}else{if(sum>0){for(int i=1; i<=n; i++) s[i] = -s[i],a[i] = -a[i];sum = -sum;}int x = (-sum)>>1;a[au] += x; a[av] += x;for(int i=1; i<=n; i++) vis[i] = 0;dfs2(1,0);ans = x;for(int i=1; i<=n; i++){ans += absl(s[i]);}printf("%lld\n",ans);}}}for(int i=1; i<=n; i++) fe[i] = vis[i] = fa[i] = 0;for(int i=1; i<=en; i++) e[i].v = e[i].nxt = 0;en = 0;return 0; }總結(jié)
以上是生活随笔為你收集整理的AtCoder AGC004F Namori (图论)的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: BZOJ 5267 特工 (类FWT)
- 下一篇: BZOJ 4823 Luogu P375