【JZOJ4805】【NOIP2016提高A组模拟9.28】跟踪

题目描述

输入

输出

样例输入

4 2 1 3
1 2
2 3
3 4

样例输出

2

数据范围

解法

预处理出两个陌生人走到各个点的距离。
从石神处开始dfs，判断走到每一个点是否会被抓；
如果会，则计算答案，并给超级答案取最大值；
如果不会，继续走下去。

---

计算答案只需简单的运算，O(1)即可。

代码

#include<iostream>
#include<stdio.h>
#include<math.h>
#include<string.h>
#include<algorithm>
#define ll long long
#define ln(x,y) int(log(x)/log(y))
#define sqr(x) ((x)*(x))
using namespace std;
const char* fin="track.in";
const char* fout="track.out";
const int inf=0x7fffffff;
const int maxn=200007,maxm=maxn*2;
int n,m1,m2,m3,i,j,k,tot,ans;
int fi[maxn],la[maxm],ne[maxm];
int dis[3][maxn];
void add_line(int a,int b){
    tot++;
    ne[tot]=fi[a];
    la[tot]=b;
    fi[a]=tot;
}
void getdis(int v,int from,int deep,int id){
    int i,j,k;
    dis[id][v]=deep;
    for (k=fi[v];k;k=ne[k]) if (la[k]!=from) getdis(la[k],v,deep+1,id);
}
void dfs(int v,int from,int time){
    int i,j,k=inf;
    /*if (v!=m1) {
        if (dis[1][v]/2<=dis[0][v]){
            if (dis[1][v]%2) k=min(k,2);
            else k=0;
        }
        if (dis[2][v]/2<=dis[0][v]){
            if (dis[2][v]%2) k=min(k,2);
            else k=0;
        }
        if (k!=inf){
            ans=max(time+k,ans);
            return ;
        }
    }*/
    if (v==m1 || dis[1][v]/2+dis[1][v]%2>dis[0][v] && dis[2][v]/2+dis[2][v]%2>dis[0][v])
        for (k=fi[v];k;k=ne[k])
            if (la[k]!=from){
                dfs(la[k],v,time+3);
            }
    ans=max(min(dis[1][v]/2*3+(dis[1][v]%2?2:0),dis[2][v]/2*3+(dis[2][v]%2?2:0)),ans);
}
int main(){
    freopen(fin,"r",stdin);
    freopen(fout,"w",stdout);
    scanf("%d%d%d%d",&n,&m1,&m2,&m3);
    for (i=1;i<n;i++){
        scanf("%d%d",&j,&k);
        add_line(j,k);
        add_line(k,j);
    }
    getdis(m1,0,0,0);
    getdis(m2,0,0,1);
    getdis(m3,0,0,2);
    dfs(m1,0,-2);
    printf("%d",ans);
    return 0;
}

启发

多出数据来调整细节。