2018TYUT暑期ACM模拟赛(3)
​​​ransaction transaction transaction HDU - 6201 ​​​
题意:商人可以选择在一个城市买书,可以选一个城市卖书。卖书-买书-两地的路费。为所获得的利润。求利润的最大值。输入n(代表n个城市)n个数每个地点书的价格。输入n-1行两地的距离。
思路。只要求出,两地的最短路径,过一遍。卖书-买书-两地距离的最短花费。求出最大值。本来求任意两地的最短路,用Floyd,由于数据量很大肯定会超时,就考虑用单源最短路,然后加上源点。写的时候一直在超时,优化了好几次spfa但是主要还是需要创建一个超级源点和超级汇点。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<vector>
#include<cmath>
const int MAXN=100000+100;//点
const int MAXM=3*100000+100;//边
const int INF=0x7f7f7f7f;
using namespace std;
int first[MAXN];//初始化为-1
bool vis[MAXN];
int cnt[MAXN]; //入队次数
int que[MAXM+10];
int dist[MAXN];
int price[MAXN];
struct Edge
{
int to,v,next;
}edge1[MAXM]; //邻接表
int tt; //初始化为
void add(int a,int b,int v) //添加边
{
edge1[tt].to=b;
edge1[tt].v=v;
edge1[tt].next=first[a];
first[a]=tt++;
}
void SPFA(int start,int n) //源点,点个数
{
int front=0,rear=0;

for(int v=0;v<=n;v++) //初始化
{
if(v==start)
{
que[rear++]=v;
vis[v]=true;
cnt[v]=1;
dist[v]=0;
}
else
{
vis[v]=false;
cnt[v]=0;
dist[v]=-INF;
}
}
while(front!=rear)
{
int u=que[front++];
if(front>=MAXN)front=0;
vis[u]=false;//循环队列
for(int i=first[u];i!=-1;i=edge1[i].next)
{
int v=edge1[i].to;
if(dist[v]<dist[u]+edge1[i].v)
{
dist[v]=dist[u]+edge1[i].v;
if(!vis[v])
{
vis[v]=true;
que[rear++]=v;
if(rear>=MAXN) rear=0; //循环队列
if(++cnt[v]>n) return ; //存在负环
}
}
}

}
return ;
}

int main()
{
int t,n;
int a,b,c;
scanf("%d",&t);
while(t--){
tt=0;
scanf("%d",&n);
// for(int i=0;i<=n;i++) dist[i]=-INF;
memset(first,-1,sizeof(first));
for(int i=1;i<=n;i++) scanf("%d",&price[i]);//cin>>price[i];//每个城市的书价
for(int i=1;i<n;i++){
scanf("%d%d%d",&a,&b,&c);
add(a,b,price[b]-price[a]-c);//直接找价值最大的路
add(b,a,price[a]-price[b]-c);
}
for(int i=1;i<=n;i++) add(0,i,0);//超级源点
int ans=0;
SPFA(0,n);
for(int j=1;j<=n;j++){
ans=max(ans,dist[j]);//遍历找出最大的值
}
printf("%d\n",ans);
//cout<<ans<<endl;
}
return 0;
}