【洛谷 P1710】地铁涨价（dfs+bfs）

#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
struct node{
int s,t,num;
}road[200010];
int a[800010],nxt[800010],p[100010],tot;
int dis[100010],dft[100010],ans[200010];
int n,m,q,nm[200010],cnt;
bool vis[100010],b[200010];
inline void add(int x,int y)
{
tot++; a[tot]=y; nxt[tot]=p[x]; p[x]=tot;
tot++; a[tot]=x; nxt[tot]=p[y]; p[y]=tot;
}
inline void bfs(int s,int dis[])
{
memset(vis,0,sizeof(vis));
queue<int>que;
dis[s]=0; vis[s]=1; que.push(s);
while(!que.empty())
{
int u=que.front(); que.pop();
for(int i=p[u];i!=-1;i=nxt[i])
if(dis[a[i]]>dis[u]+1)
{
dis[a[i]]=dis[u]+1;
if(!vis[a[i]]) vis[a[i]]=1,que.push(a[i]);
}
}
}
void dfs(int x,int fa)
{
for(int i=p[x];i!=-1;i=nxt[i])
if(a[i]!=fa&&dis[a[i]]==dft[x]+1&&dis[a[i]]!=dft[a[i]])
{
dft[a[i]]=dft[x]+1; cnt++;
dfs(a[i],x);
}
}
int main()
{
freopen("int.txt","r",stdin);
int i,j;
memset(b,1,sizeof(b));
memset(p,-1,sizeof(p));
memset(nxt,-1,sizeof(nxt));
memset(dis,127/3,sizeof(dis));
memset(dft,127/3,sizeof(dft));
scanf("%d%d%d",&n,&m,&q);
for(i=1;i<=m;++i)
{
int x,y;
scanf("%d%d",&x,&y);
road[i].s=x; road[i].t=y; road[i].num=i;
add(x,y);
}
bfs(1,dis); tot=0;
memset(p,-1,sizeof(p));
memset(nxt,-1,sizeof(nxt));
for(i=1;i<=q;++i) scanf("%d",&nm[i]),b[nm[i]]=0;
for(i=1;i<=m;++i)
if(b[i]) add(road[i].s,road[i].t);
bfs(1,dft);
for(i=q;i>0;--i)
{
int x=road[nm[i]].s,y=road[nm[i]].t;cnt=0;
if(dis[x]==dft[x]&&dis[y]==dft[x]+1&&dft[y]!=dis[y]) cnt++,dft[y]=dft[x]+1,dfs(y,x);
if(dis[y]==dft[y]&&dis[x]==dft[y]+1&&dft[x]!=dis[x]) cnt++,dft[x]=dft[y]+1,dfs(x,y);
ans[i]=cnt; add(x,y);
}
for(i=2;i<=q;++i) ans[i]+=ans[i-1];
for(i=1;i<=q;++i) printf("%d\n",ans[i]);
return 0;
}