Useful Edges
题目大意:一个无向有全图中,给出q个u,v点以及权值l,定义一条路径上以u和v分别为起点和终点的路径,且这条路径的边权和不超过l,那么这条路径上的所有边都是有用的边。求有用的边的个数
思路:枚举所有边,看他是否能成为有用边。
具体做法:枚举边的端点i,j,u->i->j->v由三段距离组成,枚举i,j等于得到了i->j的距离,那么只要再得到u->i的距离dis[i][j],边的另一个端点j离目标点的距离小于max{l-dis[i][u]}即可。由于数据范围n小于600且时间范围5s,因此最短距离dis可以用floyd。
#include <bits/stdc++.h>
#define LL long long
using namespace std;
const int kMax = 610;
LL dis[kMax][kMax],w[kMax][kMax];
LL maxx[kMax];
LL n,m,q;
vector<pair<int,int> >p[kMax];
int main()
{
scanf("%lld%lld",&n,&m);
for(int i = 1;i <= n;i++)
{
for(int j = 1;j <= n;j++)
{
dis[i][j] = 1e18;
}
}
for(int i = 1;i <= n;i++)
dis[i][i] = 0;
for(int i = 1;i <= m;i++)
{
LL u,v,val;
scanf("%lld%lld%lld",&u,&v,&val);
w[u][v] = w[v][u] = val;
dis[u][v] = dis[v][u] = min(val,dis[u][v]);
}
for(int k = 1;k <= n;k++)
{
for(int i = 1;i <= n;i++)
{
for(int j = 1;j <= n;j++)
{
if(dis[i][j] > dis[i][k] + dis[k][j])
dis[i][j] = dis[i][k] + dis[k][j];
}
}
}
int q;
cin>>q;
for(int i=1;i<=q;i++)
{
LL x,y,z;
scanf("%lld%lld%lld",&x,&y,&z);
p[x].push_back(make_pair(y,z));
p[y].push_back(make_pair(x,z));
}
LL res = 0;
for(int i = 1;i <= n;i++)
{
for(int x = 1;x <= n;x++)
{
maxx[x] = -1e18;
int size = p[x].size();
for(int j = 0;j < size;j++)
{
LL y = p[x][j].first,lim = p[x][j].second;
maxx[x] = max(maxx[x],lim-dis[i][y]);
}
}
for(int j = i + 1;j <= n;j++)
{
if(w[i][j] == 0)continue;
for(int k = 1;k <= n;k++)
{
if(maxx[k] >= dis[j][k] + w[i][j])
{
res++;
break;
}
}
}
}
/*for(int i = 1;i <= n;i++)
{
for(int j = 1;j <= n;j++)
cout<<w[i][j]<<" ";
cout<<'\n';
}*/
cout<<res<<'\n';
return 0;
}