【图论专项赛】外星人入侵

题目描述

时间限制: 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);
    }
}