SPOJ10707 COT2 Count on a tree II


Solution

我会强制在线版本! Solution戳这里

代码实现


#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<queue>
#include<set>
#include<map>
#include<iostream>
using namespace std;
#define ll long long
#define re register
#define file(a) freopen(a".in","r",stdin);freopen(a".out","w",stdout)
inline int gi()
{
    int f=1,sum=0;char ch=getchar();
    while(ch>'9' || ch<'0'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0' && ch<='9'){sum=(sum<<3)+(sum<<1)+ch-'0';ch=getchar();}
    return f*sum;
}
const int N=60010;
int Bl[N],B,P[N],ans[310][N],a[N],b[N],bl[N],num,p[N][310],Anum,rt[310],F[N];
struct array
{
    int num[210];
    int operator[](int x){return p[num[Bl[x]]][P[x]];};
    void insert(const array &pre,int x,int dep)
    {
        int block=Bl[x],t=P[x];
        memcpy(num,pre.num,sizeof(num));
        memcpy(p[++Anum],p[num[block]],sizeof(p[0]));
        p[Anum][t]=dep;num[block]=Anum;
    }
}s[N];
int to[N<<1],nxt[N<<1],front[N],cnt,dep[N],f[N][22],st[N],sta,kind;
inline void Add(int u,int v)
{
    to[++cnt]=v;nxt[cnt]=front[u];front[u]=cnt;
}
inline int dfs(int u,int fa)
{
    dep[u]=dep[fa]+1;
    f[u][0]=fa;
    s[u].insert(s[fa],a[u],dep[u]);
    st[++sta]=u;int mx=dep[u],now=sta;
    for(re int i=front[u];i;i=nxt[i])
    {
        int v=to[i];
        if(v==fa)continue;
        mx=max(mx,dfs(v,u));
    }
    if(mx-dep[u]>=B || now==1)
    {
        rt[++num]=u;
        for(re int i=now;i<=sta;i++)bl[st[i]]=num;
        sta=now-1;return dep[u]-1;
    }
    return mx;
}
int lca(int u,int v)
{
    if(dep[u]<dep[v])swap(u,v);
    for(re int i=20;~i;i--)
        if(dep[u]-(1<<i)>=dep[v])u=f[u][i];
    if(u==v)return u;
    for(re int i=20;~i;i--)
        if(f[u][i]!=f[v][i])
            u=f[u][i],v=f[v][i];
    return f[u][0];
}
inline void getans(int u,int fa,int BL)
{
    if(++F[a[u]]==1)kind++;
    ans[BL][u]=kind;
    for(re int i=front[u];i;i=nxt[i])
    {
        int v=to[i];
        if(v==fa)continue;
        getans(v,u,BL);
    }
    if(--F[a[u]]==0)kind--;
}
int solve_same(int x,int y)
{
    sta=0;
    for(kind=0;x!=y;x=f[x][0])
    {
        if(dep[x]<dep[y])swap(x,y);
        if(!F[a[x]]++)++kind,st[++sta]=a[x];
    }
    int QAQ=kind+(!F[a[x]]);
    for(;sta;sta--)F[st[sta]]=0;
    return QAQ;
}
int solve_diff(int x,int y)
{
    if(dep[rt[bl[x]]]<dep[rt[bl[y]]])swap(x,y);
    int sum=ans[bl[x]][y];
    int z=rt[bl[x]],d=dep[lca(x,y)];
    sta=0;
    for(;x!=z;x=f[x][0])
    {
        if(!F[a[x]] && s[z][a[x]]<d && s[y][a[x]]<d)
            F[st[++sta]=a[x]]=1,sum++;
    }
    for(;sta;sta--)F[st[sta]]=0;
    return sum;
}
int n,m;
void print(int x)
{
    if(x>=10)print(x/10);
    putchar(x%10+'0');
}
int main()
{
    n=gi();m=gi();B=sqrt(n);
    for(int i=1;i<=n;i++)Bl[i]=(i-1)/B+1,P[i]=i%B;
    for(re int i=1;i<=n;i++)a[i]=b[i]=gi();
    sort(b+1,b+n+1);int N=unique(b+1,b+n+1)-b-1;
    for(re int i=1;i<=n;i++)
        a[i]=lower_bound(b+1,b+N+1,a[i])-b;
    for(re int i=1;i<n;i++)
    {
        int u=gi(),v=gi();
        Add(u,v);Add(v,u);
    }
    dfs(1,1);
    for(re int i=1;i<=num;i++)getans(rt[i],rt[i],i);
    for(re int j=1;j<=20;j++)
        for(re int i=1;i<=n;i++)
            f[i][j]=f[f[i][j-1]][j-1];
    int lastans=0;
    while(m--)
    {
        int u=gi(),v=gi();
        if(bl[u]==bl[v])lastans=solve_same(u,v);
        else lastans=solve_diff(u,v);
        print(lastans);putchar('\n');
    }
    return 0;
}

【SPOJ10707】 COT2 Count on a tree II的更多相关文章

  1. 【SPOJ10707】COT2 - Count on a tree II

    题目大意:给定一棵 N 个节点的无根树,每个节点有一个颜色.现有 M 个询问,每次询问一条树链上的不同颜色数. 题解:学会了树上莫队. 树上莫队是将节点按照欧拉序进行排序,将树上问题转化成序列上的问题 ...

  2. 【树上莫队】【SP10707】 COT2 - Count on a tree II

    Description 给定一棵 \(n\) 个点的树,每个节点有一个权值,\(m\) 次询问,每次查询两点间路径上有多少不同的权值 Input 第一行是 \(n\) 和 \(m\) 第二行是 \(n ...

  3. spoj COT2 - Count on a tree II

    COT2 - Count on a tree II http://www.spoj.com/problems/COT2/ #tree You are given a tree with N nodes ...

  4. SPOJ COT2 - Count on a tree II(LCA+离散化+树上莫队)

    COT2 - Count on a tree II #tree You are given a tree with N nodes. The tree nodes are numbered from  ...

  5. COT2 - Count on a tree II(树上莫队)

    COT2 - Count on a tree II You are given a tree with N nodes. The tree nodes are numbered from 1 to N ...

  6. SPOJ10707 COT2 - Count on a tree II 【树上莫队】

    题目分析: 考虑欧拉序,这里的欧拉序与ETT欧拉序的定义相同而与倍增LCA不同.然后不妨对于询问$u$与$v$让$dfsin[u] \leq dfsin[v]$,这样对于u和v不在一条路径上,它们可以 ...

  7. 【SPOJ】10628. Count on a tree(lca+主席树+dfs序)

    http://www.spoj.com/problems/COT/ (速度很快,排到了rank6) 这题让我明白了人生T_T 我知道我为什么那么sb了. 调试一早上都在想人生. 唉. 太弱. 太弱. ...

  8. SPOJ COT2 Count on a tree II(树上莫队)

    题目链接:http://www.spoj.com/problems/COT2/ You are given a tree with N nodes.The tree nodes are numbere ...

  9. SPOJ COT2 Count on a tree II (树上莫队)

    题目链接:http://www.spoj.com/problems/COT2/ 参考博客:http://www.cnblogs.com/xcw0754/p/4763804.html上面这个人推导部分写 ...

随机推荐

  1. MongoDB相关记录

    win10中zip安装 下载地址:http://dl.mongodb.org/dl/win32/x86_64 首先解压至某文件夹, 使用管理员权限打开cmd或者powershell, 进入指定目录中的 ...

  2. PSP(3.30——4.5)以及周记录

    1.PSP 3.30 12:00 13:00 10 50 Account前端 A Y min 13:00 13:20 0 20 站立会议 A Y min 15:15 17:00 20 85 Accou ...

  3. 再读c++primer plus 004

    第九章  内存模型和名称空间 1.如果文件名包含在尖括号中,则c++编译器将在存储标准头文件的主机系统的文件系统中查找,但如果文件名包含在双引号中,则编译器将首先查找当前的工作目录或源代码目录(或其他 ...

  4. PARSEC安环境配置、运行

    1.getting started 2.run PARSEC on simulators Full-System Simulators: such as Simics, GEM5.Trace-Driv ...

  5. excel中vba将excel中数字和图表输出到word中

    参考:https://wenku.baidu.com/view/6c60420ecc175527072208af.html 比如将选区变为图片保存到桌面: Sub 将选区转为图片存到桌面() Dim ...

  6. powerdesiginer 生成oracle脚本问题,一步解决

    select * from tablename时, 总是提示错误:table or view does not exits 但是在user_tables中却可以看到刚建立的表. 原因是powerDes ...

  7. Ajax传数据到servlet

    //jsp代码 <%@ page language="java" contentType="text/html; charset=ISO-8859-1" ...

  8. WordPress-Word图片上传插件整合教程-Xproer.WordPaster

    插件下载(PHP):wordpress 3.7.1, 说明:由于许多插件可能使用相同钩子,导致冲突,所以提供手支方式整合. 1.上传插件目录. 说明:WordPress 3.7.1 使用的是TinyM ...

  9. winSockets编程(七)WSAAsyncSelect模式

    占位## #include <WinSock2.h> #include <Windows.h> #include <StrSafe.h> #pragma comme ...

  10. 2018-03-17 handler学习使用

    1.handler具体使用https://www.cnblogs.com/JohnTsai/p/5259869.html 2.handlerThread用法https://www.jianshu.co ...