图论:差分约束算法

题目链接:点击这里
在这里插入图片描述

代码如下:

#include<cstring>
#include<iostream>
#include<queue>
#include<cstdio>
#include<algorithm>
using namespace std;
struct edge{
    
    
 int to,w,next;
}kkk[10005];//存边用
int head[5005],neg[5005],dis[5005],vis[5005],numedge,n,m;
void addedge(int u,int v,int w){
    
    
 	kkk[++numedge].to=v;
 	kkk[numedge].w=w;
 	kkk[numedge].next=head[u];
 	head[u]=numedge;
}//链式前向星存边
bool spfa(int s){
    
    
 	queue<int> q;
 	memset(dis,63,sizeof(dis));
 	dis[s]=0,vis[s]=1;
 	q.push(s);
 	while(!q.empty()){
    
    
  		int u=q.front();
  		q.pop();
  		vis[u]=0;
  		for(int i=head[u];i;i=kkk[i].next){
    
    
   			int v=kkk[i].to;
   			if(dis[v]>dis[u]+kkk[i].w){
    
    
    			dis[v]=dis[u]+kkk[i].w;
    			if(!vis[v]){
    
    //不在队列中
     				vis[v]=1,neg[v]++;//判断负环
     				if(neg[v]==n) return true;
     				q.push(v);//入队
    			}
   			}
  		}
 	}
 	return false;
}
int main()
{
    
    
 	scanf("%d%d",&n,&m);
 	for(int i=1;i<=n;i++)
  		addedge(0,i,0);
 	for(int i=1;i<=m;i++){
    
    
  		int v,u,w;
  		scanf("%d%d%d",&u,&v,&w);
  		addedge(v,u,w);
 	}
 	if(spfa(0)){
    
    
 		printf("NO\n");
 	}
 	else{
    
    
  		for(int i=1;i<=n;++i){
    
    
  			printf("%d ",dis[i]);
		  }
   	}
 	return 0;
}

//最长路求法
#include<cstring>
#include<iostream>
#include<queue>
#include<cstdio>
#include<algorithm>
using namespace std;
inline int read(){
    
    
    int a=0;char x=getchar();
    while(x<'0'||x>'9')x=getchar();
    while(x>='0'&&x<='9')a=(a<<3)+(a<<1)+x-48,x=getchar();
    return a;
}
struct edge{
    
    
 	int to,w,next;
}kkk[300005];
int head[100005],neg[100005],dis[100005],vis[100005],numedge,n,m;

void addedge(int u,int v,int w){
    
    
 	kkk[++numedge].to=v;
 	kkk[numedge].w=w;
 	kkk[numedge].next=head[u];
 	head[u]=numedge;
}

bool spfa(int s){
    
    
 	queue<int>q;
 	dis[s]=0,vis[s]=1;
 	q.push(s);
 	while(!q.empty()){
    
    
  		int u=q.front();
  		q.pop();
  		vis[u]=0;
  		for(int i=head[u];i;i=kkk[i].next){
    
    
   			int v=kkk[i].to;
   			if(dis[v]<dis[u]+kkk[i].w){
    
    
    			dis[v]=dis[u]+kkk[i].w;
    			if(!vis[v]){
    
    
     				vis[v]=1,++neg[v];
     				if(neg[v]>n) return true;
     				q.push(v);
    			}
   			}
  		}
 	}
 	return false;
}
int main()
{
    
    
 	scanf("%d%d",&n,&m);
 	for(int i=n;i>=1;--i)
  		addedge(0,i,0),dis[i]=-100000;
 	for(int i=1;i<=m;++i){
    
    
  		int w,u,v;
  		scanf("%d%d%d",&u,&v,&w);
  		addedge(u,v,-w);
	}
 	if(spfa(0)){
    
    
 		printf("NO");
 	}
 	else{
    
    
 		for(int i=1;i<=n;++i) printf("%d ",dis[i]);
   	}
 	return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_45695839/article/details/109413172
今日推荐