解题报告:luogu P1144 最短路计数

题目链接:P1144 最短路计数
很简单的一道\(dfs\),然而我又跑了一遍\(dij\)和排序,时间复杂度是\(O(nlog n)\)
注意:\(1\).搜索时向\(dis[j]=dis[cur]-1\)的点\(j\)搜就好了;
\(2\).注意记录重边,而且我们存的无向图是两倍边,记得都加(\(233\)
\(3\).记得把所有不能更新的都更一遍,不要只更新最远点(就是记录一个\(deg[]\)
上代码:

\(Code\):

#include
#include
#include
#include
#define mod 100003
using namespace std;
struct node
{
    int l,r;
}a[2000005];
bool cmp(node n,node m){if(n.l^m.l) return n.l,vector >,greater > >q;
void dij()
{
    while(!q.empty())
    {
        int j=q.top().second;
        q.pop();
        if(vis[j]) continue;
        vis[j]=1;
        for(int i=head[j];i;i=e[i].nxt)
        {
            int k=e[i].to;
            if(dis[k]>dis[j]+1)
            {
                dis[k]=dis[j]+1;
                q.push(make_pair(dis[k],k));
            }
        }
    }
}
int dp[1000005],deg[1000005];
int n,m,ll=-1,rr=-1,root=-1;
int dfs(int cur)
{
    //if(deg[cur]<=1&&cur!=root) return dp[cur]=1;
    if(dp[cur]) return dp[cur];
    for(int i=head[cur];i;i=e[i].nxt)
    {
        int j=e[i].to;
        if(dis[j]==dis[cur]-1) dp[cur]=(dp[cur]+e[i].num*dfs(j)%mod)%mod;
    }
    return dp[cur]%mod;
}
int main()
{
    //freopen("data.in","r",stdin);
    //freopen("baoli.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++) scanf("%d%d",&a[i].l,&a[i].r);
    sort(a+1,a+m+1,cmp);
    for(int i=1;i<=m;i++)
    {
        if(a[i].l==ll&&a[i].r==rr) 
        {
            e[cnt].num++;
            e[cnt-1].num++;
            continue;
        }
        else
        {
            ll=a[i].l,rr=a[i].r;
            add(ll,rr);
            add(rr,ll);
            deg[ll]++,deg[rr]++;
        }
    }
    for(int i=2;i<=n;i++) dis[i]=mod/10*mod;
    q.push(make_pair(0,1));
    //for(int i=1;i<=n;i++) if(dis[i]>dis[root]) root=i;
    dij();
    dp[1]=1;
    for(int i=2;i<=n;i++) dp[i]=0;
    for(int i=1;i<=n;i++) if(!dp[i]) dfs(i);
    //dfs(root);
    for(int i=1;i<=n;i++) printf("%d\n",dp[i]%mod);
}

你可能感兴趣的:(解题报告:luogu P1144 最短路计数)