【题解】洛谷 CF141E

---

题目

原题链接

问题描述

有一个 n 个点 m 条边的无向图，每条边可能有两种颜色 S 或 M ，请你判断是否存在并输出一种这样的方案，使得其生成树中边的颜色为 S 和 M 的正好各占一半。
如果无解，输出 −1 ，否则先输出这样的生成树的边的个数，再输出一种合法的情况的边的集合。

输入/输出例子

输入1

3 3
1 2 S
1 3 M
2 3 S

输出1

2
1 2

输入2

5 6
1 1 S
1 2 M
1 3 S
1 4 M
1 5 M
2 2 S

输出2

-1

提示

无

---

解题思路

分析

我们先来分析一下什么情况是合法的，什么情况是不合法的。
显然，

1. 如果点数 n 为偶数，那么其生成树的边数 n−1 为奇数，不满足正好一半的条件，此时应输出 −1。
2. 将所有可能加入的不构成环的颜色为 S 的边加入（用并查集维护信息并判断是否有环出现）。如果能加入的边数小于 (n−1)/2 ，输出−1。之后形成的就是一个森林。
3. 将所有能加入而不构成环的颜色为 M 的边加入。如果此时不能形成一棵树，输出 -1。
4. 将能加入而不使得颜色为 M 的边构成环的边加入，直至颜色为 M 的边的边数达到 (n−1)/2。此时每加入一条边对应的就是删除一条颜色为 S 的边。如果能加入的颜色为M 的边的数量不够，输出 −1 。

Code

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=1e6+1;
struct fy
{
	int fr,to,no;
}edgeS[N<<1],edgeM[N<<1];
int fa[N],n,m,cntS,cntM,sumS,sumM;
bool f[N],fM[N],fS[N];
int find(int x)
{
    while(x!=fa[x]) 
		x=fa[x]=fa[fa[x]];
    return x;
}
void init()
{
	for(int i=1;i<=n;i++)
        fa[i]=i;
}
void kr()
{
	n--;
    for(int i=1;i<=cntS;i++)
    {
        int fx=find(edgeS[i].fr),fy=find(edgeS[i].to);
        if(fx==fy)
            continue;
        fa[fx]=fy;
        sumS++;
    }
    for(int i=1;i<=cntM;i++)
    {
        int fx=find(edgeM[i].fr),fy=find(edgeM[i].to);
        if(fx==fy)
            continue;
        fa[fx]=fy;
        sumM++;
        fM[i]=1;
    }
    if(sumS<n/2||sumS+sumM<n-1)
    	puts("-1"),exit(0);	
    n++,init(),n--;
    for(int i=1;i<=cntM;i++)
    	if(fM[i])
    	{
	        int fx=find(edgeM[i].fr),fy=find(edgeM[i].to);
	        if(fx==fy)
	            continue;
	        fa[fx]=fy;
	        f[edgeM[i].no]=1;
		}
    for(int i=1;i<=cntM&&sumM<n/2;i++)
		if(!fM[i])
		{
			int fx=find(edgeM[i].fr),fy=find(edgeM[i].to);
	        if(fx==fy)
	            continue;
	        fa[fx]=fy;
	        sumM++;
	        f[edgeM[i].no]=1;
		}
	for(int i=1;i<=cntS;i++)
    {
        int fx=find(edgeS[i].fr),fy=find(edgeS[i].to);
        if(fx==fy)
            continue;
        fa[fx]=fy;
        f[edgeS[i].no]=1;
    }
    if(sumM<n/2)
    	puts("-1"),exit(0);
}

signed main()
{
    scanf("%lld%lld",&n,&m);
    init();
    for(int i=1;i<=m;i++)
    {
    	char xy;
    	int x,y;
    	scanf("%lld%lld",&x,&y);
    	cin>>xy;
    	if(xy=='S')
    		edgeS[++cntS].fr=x,edgeS[cntS].to=y,edgeS[cntS].no=i;
    	else
    		edgeM[++cntM].fr=x,edgeM[cntM].to=y,edgeM[cntM].no=i;
    }
    if(n%2==0)
    	puts("-1"),exit(0);
    kr();
    printf("%lld\n",n);
	for(int i=1;i<=m;i++)
		if(f[i])
			printf("%lld ",i);
    return 0;
}

更多方法

更多方法