图论之路径条数专题

一直忙着金工实习+蓝桥杯,好久没有看图论了,今天就小试几题享受下被虐的快感。

1.最短路+拓扑

首先来几个结论:

1.最短路图没有环(可以用反证法证明)

2.dis[u]+edge[u,v]=dis[v],那么u,v端点的边一定在最短路图上。

因此,我们就可以先枚举起始点跑SPFA,然后把最短路找出来,假如一个边的端点为u,v,那么经过它的为cnt1[u]*cnt2[v],cnt1为正着最短路图上过u的边,cnt2为反着(注意到v结束也是一种,所以结尾后+1)。

因此,我们求两边拓扑排序即可,下面是AC代码(这里正反图用奇偶存储来区别):

cpp 复制代码
#include<bits/stdc++.h>
#define mod 1000000007
using namespace std;

const int N=1502;
const int M=100010;
bool vis[N];
int dis[N],in1[N],in2[N],cnt1[N],cnt2[N];
int n,m,u,v,w,head[N],is[M],cnt=1;
int ans[M];
struct node{
	int dian,next,zhi;
}edge[M];
void add(int u,int v,int w){//i&1为反图 
	edge[++cnt].dian=v;
	edge[cnt].zhi=w;
	edge[cnt].next=head[u];
	head[u]=cnt;
	edge[++cnt].dian=u;
	edge[cnt].zhi=w;
	edge[cnt].next=head[v];
	head[v]=cnt; 
}
void spfa(int s){
	memset(vis,0,sizeof(vis));
	memset(dis,0x7f7f7f7f,sizeof(dis));
	queue<int> q;
	vis[s]=1;
	q.push(s);
	dis[s]=0;
	while(!q.empty()){
		int ck=q.front();
		q.pop();
		vis[ck]=0;
		for(int i=head[ck];i!=-1;i=edge[i].next){
			if(i&1) continue;
			if(dis[edge[i].dian]>dis[ck]+edge[i].zhi){
				dis[edge[i].dian]=dis[ck]+edge[i].zhi;
				if(!vis[edge[i].dian]){
					vis[edge[i].dian]=1;
					q.push(edge[i].dian);
				}
			}
		}
	}
}
void new1(){
	memset(is,0,sizeof(is));
	memset(in1,0,sizeof(in1));
	memset(in2,0,sizeof(in2));
	for(int u=1;u<=n;u++){
		for(int i=head[u];i!=-1;i=edge[i].next){
			if(i&1) continue;
			if(dis[u]+edge[i].zhi==dis[edge[i].dian]){
				is[i]=1;
				is[i^1]=1;
				in1[edge[i].dian]++;
				in2[u]++;
			}
		}
	}
}
void topo(int s){
	memset(cnt1,0,sizeof(cnt1));
	memset(cnt2,0,sizeof(cnt2));
	queue<int> q;
	q.push(s);
	cnt1[s]=1;
	while(!q.empty()){
		int u=q.front();
		q.pop();
		for(int i=head[u];i!=-1;i=edge[i].next){
			if(!is[i]||(i&1)) continue;
			int v=edge[i].dian;
			in1[v]--;
			cnt1[v]=(cnt1[v]+cnt1[u])%mod;
			if(!in1[v]) q.push(v);
		}
	}
	for(int i=1;i<=n;i++){
		if(!in2[i]){
			cnt2[i]=1;
			q.push(i);
		}
	}
	while(!q.empty()){
		int u=q.front();
		q.pop();
		for(int i=head[u];i!=-1;i=edge[i].next){
			if(!is[i]||!(i&1)) continue;
			int v=edge[i].dian;
			in2[v]--;
			cnt2[v]=(cnt2[v]+cnt2[u])%mod;
			if(!in2[v]){
				q.push(v);
				cnt2[v]++;//自己 
			}
		}
	}
}
void cal(){
	for(int u=1;u<=n;u++){
		for(int i=head[u];i!=-1;i=edge[i].next){
			if((i&1)||!is[i]) continue;
			int v=edge[i].dian;
			ans[i>>1]=(ans[i>>1]+cnt1[u]*cnt2[v]%mod)%mod;
		}
	}
}
void solve(){
	for(int i=1;i<=n;i++){
		spfa(i);
		new1();
		topo(i);
		cal();
	}
	for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
}
int main(){
	cin>>n>>m;
	memset(head,-1,sizeof(head));
	for(int i=1;i<=m;i++){
		scanf("%d%d%d",&u,&v,&w);
		add(u,v,w);
	}
	solve();
	return 0;
}

2.最短路+DP

首先跑个最短路,然后我们令f[i][j]表示走到i不超过d+j的条数。

易得状态转移方程:f[x][k]=(f[y][dis[x]+k-dis[y]-len2[x][y])%p;

至于顺序我们直接记忆化搜素即可。

这里有个比较麻烦的细节:

如何判断0环?

1.不是一有0环就-1,当你的0环不在最短路+k涉及的路径上它起不了作用,那么这如何判?

注意到此时dis[x]+k-dis[y]-len2[x][y]为负值,我们判一下这种情况即可。

2.当一个二维位置即同一个点,同一个小于距离出现时,说明0环,判-1.

因此我们还要用v[n][55]来记录有没有访问过(注意dfs完后归0操作)

3.注意到0环涉及起点1的情况,这也是为什么起点设在n+1源点而不是1的原因。

下面是AC代码:

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
const int N=100000;
int t,n,m,k,p,a,b,c,flag;
vector<int> edge[N+1],len[N+1];
vector<int> edge1[N+1],len2[N+1];
int dis[N+1],vis[N+1];
int f[N+1][55];
bool v[N+1][55];
struct ty{
    int x,dis;
    bool operator< (const ty &a) const{
        return dis>a.dis;
    }
};
priority_queue<ty> q;
void dij(int s){
    memset(dis,0x7f,sizeof(dis));
    memset(vis,0,sizeof(vis));
    dis[s]=0;
    q.push({s,0});
    while(!q.empty()){
        ty tmp=q.top();
        q.pop();
        if(vis[tmp.x]) continue;
        vis[tmp.x]=1;
        for(int i=0;i<edge[tmp.x].size();i++){
            int y=edge[tmp.x][i];
            if(dis[y]>dis[tmp.x]+len[tmp.x][i]){
                dis[y]=dis[tmp.x]+len[tmp.x][i];
                q.push({y,dis[y]});
            }
        }
    }
}
int dfs(int x,int k){
    if(f[x][k]!=-1) return f[x][k];
    f[x][k]=0;
    v[x][k]=1;
    for(int i=0;i<edge1[x].size();i++){
        int y=edge1[x][i];
        int t=dis[x]+k-dis[y]-len2[x][i];
        if(t<0) continue;
        if(v[y][t]) flag=1;
        if(flag) return 0;    
        f[x][k]=(f[x][k]+dfs(y,t))%p;
    }
    v[x][k]=0;
    return f[x][k];
}
int main(){
    cin>>t;
    while(t--){
        scanf("%d%d%d%d",&n,&m,&k,&p);
        for(int i=1;i<=n+1;i++){
            edge[i].clear();
            len[i].clear();
            edge1[i].clear();
            len2[i].clear();
        }
        for(int i=1;i<=m;i++){
            scanf("%d%d%d",&a,&b,&c);
            edge[a].push_back(b);
            len[a].push_back(c);
            edge1[b].push_back(a);
            len2[b].push_back(c);
        }
        edge[n+1].push_back(1);
        len[n+1].push_back(0);
        edge1[1].push_back(n+1);
        len2[1].push_back(0);
        dij(n+1);
        memset(f,-1,sizeof(f));
        memset(v,0,sizeof(v));
        f[n+1][0]=1;
        int ans=0;
        flag=0;
        for(int i=0;i<=k;i++){
            ans=(ans+dfs(n,i))%p;
        }
        if(flag) printf("-1\n");
        else printf("%d\n",ans);
    }
}
相关推荐
Ttang236 分钟前
Leetcode:118. 杨辉三角——Java数学法求解
算法·leetcode
喜欢打篮球的普通人6 分钟前
rust模式和匹配
java·算法·rust
java小吕布20 分钟前
Java中的排序算法:探索与比较
java·后端·算法·排序算法
tumu_C28 分钟前
C++模板特化实战:在使用开源库boost::geometry::index::rtree时,用特化来让其支持自己的数据类型
c++·开源
win x37 分钟前
链表(Linkedlist)
数据结构·链表
杜若南星1 小时前
保研考研机试攻略(满分篇):第二章——满分之路上(1)
数据结构·c++·经验分享·笔记·考研·算法·贪心算法
路遇晚风1 小时前
力扣=Mysql-3322- 英超积分榜排名 III(中等)
mysql·算法·leetcode·职场和发展
Neophyte06081 小时前
C++算法练习-day40——617.合并二叉树
开发语言·c++·算法
木向1 小时前
leetcode104:二叉树的最大深度
算法·leetcode