[CF1483D]Useful Edges

Useful Edges

题解

首先我们考虑如何判断一条边合法，我们定义$u,v$之间的最短路为$di{s}_{u,v}$,询问三元组的长度为$as{k}_{u,v}$，记边为$(x,y)$，询问为$(u,v)$
那么对于边$i$,条件是
$$\exists j,di{s}_{u_j,x_i}+pai{d}_{x_i,y_i}+di{s}_{y_i,v_j}⩽as{k}_{u_j,v_j}$$
考虑移项，可得
$$\exists j,di{s}_{u_j,x_i}+pai{d}_{x_i,y_i}⩽as{k}_{u_j,v_j}-di{s}_{y_i,v_j}$$
我们可以枚举$u$和$y$，这样前面半截就只与$x$相关，后面半截就只与$v$相关。
那么，对于每个$(u,y)$，由于是存在，我们可以找到最大的$as{k}_{u,v}-di{s}_{y,v}$，此后再枚举$x$,如果有$di{s}_{u,x}+pai{d}_{x,y}⩽(as{k}_{u,v}-di{s}_{y,v}{)}_{\max }$，则该边$(x,y)$就是合法的。
我们只需要枚举$(u,y)$，算出该数对使得那些$x$合法，最后统计合法的边数即可。
至于$dis$数组，可以先通过Floyed求出来。

时间复杂度$O\left(n^3\right)$

源码

#include<bits/stdc++.h>
using namespace std;
#define MAXN 300005
#define lowbit(x) (x&-x)
#define reg register
#define mp make_pair
#define fir first
#define sec second
typedef long long LL;
typedef unsigned long long uLL;
const int INF=0x3f3f3f3f;
const LL mo=1e9+7;
const LL inv2=5e8+4;
const double Pi=acos(-1.0);
typedef pair<int,int> pii;
const double PI=acos(-1.0);
template<typename _T>
_T Fabs(_T x){return x<0?-x:x;}
template<typename _T>
void read(_T &x){
	_T f=1;x=0;char s=getchar();
	while(s>'9'||s<'0'){if(s=='-')f=-1;s=getchar();}
	while('0'<=s&&s<='9'){x=(x<<3)+(x<<1)+(s^48);s=getchar();}
	x*=f;
}
template<typename _T>
void print(_T x){if(x<0){x=(~x)+1;putchar('-');}if(x>9)print(x/10);putchar(x%10+'0');}
int n,m,q,paid[605][605],dis[605][605],ask[605][605],ans;bool vis[605][605];
struct edge{int u,v;}e[MAXN];
signed main(){
	read(n);read(m);memset(paid,0x3f,sizeof(paid));
	memset(dis,0x3f,sizeof(dis));
	for(int i=1;i<=m;i++){
		int u,v,w;read(u);read(v);read(w);e[i]=(edge){u,v};
		dis[u][v]=dis[v][u]=paid[u][v]=paid[v][u]=w;
	}
	for(int i=1;i<=n;i++)paid[i][i]=dis[i][i]=0;
	for(int k=1;k<=n;k++)for(int i=1;i<=n;i++)for(int j=1;j<=n;j++)
		if(i!=j&&i!=k&&j!=k)dis[i][j]=min(dis[i][k]+dis[k][j],dis[i][j]);
	//for(int i=1;i<=n;i++)for(int j=1;j<=n;j++)printf("dis%d %d:%d\n",i,j,dis[i][j]);
	read(q);
	for(int i=1,u,v,w;i<=q;i++)read(u),read(v),read(w),ask[u][v]=ask[v][u]=w;
	for(int u=1;u<=n;u++)
		for(int y=1;y<=n;y++){
			int maxx=0;for(int v=1;v<=n;v++)maxx=max(maxx,ask[u][v]-dis[y][v]);
			for(int x=1;x<=n;x++)if(dis[u][x]+paid[x][y]<=maxx)vis[x][y]=1;
		}
	for(int i=1;i<=m;i++)if(vis[e[i].u][e[i].v]||vis[e[i].v][e[i].u])ans++;
	printf("%d\n",ans);
	return 0;
}

谢谢！！！