题目描述
时间限制: 1 Sec 内存限制: 64 MB
外星人入侵地球。可怕的吃人外星人正在全国各地依次序建立它们的基地。
全国共有N(1≤ N ≤10,000)座城市,城市编号1~N。城市之间有M(0≤ M ≤100,000)条双向道路相连。外星人计划建立A(0≤A≤N)个基地。
你只有在距离当前所有外星人基地至少K(1≤K≤100)单位长度的城市才能得到安全。
输入
第1行:4个整数N, M, A, K
接下来M行,每行3个整数T1, T2(1≤T1
输出
共A行,第i行1个整数,表示当外星人建好第i个基地后,距离当前所有基地B1,B2,…,Bi至少K长度的城市的数量。
样例输入
7 6 3 3
1 2 1
1 3 1
2 5 1
3 6 1
1 4 1
4 7 2
2
1
4
样例输出
2
1
0
思路
我们可以先建一个图,然后用外星人基地不断更新。
为了不超时,我们可以使用记忆化搜索。
用vis与t数组来记忆化。
vis:当前城市是否安全。
t:从最近的基地到达这座城市后还剩的最大拓展量。
代码
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
const int MAXN=10005;
int n,m,k,a,x,y,z,vis[MAXN],t[MAXN];
vector<int>T[MAXN],Q[MAXN];
void dfs(int p,int g)
{
for(int i=0;i<Q[p].size();i++)
{
int v=Q[p][i];
if(t[v]<g-T[p][i]&&g-T[p][i]>0)
{
t[v]=g-T[p][i];
vis[v]=1;
dfs(v,t[v]);
}
}
}
int main()
{
scanf("%d%d%d%d",&n,&m,&a,&k);
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&x,&y,&z);
Q[x].push_back(y);
Q[y].push_back(x);
T[x].push_back(z);
T[y].push_back(z);
}
for(int i=1;i<=a;i++)
{
scanf("%d",&x);
vis[x]=1;t[x]=k;
dfs(x,k);int you=n;
for(int j=1;j<=n;j++)
if(vis[j])you--;
printf("%d\n",you);
}
}