参考链接:
https://blog.csdn.net/weixin_43581819/article/details/104121800?utm_medium=distribute.pc_relevant.none-task-blog-baidujs-4
输入样例 1:
9 12
0 1 6
0 2 4
0 3 5
1 4 1
2 4 1
3 5 2
5 4 0
4 6 9
4 7 7
5 7 4
6 8 2
7 8 4
输出样例 1:
18
输入样例 2:
4 5
0 1 1
0 2 2
2 1 3
1 3 4
3 2 5
输出样例 2:
Impossible
代码
#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <cstring>
#include <queue>
using namespace std;
int dis[100][100];
int ru[100]={0};
int n,m;
int f[100]={0};
int ans=0;
int cnt=0;
void topological_sort(){
while(true){
int flag=0;
for(int i=0;i<n;i++){
if(ru[i]==0){
//cout<<i<<endl;
flag=1;
cnt++;
ru[i]--;
for(int j=0;j<n;j++){
if(dis[i][j]!=-1){
f[j]=max(f[i]+dis[i][j],f[j]);
ans=max(ans,f[j]);
ru[j]--;
}
}
}
}
if(flag==0) break;
}
//cout<<cnt<<endl;
if(cnt==n) {
cout<<ans<<endl;
}
else
cout<<"Impossible"<<endl;
}
int main(){
cin>>n;
cin>>m;
memset(dis,-1,sizeof(dis));
// for(int i=0;i<n;i++){
// for(int j=0;j<n;j++){
// dis[i][j]=-1;
// }
// }
//memset(dis,-1,sizeof(dis));
for(int i=0;i<m;i++){
int x,y,v;
cin>>x;
cin>>y;
cin>>v;
dis[x][y]=v;
ru[y]++;
}
topological_sort();
}
优化代码:
#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <cstring>
#include <queue>
using namespace std;
int dis[100][100];
int ru[100]={0};
int n,m;
int f[100]={0};
int ans=0;
int cnt=0;
void topological_sort(){
while(true){
int flag=0;
for(int i=0;i<n;i++){
if(ru[i]==0){
//cout<<i<<endl;
flag=1;
cnt++;
ru[i]--;
for(int j=0;j<n;j++){
if(dis[i][j]!=-1){
f[j]=max(f[i]+dis[i][j],f[j]);
ans=max(ans,f[j]);
ru[j]--;
}
}
}
}
if(flag==0) break;
}
//cout<<cnt<<endl;
if(cnt==n) {
cout<<ans<<endl;
}
else
cout<<"Impossible"<<endl;
}
void topological_sort2(){
queue<int>q;
vector<int>edge[n];
for(int i=0;i<n;i++) //n:结点的总数
if(ru[i]==0) q.push(i); //将入度为0的点入队列
vector<int>ans2; //ans 为拓扑序列
while(!q.empty())
{
int p=q.front(); q.pop(); // 选一个入度为0的点,出队列
ans2.push_back(p);
for(int i=0;i<n;i++)
{
//int y=dis[p][i];
if(dis[p][i]!=-1){
f[i]=max(f[p]+dis[p][i],f[i]);
ans=max(ans,f[i]);
ru[i]--;
if(ru[i]==0) q.push(i);
}
}
}
if(ans2.size()==n){
cout<<ans<<endl;
// for(int i=0;i<ans2.size();i++)
// cout<<ans2[i]<<" ";
}else cout<<"No Answer!\n"; // ans中的长度与n不相等,存在环
}
int main(){
cin>>n;
cin>>m;
memset(dis,-1,sizeof(dis));
// for(int i=0;i<n;i++){
// for(int j=0;j<n;j++){
// dis[i][j]=-1;
// }
// }
//memset(dis,-1,sizeof(dis));
for(int i=0;i<m;i++){
int x,y,v;
cin>>x;
cin>>y;
cin>>v;
dis[x][y]=v;
ru[y]++;
}
//topological_sort();
topological_sort2();
}