G房间迷宫 最短路找因子

#include<bits/stdc++.h>
using namespace std;
#define mp make_pair
#define pb push_back
#define mem(a,b) memset(a,b,sizeof a);
typedef long long ll;
typedef pair<int,int> pii;
const int maxn=2e5+5;
const ll inf=1e18;

ll dis[maxn];
int n,m;
int a[maxn],b[maxn],c[maxn],d[maxn];
struct edge {
	int from, to, dist;
	edge(int u, int v, int d) : from(u), to(v), dist(d) {}
};
vector<int> g[maxn];
vector<edge> edges;
bool vis[maxn];
void init() {
	for(int i=0; i<=n; i++) {
		g[i].clear();		//邻接表
	}
	edges.clear();
	for(int i=0; i<=n; i++) {
		dis[i] = inf;
	}
	mem(vis,0);
}

void dijkstra(int s) {
	priority_queue<pii,vector<pii>,greater<pii> > Q;
	dis[s]=0;
	Q.push(pii(0,s));
	while(!Q.empty()) {
		pii x=Q.top();
		Q.pop();
		int u=x.second;
		if(vis[u])
			continue;
		vis[u]=true;
		for(int i=0; i<g[u].size(); i++) {
			edge &e = edges[g[u][i]];	//g[u][i] 边的位置 边的结构体
			if(vis[e.to])
				continue;
			if(dis[e.to] > dis[u] + e.dist) {
				dis[e.to] = dis[u] + e.dist;
				Q.push(pii(dis[e.to],e.to));	//花费 终点
			}
		}
	}

}
int main() {
	cin>>n;
	init();

	for(int i=1; i<=n; i++) {
		cin>>a[i]>>b[i]>>c[i]>>d[i];
	}

	for(int i=1; i <= n; i++) {	//建边后跑最短路 nlogn 
		edges.pb(edge(i,d[i],a[d[i]]));		//直达 边的结构体
		g[i].pb(edges.size()-1);	//第几条边的位置
		for(int j = 1; j*j <= c[i]; j++) {
			if(c[i] % j == 0) {
				if(i+j <= n) {
					edges.pb(edge(i,i+j,a[i+j]+b[i]));
					g[i].pb(edges.size()-1);
				}
				if(c[i]/j!=j && c[i]/j+i<=n) {
					edges.pb(edge(i,i+c[i]/j,a[i+c[i]/j]+b[i]));	//另一个因子 过路费a + b[i]开箱费
					g[i].pb(edges.size()-1);
				}
			}
		}
	}
	dijkstra(1);
	cout<<dis[n]+a[1]<<endl; 
	return 0;
}