AtCoderC - Distribution 优先队列+BFS

Posted 2021-10-27 karshey

C
参考

求最短时间。可以用优先队列。优先队列pii（greater）是按照first从大到小来排列的。
那么每次取出的都是当前最小的。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<ll,int> pii;//第一个是时间，第二个是i 因为优先队列是根据fi来排的 
#define pb push_back
#define fi first
#define se second
#define mem(a,x) memset(a,x,sizeof(a));
#define db double 

//======================
const int N=2e5+10;
int ans[N],s[N],t[N];//s是周期，t是开始 
int n;
priority_queue<pii,vector<pii>,greater<pii>>q;
int main()
{
	mem(ans,0);
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&s[i]);
	for(int i=1;i<=n;i++) scanf("%d",&t[i]);
	for(int i=1;i<=n;i++)
	{
		q.push({t[i],i});
	}
	while(q.size())
	{
		auto t=q.top();
		q.pop();
		int x=t.fi,y=t.se;
		if(ans[y]) continue;//如果已经有了就跳过
		ans[y]=x;//如果没有就赋值，因为这已经是当前最小的了
		int temp=y+1;
		if(temp==n+1) temp=1;
		q.push({ans[y]+s[y],temp});
	}
	for(int i=1;i<=n;i++) cout<<ans[i]<<endl;
	return 0; 
}