CF1027D Mouse Hunt题解
题目:
伯兰州立大学的医学部刚刚结束了招生活动。和以往一样,约80%的申请人都是女生并且她们中的大多数人将在未来4年(真希望如此)住在大学宿舍里。
宿舍楼里有nn个房间和一只老鼠!女孩们决定在一些房间里设置捕鼠器来除掉这只可怕的怪物。在ii号房间设置陷阱要花费c_ic
i
伯兰币。房间编号从11到nn。
要知道老鼠不是一直原地不动的,它不停地跑来跑去。如果tt秒时它在ii号房间,那么它将在t+1t+1秒时跑到a_ia
i
号房间,但这期间不会跑到别的任何房间里(i=a_ii=a
i
表示老鼠没有离开原来的房间)。时间从00秒开始,一旦老鼠窜到了有捕鼠器的房间里,这只老鼠就会被抓住。
如果女孩们知道老鼠一开始在哪里不就很容易吗?不幸的是,情况不是这样,老鼠在第00秒时可能会在从11到nn的任何一个房间内。
那么女孩们最少要花多少钱设置捕鼠器,才能保证老鼠无论从哪个房间开始流窜最终都会被抓到?
分析:
stack解决
a向a[x]连边
代码:
#include<cstdio>
#include<stack>
using namespace std;
int a[200005];
int cnt=0;
int vis[200005],c[200005];
int flag;
int tag;
stack<int>s;
void dfs(int x)
{
if(a[x]==0)
{
return ;
}
cnt++;
if(cnt>vis[x]&&vis[x])
{
flag=1;
tag=x;
a[x]=0;
vis[x]=0;
return ;
}
s.push(x);
vis[x]=cnt;
dfs(a[x]);
vis[x]=0;
a[x]=0;
}
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d",&c[i]);
}
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
}
int sum=0;
for(int i=1;i<=n;i++)
{
while(!s.empty())s.pop();
flag=0;
dfs(i);
if(flag)
{
int ans=2147483647;
if(!s.empty())
{
while(!s.empty()&&s.top()!=tag)
{
ans=min(ans,c[s.top()]);
s.pop();
}
if(!s.empty())
ans=min(ans,c[s.top()]);
sum+=ans;
}
}
}
printf("%d\n",sum);
return 0;
}