COGS 2507 零食店

http://www.cogs.pro/cogs/problem/problem.php?pid=2507

时间限制：1 s 内存限制：256 MB

【题目描述】

成功找到了学长之后学姐感觉到有些饿，于是决定去附近的零食店给自己和学长买些零食。

焦作市的有n家零食店，由m条道路连接着这些零食店，每条道路都有自己的长度l，每家零食店都有自己的消费指数。

由于学姐是个穷B，所以去买零食的路上不能经过某些消费指数超过一定限度的店。

同时由于学姐体力有限，所以去买零食的过程中走的路程不能太长。

想来想去学姐决定去问学长买什么零食比较好，反正到最后都是学长吃╮(╯_╰)╭

在去问之前，学姐准备先做好准备，她把焦作市（所有零食店）的地图给了你，希望你能编出一个程序快速回答她从某个零食店出发，在上述限制下有多少家零食店可供她挑选。

【输入格式】

第一行三个正整数n，m，q，分别代表零食店数，道路数和询问数。

接下来一行n个正整数，第i个正整数vi代表第i家零食店的消费指数。

接下来m行，第i行三个正整数x,y,l,代表第i条道路连接编号为x和y的两个零食店，长度为l。

接下来q行第i行三个正整数s,c,d，代表第i个询问要求从s出发，所经过的零食店的消费指数不能超过c（除了起点和终点以外），且行走路程不超过d。

【输出格式】

一共q行，第i行一个整数代表在第i个询问的要求下有多少家零食店可供学姐挑选。

【样例输入】

【样例输出】

2
3

【提示】

样例中第一个询问能去编号为2/4的零食店。

第二个询问能去编号为1/3/5的零食店。

对于40%的数据，n≤10，m≤20，q=1。

对于70%的数据，m≤500，q≤10000。

对于100%的数据，n≤100，m≤10000，q≤1000000，vi,c,d≤10^9，1≤x,y,s≤n，l≤10^6。

利用floyd 第一重循环k，所有中间点的编号都小于k

本体对消费指数的限制恰好是中间点

所以讲点按消费指数由小到大排序

f[k][i][j] 从i到j，中间经过的点的消费指数<=k的消费指数的最短路

当消费指数限制为v时，二分找到第一个消费指数<=v的点a，

从点u出发，就是在f[a][u][]里找最短路<=某个数的点的个数

排序之后二分即可

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int f[101][101][101],dy[101],a[101];
struct node
{
    int cost,id;
}e[101];
void read(int &x)
{
    x=0; char c=getchar();
    while(!isdigit(c)) c=getchar();
    while(isdigit(c)) { x=x*10+c-'0'; c=getchar(); }
}
bool cmp(node p,node q)
{
    return p.cost<q.cost;
}
int main()
{
    freopen("snackstore.in","r",stdin);
    freopen("snackstore.out","w",stdout);
    int n,m,q;
    read(n); read(m); read(q);
    for(int i=1;i<=n;i++) read(a[i]),e[i].cost=a[i],e[i].id=i;
    sort(a+1,a+n+1);
    sort(e+1,e+n+1,cmp);
    for(int i=1;i<=n;i++) dy[e[i].id]=i;
    int u,v,w;
    for(int k=0;k<=n;k++)
         for(int i=1;i<=n;i++)
             for(int j=1;j<=n;j++)
                 f[k][i][j]=1e9+1;
    while(m--)
    {
        scanf("%d%d%d",&u,&v,&w);
        u=dy[u]; v=dy[v];
        f[0][u][v]=f[0][v][u]=min(w,f[0][u][v]);
    }
    for(int i=0;i<=n;i++)
        for(int j=1;j<=n;j++)
            f[i][j][j]=0;
    for(int k=1;k<=n;k++)
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                   f[k][i][j]=min(f[k-1][i][j],f[k-1][i][k]+f[k-1][k][j]);
    for(int i=0;i<=n;i++)
        for(int j=1;j<=n;j++)
              sort(f[i][j]+1,f[i][j]+n+1);
    while(q--)
    {
        scanf("%d%d%d",&u,&v,&w);
        v=upper_bound(a+1,a+n+1,v)-a-1;
        u=dy[u];
        w=upper_bound(f[v][u]+1,f[v][u]+n+1,w)-f[v][u]-1;
        printf("%d
",w-1);
    }
}