洛谷 P1710 地铁涨价

题目描述

博艾市除了有海底高铁连接中国大陆、台湾与日本，市区里也有很成熟的轨道交通系统。我们可以认为博艾地铁系统是一个无向连通图。博艾有N个地铁站，同时有M小段地铁连接两个不同的站。

地铁计价方式很简单。从A站到B站，每经过一小段铁路（连接直接相邻的两个点的一条边），就要收取1博艾元。也就是说，从A站到B站，选择的路径不一样，要价也会不同。

我们认为凡华中学在1号地铁站。学生们通过地铁通勤，他们当然知道选择最短路来坐车的话，票价最便宜。

然而博艾地铁公司经营不善，一直亏损，于是他们打算提价。提价一次就是将一小段铁路原来收费1元改收2元。同一小段的铁路不会多次提价。他们打算提价Q次。

学生们知道，如果他们到学校的一条最短路径中的一小段提价了，可以改变路径，使总票价不变。然而随着一条一条的铁路被提价，当居住在某个站附近的学生发现，提价后，没有任何一种方案可以从家到学校的费用和初始费用相等时，就会不满。

现在地铁公司希望知道，对于每一次涨价，有多少个站，学生会因为涨价而不满呢？

输入输出格式

输入格式：

第一行为三个整数N,M,Q。

接下来M行，每行2个整数ai,bi,表示第i条铁路连接的两个站。i表示铁路编号。

接下来Q行，每行一行整数rj，表示每次涨价的铁路编号。

输出格式：

Q行。每行一个整数表示不满的车站数量。

输入输出样例

输入样例#1：

5 6 5
1 2
1 3
4 2
3 2
2 5
5 3
5
2
4
1
3

输出样例#1：

0
2
2
4
4

说明

【样例解释】

次数 车站2 车站3 车站4 车站5
初始 1     1     2     2
1    1     1     2     2
2    1     2     2     3
3    1     2     2     3
4    2     2     3     3
5    2     2     4     3

【数据范围】

对于20%的数据 N≤100, Q≤30

对于40%的数据 Q≤30

对于70%的数据 正确的输出结果中，不会有超过50种不一样的整数（数据范围剧透解法系列）

对于100%的数据 N≤100000, Q≤M≤200000

这道题全是泪啊，比赛时打的暴力，似乎还写挂了。。（只拿了20分）

后来看了题解，好气呀，要不是比赛时想多玩两盘就能想出来，（装的好逼）。

晚上没状态，照着题解敲了一遍，第一次0分，第二次10分，弃了。

第二天发现同一个边的两个顶点抄的时候抄错了（泪奔.jpg）。

还有ans数组的前缀和求反了。

感谢Silver_N指导。

#include
#include
#include
using namespace std;
queueq;
int res,n,m,t,cnt,hd[100005],dis[100005],ban[200005],ans[200005];
bool vis[100005],bn[200005],inq[100005];
struct edge
{
    int to,nxt,num;
}v[400005];
void addedge(int x,int y,int id)
{
    ++cnt;
    v[cnt].to=y;
    v[cnt].nxt=hd[x];
    v[cnt].num=id;
    hd[x]=cnt;
}
int dfs(int u)
{
    int sum=1;
    inq[u]=1;
    for(int i=hd[u];i;i=v[i].nxt)
        if(!inq[v[i].to]&&!bn[v[i].num]&&dis[v[i].to]==dis[u]+1)
            sum+=dfs(v[i].to);
    return sum;
}
void bfs()
{
    q.push(1);
    dis[1]=0;
    vis[1]=1;
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(int i=hd[u];i;i=v[i].nxt)
            if(!vis[v[i].to])
            {
                vis[v[i].to]=1;
                dis[v[i].to]=dis[u]+1;
                q.push(v[i].to);
            }
    }
}
int main()
{
	scanf("%d%d%d",&n,&m,&t);
    for(int i=1;i<=m;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        addedge(x,y,i);
        addedge(y,x,i);
    }
    bfs();
    for(int i=1;i<=t;i++)
    {
        scanf("%d",&ban[i]);
        bn[ban[i]]=1;
    }
    inq[1]=1;
    for(int i=1;i<=m;i++)
        if(!bn[i])
        {
            if(inq[v[2*i].to]&&dis[v[2*i-1].to]==dis[v[2*i].to]+1&&!inq[v[2*i-1].to])
                dfs(v[2*i-1].to);
            if(inq[v[2*i-1].to]&&dis[v[2*i].to]==dis[v[2*i-1].to]+1&&!inq[v[2*i].to])
                dfs(v[2*i].to);
        }
    for(int i=t;i>=1;i--)
    {
        bn[ban[i]]=0;
        if(inq[v[2*ban[i]].to]&&dis[v[2*ban[i]-1].to]==dis[v[2*ban[i]].to]+1&&!inq[v[2*ban[i]-1].to])
            ans[i]+=dfs(v[2*ban[i]-1].to);
        if(inq[v[2*ban[i]-1].to]&&dis[v[2*ban[i]].to]==dis[v[2*ban[i]-1].to]+1&&!inq[v[2*ban[i]].to])
            ans[i]+=dfs(v[2*ban[i]].to);
    }
    for(int i=1;i<=t;i++)
    {
    	res+=ans[i];
		printf("%d\n",res);
	}
    return 0;
}