AtCoder AGC004F Namori (图论)

题目链接

https://atcoder.jp/contests/agc004/tasks/agc004_f

题解

神仙题。。

首先考虑树的情况，树是二分图，因此假设我们对二分图进行黑白染色，那么操作就变成了，每次选择两个不同色的点来取反。然后再把黑色视作标记，那么问题就变成了，初始一些点上有标记，每次可以把标记沿着边移动到一个没标记的点，要把标记全部移动到和原来不同的位置上，求最小代价！

然后这个问题的做法就是，首先如果两种颜色个数不同就无解，否则考虑一个下界，对于每一条边而言，它至少要运送标记的次数等于其一端子树内黑白点个数差的绝对值。对所有的边求和就是答案的下界，而我们也能构造出来一种达到这个下界的方案，构造详见官方题解。

然后考虑基环树。当环是奇环和偶环时，其作用不同，因此需要分类讨论。

当环是偶环时，非树边的作用是多了一条运送标记的边。假设这条边运送了\(x\)个标记（可正可负），那么其所影响的是环上的点，需要最小化的是一个\(\sum |x-a_i|\)的形式，直接取中位数即可。

当环是奇环时，非树边的作用是可以给两个端点的标记同时\(+1\)或\(-1\). 显然\(+1\)和\(-1\)都出现是不优的，由于操作可逆可以假设是\(+1\) (否则交换初始状态和终止状态)。在这种情况下，若两种颜色个数奇偶性不同就无解，否则执行这种操作的次数\(x\)是确定的（因为初始和终止时两种颜色点数确定）。那么就可以认为给这两个端点分别加了\(x\)个标记，然后再执行树的算法即可。

时间复杂度\(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;
}