题目链接:https://icpcarchive.ecs.baylor.edu/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=5356
题目大意,有2*n个骑士,骑士分为两组,每个骑士一定会唯一的攻击另一组中的某一个骑士,要求找到一个集合,是集合内的骑士互不攻击,集合外的骑士从被某些集合内的骑士攻击(头晕~~~)。
相当于二分有向图并且有且只有n条边所以只有如下的情况
可以有若干的偶数边的环,以及指向环的若干条边
对于某个骑士,如果不被其他的攻击则一定在该集合内,否则,不存在集合内的骑士攻击该骑士;而对于他攻击的骑士一定不在集合内,否则,出现两个同在集合中的骑士攻击的情况。
对于偶数边的环,如果有边指向环则可以断开成链,否则,间隔取环的一半加入集合
代码实现的话,开始将所有入度为0的骑士加入队列,在删掉骑士攻击的骑士再攻击的骑士的边后,判断被该骑士攻击的骑士攻击的骑士是否入度为0,为0且为进入过队列,则进入队列。这样的话可以处理链以及有边相连的环。剩下的就是没有边相连的环,直接取一般输出。
代码如下:
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
queue<int> q;
int f[200050];
int d[200050];
int v[200050];
int n;
int now;
int main()
{
while (scanf("%d",&n)!=EOF)
{
memset(d,0,sizeof(d));
memset(f,0,sizeof(f));
memset(v,0,sizeof(v));
for (int i=1;i<=n;i++)
{
scanf("%d",&v[i]);
d[v[i]]++;
}
for (int i=n+1;i<=2*n;i++)
{
scanf("%d",&v[i]);
d[v[i]]++;
}
for (int i=1;i<=2*n;i++)
{
if (!d[i])
{
q.push(i);
}
}
while (!q.empty())
{
now=q.front();
q.pop();
f[now]=1;
if (f[v[now]]==-1)
{
continue;
}
f[v[now]]=-1;
d[v[v[now]]]--;
if (!d[v[v[now]]])
{
q.push(v[v[now]]);
}
}
for (int i=1;i<=2*n;i++)
{
if (f[i]==1)
{
printf("%d ",i);
}
else
{
if (i<=n && f[i]==0)
{
printf("%d ",i);
}
}
}
printf("\n");
}
return 0;
}