hdu 4358 深搜时间戳+线段树更新

其实这道题真的告诉了我一个道理，那就是仔细看题，题目中明确说出了1就是所有子树的根结点。

我们可以想一想，现在每个结点代表着一段区间的询问，所以我们必须利用深搜的时间戳来找出所有结点所代表的区间。

前向星见图自然不可少。

更新区间的话，我们必须用离线处理，[i,j]这个区间代表着第i个数在这个区内出现k次的数的个数。

#include<algorithm>
#include<iostream>
#include<string.h>
#include<cstdio>
#include<vector>
#include<map>
#include<stdlib.h>
using std::sort;
using std::vector;
using std::map;
const int N = 100111;
map<int,int> q;
vector<int> p[N];
struct Edge
{
       int e,next;
}edge[N<<1];

int head[N],weight[N],visit[N],Ti;
struct Query
{
     int l,r,id;
     bool operator <(const Query & tmp)const
     {
         return r<tmp.r;
     }
}que[N];
struct node
{
     int l,r;
}No[N];
int cover[N<<2],num[N],ans[N];
void dfs(int u)
{
     No[u].l=Ti;
     num[Ti]=weight[u];
     for(int i=head[u];i!=-1;i=edge[i].next)
     {
         if(visit[edge[i].e]==0)
         {
             visit[u]=1;
             Ti++;
             dfs(edge[i].e);
         }
     }
     No[u].r=Ti;
}
void PushDown(int t)
{
     if(cover[t])
     {
         int t1=t<<1,t2=t1|1;
         cover[t1]+=cover[t];
         cover[t2]+=cover[t];
         cover[t]=0;
     }
     return ;
}
void update(int t,int l,int r,int L,int R,int val)
{
     if(L<=l&&r<=R)
     {
         cover[t]+=val;
         return ;
     }
     PushDown(t);
     int m=(r+l)>>1;
     if(L<=m)update(t<<1,l,m,L,R,val);
     if(R>m)update(t<<1|1,m+1,r,L,R,val);
}
int query(int t,int l,int r,int i)
{
     if(l==r)return cover[t];
     PushDown(t);
     int m=(l+r)>>1;
     if(i<=m)return query(t<<1,l,m,i);
     else return query(t<<1|1,m+1,r,i);
}
int main()
{
    int t;
    int n,k,u,v,Q,T=0;
    scanf("%d",&t);
    while(t--)
    {
          scanf("%d%d",&n,&k);
          q.clear();
          int pos=1;
          for(int i=1;i<=n;i++)
          {
              scanf("%d",&weight[i]);
              if(q[weight[i]]==0)q[weight[i]]=pos++;
              weight[i]=q[weight[i]];
              head[i]=-1,visit[i]=0;
          }
          int tope=0;
          for(int i=0;i<pos;i++)
              p[i].clear();
          for(int i=1;i<n;i++)
          {
              scanf("%d%d",&u,&v);
              edge[tope].e=v;
              edge[tope].next=head[u];
              head[u]=tope++;
              edge[tope].e=u;
              edge[tope].next=head[v];
              head[v]=tope++;
          }
          visit[1]=Ti=1;
          dfs(1);
          scanf("%d",&Q);
          for(int i=1;i<=Q;i++)
          {
              scanf("%d",&u);
              que[i].l=No[u].l;
              que[i].r=No[u].r;
              que[i].id=i;
          }
          sort(que+1,que+Q+1);
          memset(cover,0,sizeof(cover));
          for(int i=1,j=1;i<=n;i++)
          {
              p[num[i]].push_back(i);
              int Count=p[num[i]].size();
              if(Count>k)
              {
                  update(1,1,n,1,p[num[i]][Count-k-1],-1);
                  update(1,1,n,p[num[i]][Count-k-1]+1,p[num[i]][Count-k],1);
              }
              else
                if(Count==k)
                   update(1,1,n,1,p[num[i]][Count-k],1);
              while(i>=que[j].r)
              {
                  ans[que[j].id]=query(1,1,n,que[j].l);
                  j++;
                  if(j>Q)break;
              }
              if(j>Q)break;
          }
          if(T!=0)printf("\n");
          printf("Case #%d:\n",++T);
          for(int i=1;i<=Q;i++)
              printf("%d\n",ans[i]);
    }
    return 0;
}