HihoCoder-1676 树上的等差数列（树形DP）

题意

给定一棵 $n$ 个节点的树，每个点都有点权。求一条最长的路径，是路径上点的点权序列形成等差数列。
$1 \leq n \leq 100000$

思路

很明显是保存以 $u$ 为根，以 $d$ 为公差的点权序列最大长度，但这个 $d$ 值飘忽不定，但个数不多，由此想到 $map$ 类型的 $dp$ （初值是 $0$ ，所以不包括路径根）。当 $u$ 作为根时，每添加一个子树 $v$ ，更新 $dp_u[V_u-V_v]$ 的值，然后用 $dp_u[V_u-V_v]+dp_u[V_v-V_u]+1$ ，注意公差 $0$ 为的值加两遍相同的值，所以特殊处理 $0$ ，存最大值和次大值。

代码

#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<map>
#define FOR(i,x,y) for(int i=(x);i<=(y);i++)
#define DOR(i,x,y) for(int i=(x);i>=(y);i--)
#define N 100003
typedef long long LL;
using namespace std;
template<const int maxn,const int maxm>struct Linked_list
{
    int head[maxn],to[maxm],nxt[maxm],tot;
    void clear(){memset(head,-1,sizeof(head));tot=0;}
    void add(int u,int v){to[++tot]=v,nxt[tot]=head[u],head[u]=tot;}
    #define EOR(i,G,u) for(int i=G.head[u];~i;i=G.nxt[i])
};
Linked_list<N,N<<1>G;
map<int,int>dp[N];
int V[N],dp0[N][2];
int n,ans;

void dfs(int u,int f)
{
    EOR(i,G,u)
    {
        int v=G.to[i];
        if(v==f)continue;
        dfs(v,u);
        if(V[u]==V[v])
        {
            if(dp0[u][0]<dp0[v][0]+1)
                dp0[u][1]=dp0[u][0],dp0[u][0]=dp0[v][0]+1;
            else if(dp0[u][1]<dp0[v][0]+1)
                dp0[u][1]=dp0[v][0]+1;
            ans=max(ans,dp0[u][0]+dp0[u][1]+1);
        }
        else
        {
            dp[u][V[u]-V[v]]=max(dp[u][V[u]-V[v]],dp[v][V[u]-V[v]]+1);
            ans=max(ans,dp[u][V[u]-V[v]]+dp[u][V[v]-V[u]]+1);
        }
    }
}

int main()
{
    G.clear();
    scanf("%d",&n);
    FOR(i,1,n)scanf("%d",&V[i]);
    FOR(i,1,n-1)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        G.add(u,v);
        G.add(v,u);
    }
    dfs(1,0);
    printf("%d\n",ans);
    return 0;
}