[LOJ]#2344. 「JOI 2016 Final」铁路票价 最短路

Solution

先BFS一次跑出最短路，然后动态维护每个点现在还有多少个直接连到它的点能够通过最短路走到它就可以了。每个点的值最多只要一次变为$0$，复杂度是线性的。

Code

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
const int Maxn=100010,Maxm=200010;
const int inf=2147483647;
int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
	return x*f;
}
int n,m,q,f[Maxn],g[Maxn];bool mark[Maxm];
bool cmp(int x,int y){return f[x]<f[y];}
vector<int>h[Maxn];
struct Edge{int x,y,next;}E[Maxm],e[Maxm<<1];
int last[Maxn],len=0;
void ins(int x,int y)
{
	int t=++len;
	e[t].y=y;e[t].next=last[x];last[x]=t;
}
void bfs()
{
	memset(f,-1,sizeof(f));
	queue<int>q;q.push(1);f[1]=0;
	while(!q.empty())
	{
		int x=q.front();q.pop();
		for(int i=last[x];i;i=e[i].next)
		{
			int y=e[i].y;
			if(f[y]!=-1)continue;
			f[y]=f[x]+1;q.push(y);
		}
	}
}
int ans=0;
void dfs(int x)
{
	ans++;
	for(int i=last[x];i;i=e[i].next)
	{
		int y=e[i].y;
		if(f[x]+1!=f[y]||mark[(i+1)>>1])continue;
		g[y]--;if(!g[y])dfs(y);
	}
}
int main()
{
	n=read(),m=read(),q=read();
	for(int i=1;i<=m;i++)
	{
		int x=read(),y=read();
		E[i].x=x,E[i].y=y;
		ins(x,y),ins(y,x);
	}
	bfs();
	g[1]=1;
	for(int x=1;x<=n;x++)
	{
		for(int i=last[x];i;i=e[i].next)
		{
			int y=e[i].y;
			if(f[x]+1==f[y])g[y]++;
		}
	}
	while(q--)
	{
		int u=read();mark[u]=true;
		int x=E[u].x,y=E[u].y;
		if(f[x]>f[y])swap(x,y);
		if(f[x]+1==f[y])
		{
			if(g[x])
			{
				g[y]--;
				if(!g[y])dfs(y);
			}
		}
		printf("%d\n",ans);
	}
}