上午把P1629 邮递员送信写了。在b站上看了最小生成路的两种方法的动态讲解。
P1629 邮递员送信
题目描述
有一个邮递员要送东西,邮局在节点 11。他总共要送 n-1n−1 样东西,其目的地分别是节点 22 到节点 nn。由于这个城市的交通比较繁忙,因此所有的道路都是单行的,共有 mm 条道路。这个邮递员每次只能带一样东西,并且运送每件物品过后必须返回邮局。求送完这 n-1n−1 样东西并且最终回到邮局最少需要的时间。
输入格式
第一行包括两个整数,nn 和 mm,表示城市的节点数量和道路数量。
第二行到第 (m+1)(m+1) 行,每行三个整数,u,v,wu,v,w,表示从 uu 到 vv 有一条通过时间为 ww 的道路。
输出格式
输出仅一行,包含一个整数,为最少需要的时间。
输入输出样例
输入 #1复制
5 10 2 3 5 1 5 5 3 5 6 1 2 8 1 3 8 5 3 4 4 1 8 4 5 3 3 5 6 5 4 2
输出 #1复制
83
说明/提示
对于 30\%30% 的数据,1 \leq n \leq 2001≤n≤200。
对于 100\%100% 的数据,1 \leq n \leq 10^31≤n≤103,1 \leq m \leq 10^51≤m≤105,1\leq u,v \leq n1≤u,v≤n,1 \leq w \leq 10^41≤w≤104,输入保证任意两点都能互相到达。
思路:邮递员每次带一份邮件去相应的点,这是用迪杰斯特拉找最小,但是回来时这条路不一定是最短的。所以对它反向使用dij。
代码实现:
#include<bits/stdc++.h>
using namespace std;
int n,m;
int dvis[1000000];
int fvis[1000000];
int dlow[1000000];
int flow[1000000];
int maxm=1e9;
struct edge{
int to;
int v;
int next;
}dmp[100000],fmp[100000];
int dhead[1000000];
int fhead[1000000];
struct node
{
int dis;
int v;
bool operator <(const node &x) const
{
return x.v<v;
}
};
int cnt=0;
void add(int a,int b,int v)
{
cnt++;
dmp[cnt].to=b;
dmp[cnt].v=v;
dmp[cnt].next=dhead[a];
dhead[a]=cnt;
fmp[cnt].to=a;
fmp[cnt].v=v;
fmp[cnt].next=fhead[b];
fhead[b]=cnt;
}
void dij()
{
priority_queue<node>q;
for(int i=1;i<=n;i++)
dlow[i]=maxm;
dlow[1]=0;
q.push((node){1,0});
while(q.size()!=0)
{
node x=q.top();
q.pop();
int dis=x.dis;
if(dvis[dis]==1)
continue;
dvis[dis]=1;
for(int i=dhead[dis];i!=0;i=dmp[i].next)
{
int t=dmp[i].to;
if(dlow[t]>dlow[dis]+dmp[i].v)
{
dlow[t]=dlow[dis]+dmp[i].v;
q.push((node){t,dlow[t]});
}
}
}
}
void fdij()
{
for(int i=1;i<=n;i++)
flow[i]=maxm;
flow[1]=0;
priority_queue<node>q;
q.push((node){1,0});
while(q.size()!=0)
{
node x=q.top();
q.pop();
int dis=x.dis;
if(fvis[dis]==1)
continue;
fvis[dis]=1;
for(int i=fhead[dis];i!=0;i=fmp[i].next)
{
int t=fmp[i].to;
if(flow[t]>flow[dis]+fmp[i].v)
{
flow[t]=flow[dis]+fmp[i].v;
q.push((node){t,flow[t]});
}
}
}
}
int main()
{
cin>>n>>m;
int a,b,v;
for(int i=1;i<=m;i++)
{
cin>>a>>b>>v;
add(a,b,v);
}
dij();
fdij();
int sum=0;
for(int i=2;i<=n;i++)
{
sum=sum+dlow[i]+flow[i];
}
cout<<sum;
}
这里dlow,dhead表示来的时候,flow,fhead表示回来的时候。
下午学习了最小生成树Kruskal 写法。这个算法就是先给路劲从小到大排序,然后一个取,在保证没有环的情况下寻找。如果形成了环,这条路就不用连接。知道所有点都经过后结束。
晚上完成了一个模板题
P3366 【模板】最小生成树
题目描述
如题,给出一个无向图,求出最小生成树,如果该图不连通,则输出 orz
。
输入格式
第一行包含两个整数 N,MN,M,表示该图共有 NN 个结点和 MM 条无向边。
接下来 MM 行每行包含三个整数 X_i,Y_i,Z_iXi,Yi,Zi,表示有一条长度为 Z_iZi 的无向边连接结点 X_i,Y_iXi,Yi。
输出格式
如果该图连通,则输出一个整数表示最小生成树的各边的长度之和。如果该图不连通则输出 orz
。
输入输出样例
输入 #1复制
4 5 1 2 2 1 3 2 1 4 3 2 3 4 3 4 3
输出 #1复制
7
说明/提示
数据规模:
对于 20\%20% 的数据,N\le 5N≤5,M\le 20M≤20。
对于 40\%40% 的数据,N\le 50N≤50,M\le 2500M≤2500。
对于 70\%70% 的数据,N\le 500N≤500,M\le 10^4M≤104。
对于 100\%100% 的数据:1\le N\le 50001≤N≤5000,1\le M\le 2\times 10^51≤M≤2×105,1\le Z_i \le 10^41≤Zi≤104。
样例解释:
所以最小生成树的总边权为 2+2+3=72+2+3=7。
思路:运用Kruskal 算法。
代码实现:
#include<bits/stdc++.h>
using namespace std;
int n,m;
struct node{
int a;//起点
int b;//终点
int v;//距离
}mp[200010];
int father[5001];
bool compare(node a,node b)
{
return a.v<b.v;
}
int find_root(int n)
{
if(father[n]==n)
return father[n];
else
return find_root(father[n]);
}
void k()
{
int flag=0;
sort(mp,mp+m,compare);
int cnt=0;
int sum=0;
for(int i=0;i<m;i++)
{
int a=find_root(mp[i].a);
int b=find_root(mp[i].b);
if(a!=b)
{
sum+=mp[i].v;
father[b]=a;
cnt++;
}
if(cnt==n-1)
{
flag=1;
break;
}
}
if(flag==0)
cout<<"orz";
else if(flag==1)
cout<<sum;
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
father[i]=i;
int a,b,v;
for(int i=0;i<m;i++)
{
cin>>a>>b>>v;
mp[i].a=a;
mp[i].b=b;
mp[i].v=v;
}
k();
}
明天学习prim写法。