• noip2016 天天爱跑步


    没看过正解。。应该是些乱七八糟想不出来的东西

    解法1:

    首先,必须要做的是将每条路径拆成2个直的路径

    那么对于那条从深度大的到深度小的路径 dep[x]-dep[y]应该等于观察时间

    那么就可以在这些点打标记

    那问题在于怎么找这些点

    可以把深度为x的数组用vector搞出来

    然后每次判断一下x里面的元素是否在他的子树中(判断一下lca就好了)

    树剖判断一下每个点的覆盖次数

    对于另一条路径同理做

    代码:

    解法2:

    这个解法复杂度是可以分析出来的

    首先还是把路径拆成两条

    那么对于那条从深度大的到深度小的路径

    考虑启发式合并

    对于每个点维护一颗splay表示子树中出现的dep值的个数

    将路径顶端和底端分别在该点打标记

    访问到顶端时删除底端时插入

    由于每一次向上会导致所有元素的权值变化所以可以加一个标记(插入时减掉这个值)

    那么到查询点的时候就在splay上查询就可以了

    那么复杂度是nlogn*splay的复杂度 也就是nlog^2n  不过splay的常数感觉这方法还是很虚啊。。。

    还有就是splay的insert操作之后要splay一下

    防止是这颗树的第一个节点

    代码:

    #include <bits/stdc++.h>
    using namespace std;
    #define maxn 6200000
    #define maxn2 310000
    int count2[maxn],leftson[maxn],rightson[maxn],fa[maxn],root[maxn2];
    struct re{
      int a,b,c;
    }a[maxn2*2];
    int ans[maxn2],bz[maxn2][20],dep[maxn2],vw[maxn2],head[maxn2],l,last[maxn2];
    int hf[maxn2],cc[maxn2],dd[maxn2],lcaa[maxn2],num[maxn],num2,data[maxn];
    int n,m;
    vector<int> ve[maxn];
    inline void updata(int x)
    {
      count2[x]=count2[leftson[x]]+count2[rightson[x]]+1;
    }
    inline void rotate(int x,int y)
    {
      int father=fa[x];
      if (y==1)
      {
        rightson[father]=leftson[x];
        if (leftson[x]) fa[leftson[x]]=father;
      } else
      {
        leftson[father]=rightson[x];
        if (rightson[x]) fa[rightson[x]]=father;
      }
      fa[x]=fa[father];
      if (fa[father])
      {
        if (leftson[fa[father]]==father) leftson[fa[father]]=x;
        else rightson[fa[father]]=x;
      }
      fa[father]=x;
      if (y==1) leftson[x]=father; else rightson[x]=father;
      updata(father); updata(x);
    }
    inline void splay(int k,int x,int goal)
    {
      if (x==root[k]) return;
      int father=fa[x];
      while (father!=goal)
      {
        if (fa[father]==goal)
        {
          if (x==leftson[father]) rotate(x,2); else rotate(x,1);
        } else
        {
          if (father==leftson[fa[father]])
          {
            if (x==leftson[father]) rotate(father,2),rotate(x,2);
            else rotate(x,1),rotate(x,2);
          } else
          {
            if (x==rightson[father]) rotate(father,1),rotate(x,1);
            else rotate(x,2),rotate(x,1);
          }
        }
        father=fa[x];
      }
      if (goal==0) root[k]=x;
    }
    inline void dfs(int x,int y)
    {
      bz[x][0]=y; dep[x]=dep[y]+1; 
      int u=head[x];
      while (u)
      {
        int v=a[u].b;
        if (v!=y) dfs(v,x);
        u=a[u].a;
      }
    }
    inline int get_lca(int x,int y)
    {
       if (dep[x]<dep[y]) swap(x,y);
       for (int i=19;i>=0;i--)
         if (dep[bz[x][i]]>=dep[y]) x=bz[x][i];
       if (x==y) return(x);
       for (int i=19;i>=0;i--)
         if (bz[x][i]!=bz[y][i])
         {
           x=bz[x][i],y=bz[y][i];
         }
       return(bz[x][0]);
    }
    inline void arr(int x,int y)
    {
      a[++l].a=head[x];
      a[l].b=y;
      head[x]=l;
    }
    inline void insert(int x,int v,int w)
    {
      int tmp=x;x=root[x];
      while (x)
      {
        if (data[x]==v) break;
        count2[x]++;
        if (v<data[x])
        {
          if (!leftson[x]) break;
          x=leftson[x];
        } else
        {
          if (!rightson[x]) break;
          x=rightson[x];
        }
      }
      if (x!=0&&data[x]==v) num[x]+=w; else
      {
        data[++num2]=v; count2[num2]=1; fa[num2]=x;
        num[num2]=w;
        if (x!=0)
        {
          if (v>data[x]) rightson[x]=num2;
          else leftson[x]=num2;
        }
        splay(tmp,num2,0);
      }
    }
    inline void delete1(int k)
    {
      int x=leftson[root[k]];
      if (x==0)
      {
        root[k]=rightson[root[k]]; fa[root[k]]=0; return;
      }
      while (rightson[x]) x=rightson[x];
      splay(k,x,root[k]);
      rightson[x]=rightson[root[k]];
      if (rightson[root[k]]) fa[rightson[root[k]]]=x;
      updata(x);
      fa[x]=0; root[k]=x;
    }
    inline int search(int x,int goal)
    {
      x=root[x];
      while (1)
      {
        if (data[x]==goal) return(x);
        if (data[x]<goal)
        {
           if (!rightson[x]) return(x); 
           x=rightson[x];
        }
        else
        {
           if (!leftson[x]) return(x); 
           x=leftson[x];
        }
      }
    }
    inline void merge(int x,int y,int z)
    {
      while (root[x])
      {
        int tmp=data[root[x]]-last[y]+last[x];
        int o=search(y,tmp);
        if (o!=0&&data[o]==tmp) num[o]+=num[root[x]];
        else insert(y,tmp,num[root[x]]);
        delete1(x);
      }
      root[z]=root[y]; last[z]=last[y];
    }
    inline void dfs2(int x,int fa,int lq)
    {
      int u=head[x],pp=ve[x].size();
      for (int i=0;i<=pp/2-1;i++)
      if (!ve[x][i*2+1])
      {
          insert(x,ve[x][i*2],1);
      }
      while (u)
      {
        int v=a[u].b;
        if (v!=fa)
        {
           dfs2(v,x,lq);
           last[v]+=lq;
           if (count2[root[x]]<count2[root[v]]) merge(x,v,x);
           else merge(v,x,x);
        }
        u=a[u].a;
      }
      int tmp=search(x,vw[x]-last[x]);
      if (data[tmp]==vw[x]-last[x]) ans[x]+=num[tmp];
      for (int i=0;i<=pp/2-1;i++)
      if (ve[x][i*2+1])
      {
          int tmp=search(x,ve[x][i*2]-last[x]);
          if (num[tmp]>1) num[tmp]--;
          else 
          {
            splay(x,tmp,0);
             delete1(x);
          }
      }
    }
    int main()
    {
      freopen("noip.in","r",stdin);
      freopen("noip.out","w",stdout);
      std::ios::sync_with_stdio(false);
      cin>>n>>m;
      int c,d;
      for (int i=1;i<=n-1;i++)
        cin>>c>>d,arr(c,d),arr(d,c);
      for (int i=1;i<=n;i++)
        cin>>vw[i];
      dfs(1,0);
      for (int i=1;i<=19;i++)
        for (int j=1;j<=n;j++)
          bz[j][i]=bz[bz[j][i-1]][i-1];
      for (int i=1;i<=m;i++)
      {
        cin>>c>>d;
        cc[i]=c; dd[i]=d;
        int w=get_lca(c,d);
        lcaa[i]=w;
        hf[i]=dep[c]-dep[w];
      }
      for (int i=1;i<=m;i++)
      {
        ve[cc[i]].push_back(0); ve[cc[i]].push_back(0);
        ve[lcaa[i]].push_back(hf[i]); ve[lcaa[i]].push_back(1);
      }
      dfs2(1,0,1);
      memset(leftson,0,sizeof(leftson));
      memset(rightson,0,sizeof(rightson));
      memset(num,0,sizeof(num));
      memset(count2,0,sizeof(count2));
      memset(fa,0,sizeof(fa));
      memset(root,0,sizeof(root));
      memset(last,0,sizeof(last)); num2=0;
      memset(data,0,sizeof(data));
      for (int i=1;i<=n;i++)
      {
        vector<int> tmp; swap(tmp,ve[i]); 
       // ve[i].clear();
      }
      for (int i=1;i<=m;i++)
      {
        ve[dd[i]].push_back(hf[i]+dep[dd[i]]-dep[lcaa[i]]);
        ve[dd[i]].push_back(0);
        ve[lcaa[i]].push_back(hf[i]); 
        ve[lcaa[i]].push_back(1);
      }
      dfs2(1,0,-1);
      for (int i=1;i<=m;i++)
      {
        if (hf[i]==vw[lcaa[i]]) ans[lcaa[i]]--;
      }
      for (int i=1;i<=n-1;i++) cout<<ans[i]<<" ";
      cout<<ans[n];
      return 0;
    }
  • 相关阅读:
    被Play framework狠狠的play了一把
    ant导入Zookeeper到Eclipse错误path contains invalid character
    Hadoop2.4代码的坑
    uml类关系
    Hadoop 源码编译导出
    Eclipse 导入 Hadoop 源码
    js中的prototype和constructor
    react之路:使用redux-immutable
    react之路:使用actionCreators和constants
    react之路:使用combineReducers拆分reducer
  • 原文地址:https://www.cnblogs.com/yinwuxiao/p/8534832.html
Copyright © 2020-2023  润新知