这道题只能骗到90 分,想法在代码注释中写着,注意:
这道题没有拿到全部分数,算法运行超时了,放到这,只是为了保存代码。日后回来再改吧。
#include <iostream>
#include <queue>
#include <cstring>
#include <algorithm>
using namespace std;
#define INF 0x7fffffff
int n,m,k,d;
struct node{
int x,y;
int bushu;
node(int x,int y,int bushu):x(x),y(y),bushu(bushu){
}
node(){
}
};
int dist[1005][1005];
int vis[1005][1005];
// 以每个分店所在点为起点,调用一次BFS , 得到 每个客户所在点离最近分店的最短路,然后乘以订餐数;
queue<node> s;
// 编完再考虑实际的位置和题目中的位置有所不同。
void bfs(int startx,int starty){
dist[startx][starty]=0;
int dx[]={0,-1,0,1},dy[]={1,0,-1,0};
s.push(node(startx,starty,0));
node no;
int x,y,bushu;
while(!s.empty()){
no=s.front();
s.pop();
x=no.x;
y=no.y;
bushu=no.bushu;
for(int i=0;i<4;i++){
int nx=x+dx[i],ny=y+dy[i],nbushu=bushu+1;
//考虑不能走不通,用什么来记录,就用 vis 好了。
if(nx>=0&&nx<n&&ny>=0&&ny<n&& !vis[nx][ny]){
dist[nx][ny]=nbushu;
vis[nx][ny]=1;
s.push(node(nx,ny,nbushu));
}
}
}
}
pair<int,int> fendian[1000005],cv[1000005];
struct kehu{
int x,y,amount,mindis;
};
kehu ke[1000005];
int main(){
cin>>n>>m>>k>>d;
int x,y;
for(int i=0;i<m;i++){
cin>>x>>y;
fendian[i].first=n-y;
fendian[i].second=x-1;
}
for(int i=0;i<k;i++){
cin>>x>>y;
ke[i].x=n-y;ke[i].y=x-1;
cin>>ke[i].amount;
ke[i].mindis=INF;
}
for(int i=0;i<d;i++){
cin>>x>>y;
cv[i].first=n-y;
cv[i].second=x-1;
}
for(int i=0;i<m;i++){
int u=fendian[i].first,v=fendian[i].second;
memset(vis,0,sizeof(vis));
//memset(dist,0,sizeof(dist));
for(int i=0;i<n;i++)
for(int j=0;j<n;j++){
dist[i][j]=INF;
}
for(int j=0;j<d;j++){
vis[cv[j].first][cv[j].second]=1;
}
bfs(u,v);
for(int j=0;j<k;j++){
ke[j].mindis=min(ke[j].mindis,dist[ke[j].x][ke[j].y]);
}
}
// 输出结果
long long ans=0;
for(int j=0;j<k;j++){
int amount=ke[j].amount;
ans+=ke[j].mindis*amount;
}
cout<<ans<<endl;
return 0;
}