[模拟赛]路途

题意

给定\(n\)个点\((n\le 40)\)\(m\)\((m\le 1000)\)长为\(1\)的边(无重边有自环)。
\(q\)个询问\((q\le 40)\),每次询问两个点之间长度在\(l,r(l,r\le 10^9,(r-1)\le 200)\)之间的路径数。

分析

一开始打了个爆搜结果一分都没有。
正解是弗洛伊德。

30pts

\(f[i][j][k]\)表示从\(i\)\(j\)路径长度为\(k\)的方案数。
很明显\(k=1\)的时候\(f[i][j][k]=g[i][j]\)\(g[i][j]\)为邻接矩阵,表示\(i,j\)之间有一条边。
我们可以从\(2\)\(l-1\)递推一遍,然后再从\(l\)\(r\)递推一遍,这次统计方案数。
注意\(l=1\)时的特判,这样子30分到手。
时间复杂度\(O(n^3qr)\)

70pts

然后我们可以注意到,上面的运算和矩阵的运算有点相似。
我们发现,邻接矩阵就是系数矩阵,代表\(k-1\)长度的路径能否转移到\(k\)长度的路径。
我们递推一次就相当于乘以一个邻接矩阵,我们可以用矩阵快速幂快速递推出\(k=l\)时的f矩阵。
时间复杂度\(O(n^3q((r\ −\ l)\ +\ log\ n))\)

100pts

我们要拿满分还必须优化掉\((r-l)\)前面的系数。
我们预处理一遍前缀和,用一个\(sum\)矩阵保存\(1~200\)的系数前缀和。
算出\(A^l\)之后我们只要乘上这个前缀和矩阵就可以直接算出答案了,这样我们就能把\((r-l)\)前的系数优化到\(1\)了。

代码

30pts

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#define ll long long
using namespace std;
const int M=1009,N=50;
const int Maxn=10000009;
const int mod=2333;
int read(){
    char c;int num,f=1;
    while(c=getchar(),!isdigit(c))if(c=='-')f=-1;num=c-'0';
    while(c=getchar(), isdigit(c))num=num*10+c-'0';
    return f*num;
}
int n,m,s,t,l,r,g[50][50];
int f[50][50][5],Case;
void work(){
    s=read();t=read();l=read();r=read();
    int ans=0;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++){
            f[i][j][1]=g[i][j];
            if(l==1&&i==s&&j==t)
                ans+=f[i][j][1];
        }
    for(int k=2;k<l;k++)
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++){
                f[i][j][k%2]=0;
                for(int mid=1;mid<=n;mid++)
                    f[i][j][k%2]=(f[i][j][k%2]+f[i][mid][(k-1)%2]*g[mid][j])%mod;
            }
    for(int k=max(l,2);k<=r;k++)
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++){
                f[i][j][k%2]=0;
                for(int mid=1;mid<=n;mid++)
                    f[i][j][k%2]=(f[i][j][k%2]+f[i][mid][(k-1)%2]*g[mid][j])%mod;
                if(i==s&&j==t)
                    ans=(ans+f[i][j][k%2])%mod;
            }
    printf("%d\n",ans);
}
int main()
{
    freopen("l.in","r",stdin);
    freopen("l.out","w",stdout);
    n=read();m=read();
    int u,v;
    for(int i=1;i<=m;i++){
        u=read();v=read();
        g[u][v]=g[v][u]=1;
    }
    Case=read();
    while(Case--)work();
    return 0;
}

70pts

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#define ll long long
using namespace std;
const int M=1009,N=50;
const int Maxn=10000009;
const int mod=2333;
struct mat{
    int a[50][50];
    mat(){
        memset(a,0,sizeof(a));
    }
}f[209],g,now;
int read(){
    char c;int num,f=1;
    while(c=getchar(),!isdigit(c))if(c=='-')f=-1;num=c-'0';
    while(c=getchar(), isdigit(c))num=num*10+c-'0';
    return f*num;
}
int n,m,s,t,l,r,Case;
mat operator *(mat A,mat B){
    mat C;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            C.a[i][j]=0;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            for(int k=1;k<=n;k++)
                C.a[i][j]=(C.a[i][j]+A.a[i][k]*B.a[k][j]%mod)%mod;
    return C;
}
mat operator ^(mat A,int p){
    mat C=A;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++)
            C.a[i][j]=0;
        C.a[i][i]=1;
    }
    for(;p;p>>=1,A=A*A)
        if(p&1)C=C*A;
    return C;
}
mat operator +(mat A,mat B){
    mat ans;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            ans.a[i][j]=A.a[i][j]+B.a[i][j];
            if(ans.a[i][j]>=mod)ans.a[i][j]-=mod;
        }
    }
    return ans;
}
void work(){
    s=read();t=read();l=read();r=read();
    int ans=0;
    if(l>r){printf("0\n");return ;}
    now=g^l;
    for(int k=1;k<=(r-l+1);k++){
        ans=(ans+now.a[s][t])%mod;
        now=now*g;
    }
    printf("%d\n",ans);
    return ;
}
int main()
{
    freopen("l.in","r",stdin);
    freopen("l.out","w",stdout);
    n=read();m=read();
    int u,v;
    for(int i=1;i<=m;i++){
        u=read();v=read();
        g.a[u][v]=g.a[v][u]=1;
    }
    Case=read();
    while(Case--)work();
    return 0;
}

100pts

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#define ll long long
using namespace std;
const int M=1009,N=50;
const int Maxn=10000009;
const int mod=2333;
struct mat{
    int a[50][50];
    mat(){
        memset(a,0,sizeof(a));
    }
}sum[209],g,now;
int read(){
    char c;int num,f=1;
    while(c=getchar(),!isdigit(c))if(c=='-')f=-1;num=c-'0';
    while(c=getchar(), isdigit(c))num=num*10+c-'0';
    return f*num;
}
int n,m,s,t,l,r,Case;
mat operator *(mat A,mat B){
    mat C;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            C.a[i][j]=0;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            for(int k=1;k<=n;k++)
                C.a[i][j]=(C.a[i][j]+A.a[i][k]*B.a[k][j]%mod)%mod;
    return C;
}
mat operator ^(mat A,int p){
    mat C=A;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++)
            C.a[i][j]=0;
        C.a[i][i]=1;
    }
    for(;p;p>>=1,A=A*A)
        if(p&1)C=C*A;
    return C;
}
mat operator +(mat A,mat B){
    mat ans;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            ans.a[i][j]=A.a[i][j]+B.a[i][j];
            if(ans.a[i][j]>=mod)ans.a[i][j]-=mod;
        }
    }
    return ans;
}
void work(){
    s=read();t=read();l=read();r=read();
    int ans=0;
    if(l>r){printf("0\n");return ;}
    now=g^l;
    now=now*sum[r-l];
    printf("%d\n",now.a[s][t]);
    return ;
}
int main()
{
    freopen("l.in","r",stdin);
    freopen("l.out","w",stdout);
    n=read();m=read();
    int u,v;
    for(int i=1;i<=m;i++){
        u=read();v=read();
        g.a[u][v]=g.a[v][u]=1;
    }
    for(int i=1;i<=n;i++)now.a[i][i]=1;
    sum[0]=now;
    for(int i=1;i<=201;i++){
        now=now*g;
        sum[i]=sum[i-1]+now;
    }
    Case=read();
    while(Case--)work();
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/onglublog/p/9887194.html