大概题意:
根据要求在飞机场建服务区,如果可以满足要求的话输出最少的数量,否则输出impossible。
第一行输入n,m,代表有n个飞机场和m个要求。然后m行,每行三个数,分别是i,j,k。代表第 i 个机场和第 j 个机场一共有k个服务区,而且每个机场最多只能有一个服务区。
思路:
如果两个服务区,那么就给两个机场都染色成color1,如果是零的话,就都染成color2。如果起了冲突,那么直接输出impossible。
然后只需要去判断两个机场只有一个服务区的情况。我们将只有一个服务区的情况建无向图。然后将这个图分成两种情况,一种是图中已经有一些点被染色了,另一种是整个图上没有被染过一次色。
所以,我们从被染了色的点开始遍历它能所到达的所有点,组成一种图。剩余的点就能组成一种没有一个点被染色的图。
然后对于第一种图,我们只需要从任意一个点开始遍历,判断与它相连的点是否颜色相同,颜色相同就impossible,否则就染成不同的色,这种图的服务区个数是颜色color1的数量。
对于第二种图,我们选择任意一个点开始染色即可。服务区数量:分别统计两种颜色的数量,取最小值作为服务区数量。
代码:
#include<bits/stdc++.h>
using namespace std;
const int N = 1e6+5;
#define int long long
typedef pair<int, int>PII;
const int Max = 0x3f3f3f3f3f3f3f3f;
const int Min = -0x3f3f3f3f3f3f3f3f;
int n, m, k;
int T;
vector<int>v[N];
int col[N];
bool st[N];
int cnt1, cnt2, ans;
bool dfs(int u, int c, int z) //分别代表:当前点,颜色,属于哪种图
{
if(z) //如果是第二种图,分别统计两个颜色数量,dfs后取小的值加进答案
{
if(c==1) cnt1++;
else cnt2++;
}
else if(c==2) ans++; //第一种图,一开始两种机场都染色的这种颜色,这个颜色的数量加入答案
if(!col[u]) col[u] = c; //如果没染色,就染上
if(col[u] != c) return false; //如果已经染过的颜色和应该要染的颜色不同,impossible
for(int i = 0; i < v[u].size(); i ++) //染色
{
int t = v[u][i];
if(!st[t])
{
st[t] = true;
if(!dfs(t, 3-c, z)) return false;
}
else if(col[u]!=3-col[t]) return false;
}
return true;
}
signed main()
{
ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
cin >> n >> m;
int f = 0;
while(m --)
{
int a, b, c;
cin >> a >> b >> c;
if(c==2)
{
if(col[a]==1||col[b]==1)
{
f = 1;
}
col[a] = 2, col[b] = 2;
}
else if(c==0)
{
if(col[a]==2||col[b]==2)
{
f = 1;
}
col[a] = 1, col[b] = 1;
}
else //建图
{
v[a].push_back(b);
v[b].push_back(a);
}
}
//如果0和2情况的染色冲突就impossible
if(f)
{
cout << "impossible\n";
return 0;
}
//含有被染过色的点的图,这种图的服务区数量直接在dfs中处理加上了
for(int i = 1; i <= n; i ++)
{
if(!st[i]&&col[i])
{
st[i] = true;
cnt1 = 0, cnt2 = 0;
if(!dfs(i, col[i], 0))
{
cout << "impossible\n";
return 0;
}
}
}
//没有一个点被染过色
for(int i = 1; i <= n; i ++)
{
if(!st[i])
{
cnt1 = 0, cnt2 = 0;
st[i] = true;
if(!dfs(i, 1, 1))
{
cout << "impossible\n";
return 0;
}
ans += min(cnt1, cnt2);//取小值
}
}
cout << ans;
return 0;
}