POJ 3162 树形DP+单调队列

题目链接

题意:

给一颗树先求树上各点最远可到距离(HDU 2196),然后求最长区间长度,并且这个区间内最远距离的最值差<=m。

思路:

求最远可到距离,参考HDU 2196。求最值差,我们用单调队列进行计算。

q1将在当前区间的最小值保存在队列开头

q2将在当前区间的最大值保存在队列开头

当右端点向右移动时,为保证q1的队列开头为最小值,需要将q1内所有大于等于当前值的数全部扔出队列,同理为保证q2的队列开头为最大值,需要将q2内所有小于等于当前值的数全部扔出队列。

当左端点向右移动时,为保证q1,q2队列中的最值为有效的所以需要将不是区间内的数全部扔出队列。

C++代码:

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
const int maxn = 1000010;

int n,m,tol,head[maxn];
struct edge
{
    int to,next,cost;
}es[maxn];

void addedge( int u , int v , int w )
{
    es[tol].to = v;
    es[tol].cost = w;
    es[tol].next = head[u];
    head[u]  = tol++;
}

int dp[maxn][5],son[maxn];

void dfs1( int u , int f )
{
    for ( int i=head[u] ; i!=-1 ; i=es[i].next )
    {
        int v = es[i].to,w = es[i].cost;
        if ( v!=f )
        {
            dfs1( v , u );
            if ( son[u]==-1||dp[v][0]+w>dp[u][0] )
            {
                dp[u][1] = dp[u][0];
                dp[u][0] = dp[v][0]+w;
                son[u] = v;
            }
            else if ( dp[v][0]+w<=dp[u][0]&&dp[v][0]+w>dp[u][1] )
            {
                dp[u][1] = dp[v][0]+w;
            }
        }
    }
}

void dfs2( int u , int f )
{
    for ( int i=head[u] ; i!=-1 ; i=es[i].next )
    {
        int v = es[i].to,w = es[i].cost;
        if ( v!=f )
        {
            if ( v!=son[u] )
                dp[v][2] = w+max(dp[u][0],dp[u][2]);
            else
                dp[v][2] = w+max(dp[u][1],dp[u][2]);
            dfs2( v , u );
        }
    }
}

int q1[maxn],l1,r1;
int q2[maxn],l2,r2;

int slove()
{
    int ans = 0;
    l1 = r1 = 0;
    l2 = r2 = 0;
    for ( int i=1,j=1 ; j<=n ; j++ )
    {
        while( l1=dp[j][3] ) r1--;
        q1[r1++] = j;
        while( l2

你可能感兴趣的:(树形DP)