並查集思路:
- 首先利用並查集計算出初始時連通塊的數量。
- 然後攻掉一個城市,即去掉一個點,再次計算連通塊的數量,注意:被去掉的點此刻算作一個連通塊,連通塊的數量是一直遞增的!
- 找到 去掉一個點時連通塊數量變化的規律來設置判斷是否要發出紅色警報。如果被攻佔的城市之前是單獨一個城市或者是連通塊的不影響其他城市連通的城市,cnt=cnt0;如果是連通塊的橋的時候,被攻佔了cnt=cnt0+2;
關於並查集,這裏還有詳細的介紹和例子!!
https://blog.csdn.net/weixin_43535668/article/details/104413573
https://blog.csdn.net/weixin_43535668/article/details/104436391
#include<iostream>
#include<algorithm>
#include<string>
#include<stdio.h>
using namespace std;
#define N 505
int fa[N];//並查集記錄數組
int edge[N][N];
int vis[N];
int lost[N];
void init()
{
for (int i = 0; i <=N ; i++)
{
fa[i] = i;
}
}
int find(int x)
{
if (fa[x] == x)
return x;
else
return find(fa[x]);
}
void merge(int x, int y)
{
int rx = find(x);
int ry = find(y);
if (rx != ry)
{
fa[rx] = ry;
}
}
int count(int n)//返回連通塊的數量
{
int res = 0;
for (int i = 0; i < n; i++)
{
if (fa[i] == i)
res++;
}
return res;
}
int main()
{
int n, m, x, y;
cin >> n >> m;//城市數目 道路數目
int con[5000][2];
init();//並查集數組初始化
for (int i = 0; i < m; i++)
{
cin >> con[i][0] >> con[i][1];//記錄道路信息
edge[con[i][0]][con[i][1]] = 1;
edge[con[i][1]][con[i][0]] = 1;
merge(con[i][0], con[i][1]);
}
int t, cnt0 = count(n);//返回連通塊的數量
cin >> t;
while (t--)
{
cin >> x;
lost[x] = 1;
init();
for (int i = 0; i < m; i++)
{
if (!(lost[con[i][0]] || lost[con[i][1]]))//重新統計丟失城市後的連通塊
{
merge(con[i][0], con[i][1]);
}
}
for (int i = 0; i < m; i++) {
if (lost[con[i][0]] || lost[con[i][1]]);
else merge(con[i][0], con[i][1]);
}
int cnt = count(n);
if (cnt == cnt0 + 1 || cnt <= cnt0)//這個判斷條件要注意
//如果被攻佔的城市之前是單獨一個城市或者是連通塊的不影響其他城市連通的城市,cnt=cnt0;
//如果是連通塊的橋的時候,被攻佔了cnt=cnt0+2;
{
printf("City %d is lost.\n", x);
}
else printf("Red Alert: City %d is lost!\n", x);
cnt0 = cnt;
}
if(cnt0==n)
printf("Game Over.\n");
return 0;
}
BFS思路:
使用BFS統計聯通分量時,沒有計算失守的城市,若本次連通分量個數大於上一次的則要紅色警報,否則只需要打印lost即可。
思路和上面大同小異,就是計算連通塊數量。
#include<iostream>
#include<queue>
#include<algorithm>
#include<stdio.h>
#include<string>
#include<string.h>
using namespace std;
#define N 505
int edge[N][N], vis[N], lost[N], n;
queue<int> qu;
void bfs()
{
while (!qu.empty())
{
int e = qu.front();
qu.pop();
vis[e] = 1;
for (int i = 0; i < n; i++)
{
if ((edge[e][i] || edge[i][e]) && !vis[i] && !lost[i])
{
qu.push(i);
vis[i] = 1;
}
}
}
}
int main()
{
int m, x, y, t, precnt = 0, cnt = 0;
cin >> n >> m;//城市數量 道路數量
for (int i = 0; i < m; i++)
{
cin >> x >> y;
edge[x][y] = 1; edge[y][x] = 1;
}
for (int i = 0; i < n; i++)
{
if (!vis[i])
{
vis[i] = 1;
qu.push(i);//進隊列
bfs();
precnt++;//計算連通塊的個數
}
}
cin >> t;
while (t--)
{
memset(vis, 0, sizeof(vis));
cnt = 0;
cin >> x;
lost[x] = 1;
for (int i = 0; i < n; i++)
{
if (!vis[i] && !lost[i])
{
vis[i] = 1;
qu.push(i);
bfs();
cnt++;
}
}
if(cnt<=precnt)//這裏的判斷條件與並查集的不同,這裏的城市被攻佔了就是去掉了
printf("City %d is lost.\n", x);
else printf("Red Alert: City %d is lost!\n", x);
precnt = cnt;
}
if(precnt==0)
printf("Game Over.\n");
system("pause");
return 0;
}