题意
给定一棵
n
n
个节点的树,每个点都有点权。求一条最长的路径,是路径上点的点权序列形成等差数列。
思路
很明显是保存以 u u 为根,以 为公差的点权序列最大长度,但这个 d d 值飘忽不定,但个数不多,由此想到 类型的 dp d p (初值是 0 0 ,所以不包括路径根)。当 作为根时,每添加一个子树 v v ,更新 的值,然后用 dpu[Vu−Vv]+dpu[Vv−Vu]+1 d p u [ V u − V v ] + d p 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;
}