C. Bulmart codeforces 730C

https://codeforces.com/problemset/problem/730/C

题目大意：有n个城市互相由m条无向边连接，走每条边耗时1单位时间，其中w个点为超市的位置，每个超市有k个商品，单价p元，现在有q次询问，每次问位于第g号点的顾客要用不超过a元获得r个商品，最短的时间是多少

1<=n,m<=5000;1<=w<=n;1<=k,p<=2e5;1<=r,a<=1e9

思路：因为每个超市可以任意选择数量，所以我们肯定优先选取单价最便宜的那个，那么我们可以二分答案，先dfs找出顾客所在位置到所有超市的最短距离，然后再判断在二分枚举的距离内的超市按照价格从小到大的顺序拿，直到拿够商品然后判断实际价格有没有超过预算，没超过，即可以把预设距离缩短试试，如果商品那不够，或者超出预算，就要加大预设距离

#include<iostream>
#include<cstdio>
#include<queue>
#include<map>
#include<algorithm>
using namespace std;
const int N = 5005;
typedef long long ll;
struct Edge
{
	int u, v, w, next;
}e[N * N];
int head[N];
int cnt = 0;
void addedge(int u, int v, int w)
{
	e[++cnt].u = u;
	e[cnt].v = v;
	e[cnt].w = w;
	e[cnt].next = head[u];
	head[u] = cnt;
}//链式前向星存图
int n, m;
struct store
{
	int id;//商店的是几号点
	ll num, pri;//商品数量和单价
}st[N];
int dis[N];
void init()
{
	for (int i = 1; i <= n; i++)
	{
		head[i] = -1;
		st[i].pri = 1e9 + 1;
	}
}
bool vis[N];
void bfs(int s)
{//算出从s点开始的到所有点的最短距离
	queue<int>q;
	q.push(s);
	for (int i = 1; i <= n; i++)
	{
		vis[i] = 0;
		dis[i] = 5001;//将所有点的距离初始化为极大值，这样s点到没有联通的点的距离就是无穷大
	}
	vis[s] = 1;
	dis[s] = 0;
	while (!q.empty())
	{
		int u = q.front();
		q.pop();
		for (int i = head[u]; ~i; i = e[i].next)
		{
			int v = e[i].v;
			if (vis[v])
				continue;
			vis[v] = 1;
			dis[v] = dis[u] + 1;//相邻点距离+1
			q.push(v);
		}
	}
}
int w;
bool cmp(store a, store b)
{//按单价从小到大排序
	return a.pri < b.pri;
}
ll numn, pril;
bool check(int x)
{
	ll cnum = 0, cpri = 0;//当前已取的数量和总价
	for (int i = 1; i <= w; i++)
	{
		if (dis[st[i].id] > x)
			continue;//到这个商店距离超过预设距离
		if (st[i].num > numn - cnum)
		{//当前商店拥有的商品数大于剩余需要拿的数量
			cpri += st[i].pri * (numn - cnum);
			cnum += numn - cnum;
			break;
		}
		cnum += st[i].num;//取完这个商店的物品
		cpri += st[i].num * st[i].pri;//维护总价
	}
	return cnum == numn && cpri <= pril;//判断物品数量是否拿够，总价是否小于限制价格
}
int main()
{	
	cin >> n >> m;
	init();
	for (int i = 1; i <= m; i++)
	{
		int u, v;
		scanf("%d%d", &u, &v);
		addedge(u, v, 1);
		addedge(v, u, 1);//存双向边，边权为1
	}
	scanf("%d", &w);
	for (int i = 1; i <= w; i++)
	{
		scanf("%d%lld%lld", &st[i].id, &st[i].num, &st[i].pri);
	}
	sort(st + 1, st + 1 + w,cmp);
	int q;
	cin >> q;
	for (int i = 1; i <= q; i++)
	{
		int s;
		scanf("%d%lld%lld", &s, &numn, &pril);
		bfs(s);
		int l = 0, r = 5000, mid;//最短时间0，最长等于边数5000
		int ans = 5001;//初始化答案为极大值
		while (l <= r)
		{
			mid = (l + r) >> 1;
			if (check(mid))
			{//当前距离内能满足，试试减小距离
				ans = mid;
				r = mid - 1;
			}
			else
			{//当前距离满足不了，增大距离
				l = mid + 1;
			}
		}
		if (ans == 5001)
		{//如果答案仍为极大值，说明没有可行方案
			printf("-1\n");
			continue;
		}
		printf("%d\n", ans);
	}
	return 0;
}