tkj
文章143
标签102
分类0
bzoj 1875: [SDOI2009]HH去散步

bzoj 1875: [SDOI2009]HH去散步

.

题意:无向图上有多少种方案可以从a刚好走t步走到b
题解:DP+矩阵快速幂
设f[i][j]表示走到i这条边j步的方案数。f[i][j]=∑f[k][j−1] (k.y=i.x且i,k不是同一条边)
矩阵就很好构造了。
代码:

#include<bits/stdc++.h>
using namespace std;
int n,m,t,A,B,fst[30],num=-1;
const int mod=45989;
struct matrix
{
    int a[150][150];
    int x,y;
    matrix()
    {
        x=y=0;
        memset(a,0,sizeof(a));
    }
    int* operator[](int x)
    {
        return a[x];
    }
}a,one,ans;
struct edge
{
    int x,y,n;
}e[150];
matrix operator*(matrix x,matrix y)
{
    matrix ans;
    ans.x=x.x;
    ans.y=y.y;
    for(int i=0;i<ans.x;i++)
    for(int j=0;j<ans.y;j++)
    for(int k=0;k<y.x;k++)
    ans[i][j]=(ans[i][j]+x[i][k]*y[k][j]%mod)%mod;
    return ans;
}
matrix pow(matrix x,int y)
{
    if(y==0)
    return one;
    if(y==1)
    return x;
    matrix ans=pow(x,y>>1);
    ans=ans*ans;
    if(y&1)
    ans=ans*x;
    return ans;
}
void ins(int x,int y)
{
    e[++num]={x,y,fst[x]};
    fst[x]=num;
}
int main()
{
    memset(fst,-1,sizeof(fst));
    scanf("%d%d%d%d%d",&n,&m,&t,&A,&B);
    for(int i=0;i<m;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        ins(x,y);
        ins(y,x);
    }
    one.x=one.y=num+1;
    for(int i=0;i<one.x;i++)
    one[i][i]=1;
    ans.x=1,ans.y=num+1;
    for(int i=0;i<=num;i++)
    if(e[i].x==A)
    ans[0][i]=1;
    a.x=a.y=num+1;
    for(int i=0;i<=num;i++)
    {
        for(int j=fst[e[i].y];~j;j=e[j].n)
        {
            if(j^i^1)
            {
                a[i][j]++;
            }
        }
    }/*
    for(int i=0;i<=num;i++)
    {
        for(int j=0;j<=num;j++)
        printf("%d ",a[i][j]);
        puts("");
    }*/
    a=pow(a,t-1);
    ans=ans*a;
    int Ans=0;
    for(int i=0;i<=num;i++)
    if(e[i].y==B)
    Ans=(Ans+ans[0][i])%mod;
    printf("%d",Ans);
}
本文作者:tkj
本文链接:https://tkj666.github.io/29/
版权声明:本文采用 CC BY-NC-SA 3.0 CN 协议进行许可