时间限制求助
查看原帖
时间限制求助
1085788
Feynman5210楼主2024/8/18 13:56

分层图+dijkstra,洛谷上不吸氧55TLE,吸氧AC;UOJ上TLE最后一个点。

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
using namespace std;
const int maxn=30300,maxm=110;
struct edge{
	int ya,yb,q;
};
struct edgge{
    unsigned short ya,yb,q;
};vector<edgge>a[maxn][maxm];
inline bool operator<(edge c,edge d){
	return c.q>d.q;
}
int n,m,b,p,kk,s,t,zdl[maxn][maxm];bool used[maxn][maxm];
priority_queue<edge>q;
inline void jb(){
	for(int i=1;i<=kk;i++)for(int j=0;j<n;j++){
		if(j>=i)a[j][i].push_back({j-i,i,1});
		if(i+j<n)a[j][i].push_back({j+i,i,1});
		a[j][i].push_back({j,0,0});
	}
}
inline void jianbian(int b,int p){
	if(p<=kk)a[b][0].push_back({b,p,0});
	else{
		for(int i=b;i>=p;i-=p)a[b][0].push_back({i-p,0,(b-i+p)/p});
		for(int i=b;i+p<n;i+=p)a[b][0].push_back({i+p,0,(i+p-b)/p});
	}
}
inline void input(){
	ios::sync_with_stdio(0);
	cin.tie(0);cout.tie(0);
	cin>>n>>m;kk=(int)sqrt(n/3);
	for(int i=0;i<m;i++){
		cin>>b>>p;
		if(i==0)s=b;if(i==1)t=b;
		jianbian(b,p);
	}
	jb();
}
inline int dijkstra(){
	memset(zdl,63,sizeof(zdl));
	zdl[s][0]=0,q.push({s,0,0});
	while(!q.empty()){
		edge r=q.top();q.pop();if(r.ya==t)return r.q;
		if(used[r.ya][r.yb])continue;used[r.ya][r.yb]=1;
		for(int i=0;i<a[r.ya][r.yb].size();i++)
			if(r.q+a[r.ya][r.yb][i].q<zdl[a[r.ya][r.yb][i].ya][a[r.ya][r.yb][i].yb]){
				zdl[a[r.ya][r.yb][i].ya][a[r.ya][r.yb][i].yb]=r.q+a[r.ya][r.yb][i].q;
				q.push({a[r.ya][r.yb][i].ya,a[r.ya][r.yb][i].yb,zdl[a[r.ya][r.yb][i].ya][a[r.ya][r.yb][i].yb]});
			}
	}
	return -1;
}
int main(){
	input();cout<<dijkstra();return 0;
}
2024/8/18 13:56
加载中...