指针优化树形DP
BZOJ3522的n只有5000,可以随便用一个n^2的大暴力搞过去,就不讲了。
BZOJ4543好丧啊,看到题解我当时就震惊了,题解点我
n有十万,使人一看到就会放弃一些树形DP的思路,而正解正是树形DP,难点就在内存和时间的优化。
可以证明复杂度是O(n),详见上面的题解呀。
#include<cstdio>
#include<cstring>
#define N 5005
#define ll long long
using namespace std;
struct edge{int next,to;}e[N<<1];
int ecnt=0, last[N], cnt[N], vis[N], timer, mx, mmx;
ll f[N][4];
void addedge(int a, int b)
{
e[++ecnt]=(edge){last[a],b};
last[a]=ecnt;
}
void dfs(int x, int fa, int dep)
{
if(mx<dep) mx=dep;
if(vis[dep] < timer)vis[dep] = timer, cnt[dep] = 1;
else cnt[dep]++;
for(int i = last[x]; i; i=e[i].next)
{
int y=e[i].to;
if(y==fa)continue;
dfs(y,x,dep+1);
}
}
int main()
{
int n;
ll ans=0;
scanf("%d",&n);
for(int i = 1; i < n; i++)
{
int a, b;
scanf("%d%d",&a,&b);
addedge(a,b);
addedge(b,a);
}
for(int x = 1; x <= n; x++)
{
memset(f,0,sizeof(f));
for(int i = last[x]; i; i=e[i].next)
{
timer++;
int y=e[i].to;
mx=0;
dfs(y,x,1);
if(mmx<mx)mmx=mx;
for(int j = 1; j <= mx; j++)
{
f[j][3] += f[j][2]*cnt[j];
f[j][2] += f[j][1]*cnt[j];
f[j][1] += cnt[j];
}
}
for(int i = 1; i <= mmx; i++)
ans += f[i][3];
}
printf("%lld\n",ans);
}
#include<cstdio>
#define ll long long
#define N 100005
using namespace std;
ll *f[N], *g[N], temp[N*20], *cur, ans;
int n,ecnt, last[N], under[N], dep[N];
struct edge{int next,to;}e[N<<1];
void addedge(int a, int b)
{
e[++ecnt]=(edge){last[a],b};
last[a]=ecnt;
}
void dfs(int x, int fa)
{
under[x]=x;
dep[x]=dep[fa]+1;
for(int i = last[x]; i; i=e[i].next)
{
int y=e[i].to;
if(y==fa)continue;
dfs(y,x);
if(dep[under[y]] > dep[under[x]])
under[x]=under[y];
}
for(int i = last[x]; i; i=e[i].next)
{
int y = e[i].to;
if(y==fa || (under[x] == under[y] && x!=1))continue;
int z = under[y];
cur += dep[z] - dep[x] + 5;
f[z] = cur;
g[z] = (cur+=5);
cur += (dep[z] - dep[x])*2 + 5;
}
}
void dp(int x, int fa)
{
for(int i = last[x]; i; i=e[i].next)
{
int y=e[i].to;
if(y==fa)continue;
dp(y,x);
if(under[y] == under[x])
{
f[x] = f[y] - 1;
g[x] = g[y] + 1;
}
}
f[x][0] = 1;
ans += g[x][0];
for(int i = last[x]; i; i=e[i].next)
{
int y=e[i].to;
if(y==fa || under[y] == under[x])continue;
for(int j = 0; j <= dep[under[y]] - dep[x]; j++)
{
ans += g[y][j] * f[x][j-1];//用y的f更新
ans += f[y][j] * g[x][j+1];//用y的g更新
}
for(int j = 0; j <= dep[under[y]] - dep[x]; j++)
{
g[x][j-1] += g[y][j];
g[x][j+1] += f[x][j+1] * f[y][j];
f[x][j+1] += f[y][j];
}
}
}
int main()
{
scanf("%d",&n);
for(int i = 1, a, b; i < n; i++)
{
scanf("%d%d",&a,&b);
addedge(a,b);
addedge(b,a);
}
cur = temp + 5;
dfs(1,0);
dp(1,0);
printf("%lld\n",ans);
}