最短カウント

zuiduan

アイデア:重みはすべて1なので、BFSを使用して最短パスを見つけます。BFSでは、cnt [i]を使用して位置iまでの最短パスの数を記録します。最短距離を更新できる場合は、cntを更新します。 [i]の値。同じ距離が見つかった場合は、別のパスの番号が追加されます。

コード:

#pragma GCC optimize(2)
#include<bits/stdc++.h>
 
using namespace std;
typedef long long ll;
#define SIS std::ios::sync_with_stdio(false)
#define space putchar(' ')
#define enter putchar('\n')
#define lson root<<1
#define rson root<<1|1
typedef pair<int,int> PII;
typedef pair<int,PII> PIII;
const int mod=100003;
const int N=2e6+5;
const int inf=0x7f7f7f7f;

int gcd(int a,int b)
{
    
    
    return b==0?a:gcd(b,a%b);
}
 
ll lcm(ll a,ll b)
{
    
    
    return a*(b/gcd(a,b));
}
 
template <class T>
void read(T &x)
{
    
    
    char c;
    bool op = 0;
    while(c = getchar(), c < '0' || c > '9')
        if(c == '-')
            op = 1;
    x = c - '0';
    while(c = getchar(), c >= '0' && c <= '9')
        x = x * 10 + c - '0';
    if(op)
        x = -x;
}
template <class T>
void write(T x)
{
    
    
    if(x < 0)
        x = -x, putchar('-');
    if(x >= 10)
         write(x / 10);
    putchar('0' + x % 10);
}
ll qsm(int a,int b,int p)
{
    
    
    ll res=1%p;
    while(b)
    {
    
    
        if(b&1)
            res=res*a%p;
        a=1ll*a*a%p;
        b>>=1;
    }
    return res;
}
struct node
{
    
    
    int to,nex,w;
}edge[N];
int head[N],dis[N],cnt[N];
int tot,bcnt;
int n,m,s,k;

void add(int u,int v)
{
    
    
    edge[tot].to=v;
    edge[tot].nex=head[u];
    head[u]=tot++;
}
void bfs()
{
    
    
    queue<int> q;
    memset(dis,inf,sizeof dis);
    dis[1]=0;
    cnt[1]=1;
    q.push(1);
    while(q.size())
    {
    
    
        int u=q.front();q.pop();
        for(int i=head[u];~i;i=edge[i].nex)
        {
    
    
            int v=edge[i].to;
            if(dis[v]>dis[u]+1)
            {
    
    
                dis[v]=dis[u]+1;
                cnt[v]=cnt[u];
                q.push(v);
            }
            else if(dis[v]==dis[u]+1)
            {
    
    
                cnt[v]=(cnt[v]+cnt[u])%mod;
            }
        }
    }
}



int main()
{
    
    
  scanf("%d%d",&n,&m);
  memset(head,-1,sizeof head);
  while(m--)
  {
    
    
      int a,b;
      scanf("%d%d",&a,&b);
      add(a,b);add(b,a);
    
  }
  bfs();
  for(int i=1;i<=n;i++)printf("%d\n",cnt[i]);

   return 0;

}


おすすめ

転載: blog.csdn.net/qq_43619680/article/details/112995777