JZOJ1243. TreeCount

题目

Description

给出一个有N(2<=N<=1000)个顶点M(N-1<=M<=N*(N-1)/2)条边的无向连通图。设dist1[i]表示在这个无向连通图中，顶点i到顶点1的最短距离。
现在要求你在这个图中删除M-(N-1)条边，使得这个图变成一棵树。设dist2[i]表示在这棵树中，顶点i到顶点1的距离。
你的任务是求出有多少种删除方案，使得对于任意的i，满足dist1[i]=dist2[i]。

Input

第一行，两个整数，N，M，表示有N个顶点和M条边。
接下来有M行，每行有3个整数 ，表示顶点x和顶点y有一条长度为len的边。 数据保证不出现自环、重边。

Output

输出一个整数，表示满足条件的方案数 mod 2147483647的答案。

Sample Input

3 3
1 2 2
1 3 1
2 3 1

Sample Output

2

Hint

【样例解释】 删除第一条边或者第三条边都能满足条件，所以方案数为2。
【数据规模】 对于30%的数据,2<=N<=5,M<=10
对于50%的数据，满足条件的方案数不超过10000
对于100%的数据，2<=N<=1000

分析

如果要保证最短路不变，那么我们可以知道：连接个点和根节点之间，至少有一条最短路没有被改变。

我们沿着这个思路向下想，如果某些边不在最短路上面，那它们肯定会被删除。因为保留下来的边都是最短路上面的。

而考虑最短路上面的边，这些边我们就要有选择地删除，不能全部删除，也不能保留太多。

如果一个点有多个点可以更新它，那么我们只需要保留其中一条边就可以了，其他的都必须删掉。此时，产生的方案数就是可以更新它的点的数量。

题解

求有多少个点来更新一个点可以在SPFA中处理一下。

• 具体做法：

设$f_i$表示有多少个点可以更新当前这个点。

如果现在找到更优的方案，那么$f_i=1$

如果现在找到的方案等于已经知道的最优解，那么$f_i++$

最后，只需要将所有的$f_i$乘起来就是答案了

• 注意：要开long long

• 特别要注意：$f_i$表示有多少个点可以更新当前这个点，而不是到达这个点最短路的条数。

code

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string.h>
#include <cmath>
#include <stdlib.h>
#include <math.h>
#define ll long long 
#define mo %2147483647
using namespace std;

int dis[1003],b[1003],f[1003],data[3000003];
int next[1000003],a[1000003],v[1000003];
int n,m,x,y,z;
bool bz[1003];
ll ans;

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&x,&y,&z);

        next[i]=b[x];
        a[i]=y;
        v[i]=z;
        b[x]=i;

        next[i+m]=b[y];
        a[i+m]=x;
        v[i+m]=z;
        b[y]=i+m;
    }

    int i=0,j=1;
    data[1]=1;
    memset(bz,1,sizeof(bz));
    bz[1]=0;
    memset(dis,127,sizeof(dis));
    dis[1]=0;
    for(int i=1;i<=n;i++)
        f[i]=1;

    while(i<j)
    {
        i++;
        for(int k=b[data[i]];k;k=next[k])
        {
            if(dis[a[k]]>dis[data[i]]+v[k])
            {
                dis[a[k]]=dis[data[i]]+v[k];
                f[a[k]]=1;
                if(bz[a[k]])
                {
                    j++;
                    data[j]=a[k];
                    bz[a[k]]=0;
                }
            }
            else if(dis[a[k]]==dis[data[i]]+v[k])f[a[k]]++;
        }
        bz[data[i]]=1;
    }
    ans=1;
    for(int i=1;i<=n;i++)
        ans=(ans*f[i])mo;
    printf("%lld",ans);
}