比赛 |
NOIP模拟赛by mzx Day1 |
评测结果 |
AWAWWWWTTT |
题目名称 |
零食店 |
最终得分 |
20 |
用户昵称 |
ciyou |
运行时间 |
3.326 s |
代码语言 |
C++ |
内存使用 |
0.29 MiB |
提交时间 |
2016-10-19 21:45:45 |
显示代码纯文本
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
using namespace std;
struct edge{
int from,to,cost;
};
struct vertix{
int dis,id;
bool operator < (const vertix& c) const{
return dis>c.dis;
}
};
vector<edge> edges;
vector<int> G[101];
priority_queue<vertix> Q;
int dis[101],vis[101],C[101];
int n,m,q;
int dijkstra(int,int,int);
edge make_edge(int,int,int);
vertix make_vertix(int,int);
void add_edge(int,int,int);
int main(){
freopen("snackstore.in","r",stdin);
freopen("snackstore.out","w",stdout);
scanf("%d%d%d",&n,&m,&q);
edges.clear();
for(int i=1;i<=n;i++) G[i].clear();
for(int i=1;i<=n;i++) scanf("%d",&C[i]);
for(int i=1;i<=m;i++){
int x,y,l;
scanf("%d%d%d",&x,&y,&l);
add_edge(x,y,l);
}
for(int i=1;i<=q;i++){
int s,c,d;
scanf("%d%d%d",&s,&c,&d);
printf("%d\n",dijkstra(s,c,d));
}
fclose(stdin);
fclose(stdout);
return 0;
}
edge make_edge(int u,int v,int w){
edge temp;
temp.from=u;
temp.to=v;
temp.cost=w;
return temp;
}
vertix make_vertix(int id,int dis){
vertix temp;
temp.dis=dis;
temp.id=id;
return temp;
}
void add_edge(int u,int v,int w){
edges.push_back(make_edge(u,v,w));
G[u].push_back(edges.size()-1);
edges.push_back(make_edge(v,u,w));
G[v].push_back(edges.size()-1);
return;
}
int dijkstra(int s,int c,int d){
int ans=0;
memset(dis,0x3f,sizeof(dis));
memset(vis,0,sizeof(vis));
dis[s]=0;
Q.push(make_vertix(s,0));
while(!Q.empty()){
int p=Q.top().id;
Q.pop();
if(vis[p]) continue;
else vis[p]=1;
for(int i=0;i<G[p].size();i++){
edge e=edges[G[p][i]];
if(dis[e.to]>dis[p]+e.cost){
dis[e.to]=dis[p]+e.cost;
if(dis[e.to]<=d) ans++;
if(dis[e.to]<=d&&C[e.to]<=c) Q.push(make_vertix(e.to,dis[e.to]));
}
}
}
return ans;
}