GYM 100962F Problem F. Frank Sinatra(树上莫队+分块)

题目链接

Problem F. Frank Sinatra

分析

这题和前面那个题唯一不一样的地方是，这题访问的是边上的，因此可以将边上的值算做入边顶点的值，这样就　u,v 对应的区间就是　[dfl[u]+1,dfl[v]], 这样开个桶记录访问到的数就行了.

code

#include <bits/stdc++.h>using namespace std;#define ms(x,v) (memset((x),(v),sizeof(x)))#define pb push_back#define mp make_pair#define fi first#define se secondtypedef long long LL;typedef pair<int,int > Pair;const int maxn = 1e5+10;int n,m;const int S = 320;struct Query{    int id,l,r,backet;    bool operator<(const Query & o)const{        return backet==o.backet?(backet&1?r>o.r : r <o.r) : l<o.l;    }} q[maxn];std::vector<Pair> G[maxn];int dfl[maxn],dfr[maxn],dfn[maxn<<1],val[maxn],dft=0;int cnt[maxn],vis[maxn],sum[maxn],ret[maxn];void dfs(int x) {    dfl[x] = ++dft;dfn[dft] = x;    for(auto v : G[x])        if(!dfl[v.fi])val[v.fi] =v.se, dfs(v.fi);    dfr[x] = ++dft;dfn[dft] = x;}inline void move(int node) {    if(val[node]>n)return;    if(vis[node] && --cnt[val[node]]==0)--sum[val[node]/S];    else if(!vis[node] && ++cnt[val[node]]==1)++sum[val[node]/S];    vis[node] ^=1;}inline void mo(/* arguments */) {    int l=  q[0].l,r = q[0].l-1,L,R;    for(int i=0 ;i<m ; ++i){        L = q[i].l,R = q[i].r;        while (l > L)move(dfn[--l]);        while (r < R)move(dfn[++r]);        while (l < L)move(dfn[l++]);        while (r > R)move(dfn[r--]);        int j;        for( j=0;sum[j]==S ; ++j);        for(j*=S ; cnt[j]; ++j);        ret[q[i].id]  = j;    }}int main(int argc, char const *argv[]) {    scanf("%d%d",&n,&m );    for(int i=1 ; i<n ; ++i){        int u,v,c;        scanf("%d%d%d",&u,&v,&c );        G[u].pb(mp(v,c));G[v].pb(mp(u,c));    }    dfs(1);    for(int i=0 ; i<m ; ++i){        int u,v;        scanf("%d%d",&u,&v );        if(dfl[u]> dfl[v])swap(u,v);        q[i].id = i;        q[i].l= dfl[u]+1;q[i].r = dfl[v];q[i].backet = q[i].l/S;    }    sort(q,q+m);    mo();    for(int i=0 ; i<m ; ++i)printf("%d\n",ret[i]);    return 0;}