E - Souvenir(图论典型例题)

思路:对于有很多询问的题,一般都是先初始化。我们求出每个点到其他点的最短路径以及相同路径下最大的价值和即可。

代码:

cpp 复制代码
#include <bits/stdc++.h>
#define pb push_back
#define a first
#define b second
using namespace std;
typedef long long int ll;
typedef pair <int,int> P;
typedef unsigned long long ull;

char str[1005];
int main(){
	int n;
	scanf("%d",&n);
	vector <ll> A(n);
	for(int i=0;i<n;i++) scanf("%lld",&A[i]);
	vector < vector<int> > mp(n);
	for(int i=0;i<n;i++){
		scanf("%s",&str);
		mp[i].resize(n);
		for(int j=0;j<n;j++){
			mp[i][j]=str[j]=='Y';
		}
		mp[i][i]=1;
	}
	ll inf=1000000000000000;
	vector < vector<ll> > ans(n);
	vector < vector<int> > vd(n);
	for(int i=0;i<n;i++){
		ans[i].resize(n);
		for(int j=0;j<n;j++) ans[i][j]=-inf;
		vector <int> dist(n,n+1);
		queue <int> que;
		dist[i]=0;
		ans[i][i]=A[i];
		que.push(i);
		while(!que.empty()){
			int v=que.front();que.pop();
			for(int j=0;j<n;j++){
				if(mp[v][j]){
					if(dist[j]==n+1){
						dist[j]=dist[v]+1;
						que.push(j);
					}
					if(dist[j]==dist[v]+1){
						ans[i][j]=max(ans[i][j],ans[i][v]+A[j]);
					}
				}
			}
		}
		vd[i]=dist;
	}
	int Q;
	scanf("%d",&Q);
	for(int i=0;i<Q;i++){
		int u,v;
		scanf("%d%d",&u,&v);u--,v--;
		if(ans[u][v]==-inf) puts("Impossible");
		else printf("%d %lld\n",vd[u][v],ans[u][v]);
	}
}
相关推荐
火星机器人life1 小时前
基于ceres优化的3d激光雷达开源算法
算法·3d
虽千万人 吾往矣1 小时前
golang LeetCode 热题 100(动态规划)-更新中
算法·leetcode·动态规划
arnold662 小时前
华为OD E卷(100分)34-转盘寿司
算法·华为od
ZZTC2 小时前
Floyd算法及其扩展应用
算法
lshzdq3 小时前
【机器人】机械臂轨迹和转矩控制对比
人工智能·算法·机器人
2401_858286113 小时前
115.【C语言】数据结构之排序(希尔排序)
c语言·开发语言·数据结构·算法·排序算法
猫猫的小茶馆3 小时前
【数据结构】数据结构整体大纲
linux·数据结构·算法·ubuntu·嵌入式软件
u0107735144 小时前
【字符串】-Lc5-最长回文子串(中心扩展法)
java·算法
帅逼码农4 小时前
K-均值聚类算法
算法·均值算法·聚类
姚先生974 小时前
LeetCode 209. 长度最小的子数组 (C++实现)
c++·算法·leetcode