Codeforces Round #495 (Div. 2)E. Sonya and Ice Cream

在树上找k个点，使树上每个点距能够找到的点距离最短。首先这k个点肯定是在直径上，其次怎么在直径上找k个点，我们可以用滑动窗口，学滑动窗口就要回单调队列，这玩意自己以前一直不会，最近才写题的时候遇到的。哎。首先用单调队列判断队尾元素是否该继续留在里面，如果不留在里面，就弹出来，然后把当前元素塞进去，然后判断队首元素是否超出滑动窗口的长度限制，如果超出，就弹出队首元素。代码中的dfs是用来找当前这个点距离最长的且没有被访问过的点的距离

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
const int INF=0x3f3f3f3f;
vector<pair<int,int>>E[maxn];
int len[maxn],father[maxn];
bool vis[maxn];

void dfs(int u,int fa,int &mx) {
    father[u]=fa;mx=max(mx,len[u]);
    for(auto it:E[u]) {
        int v=it.first,w=it.second;
        if(v==fa||vis[v]) continue;
        len[v]=len[u]+w;
        dfs(v,u,mx);
    }
}

int main()
{
    ios::sync_with_stdio(0);cin.tie(0);
    int n,k;cin>>n>>k;
    for(int i=1;i<n;i++) {
        int x,y,z;cin>>x>>y>>z;
        E[x].push_back({y,z});
        E[y].push_back({x,z});
    }
    int du=1,mx=0;
    dfs(du,len[du]=0,mx);
    for(int i=1;i<=n;i++) {
        if(len[i]>len[du]) du=i;
    }
    int dv=1;
    dfs(du,len[du]=0,mx);
    for(int i=1;i<=n;i++) {
        if(len[i]>len[dv]) dv=i;
    }
    vector<int>point,dis,dep;
    while(dv) {
        point.push_back(dv);
        dis.push_back(len[dv]);
        vis[dv]=true;
        dv=father[dv];
    }
    reverse(point.begin(),point.end());
    reverse(dis.begin(),dis.end());
    int d=point.size();
    k=min(k,d);
    int res=INF;
    deque<pair<int,int>>Q;
    for(int i=0;i<d;i++) {
        len[point[i]]=mx=0;
        dfs(point[i],0,mx);
        dep.push_back(mx);
    }
    for(int i=0,j=0;i+k<=d;i++) {
        while(j<i+k) {
            while(!Q.empty()&&Q.back().first<dep[j]) Q.pop_back();
            Q.push_back({dep[j],j});
            j++;
        }
        while(Q.front().second<i) Q.pop_front();
        res=min(res,max({dis[i],dis.back()-dis[i+k-1],Q.front().first}));
    }
    cout<<res<<endl;
    return 0;
}