题目
问题描述
有一个 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
提示
无
解题思路
分析
我们先来分析一下什么情况是合法的,什么情况是不合法的。
显然,
- 如果点数 n 为偶数,那么其生成树的边数 n−1 为奇数,不满足正好一半的条件,此时应输出 −1。
- 将所有可能加入的不构成环的颜色为 S 的边加入(用并查集维护信息并判断是否有环出现)。如果能加入的边数小于 (n−1)/2 ,输出−1。之后形成的就是一个森林。
- 将所有能加入而不构成环的颜色为 M 的边加入。如果此时不能形成一棵树,输出 -1。
- 将能加入而不使得颜色为 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;
}