POJ - 3162 Walking Race 树形dp 单调队列

  1 #include<cstdio>
  2 #include<queue>
  3 #include<algorithm>
  4 using namespace std;
  5 const int N=1000118;
  6 struct Side{
  7     int v,ne,len;
  8 }S[2*N];
  9 int sn,n,m,head[N],dis1[N],dis2[N],path[N];
 10 void add(int u,int v,int c)
 11 {
 12     S[sn].v=v;
 13     S[sn].len=c;
 14     S[sn].ne=head[u];
 15     head[u]=sn++;
 16 }
 17 void updata(int u,int v,int dis)//更新距离 
 18 {
 19     if(dis>dis1[u])//更新最大距离的来源点 
 20         path[u]=v;
 21     dis2[u]=max(dis2[u],min(dis1[u],dis)); 
 22     dis1[u]=max(dis1[u],dis);
 23 }
 24 void dfs1(int u,int f)
 25 {
 26     for(int i=head[u];i!=-1;i=S[i].ne)
 27     {
 28         int v=S[i].v;
 29         if(v!=f)
 30         {
 31             dfs1(v,u);//先搜索子节点 
 32             updata(u,v,dis1[v]+S[i].len);//更新 
 33         }
 34     }
 35 }
 36 void dfs2(int u,int f,int len)
 37 {
 38     updata(u,f,len);//由父节点传入的距离更新距离 
 39     for(int i=head[u];i!=-1;i=S[i].ne)
 40     {
 41         int v=S[i].v;
 42         if(v!=f)
 43         {
 44             if(v==path[u])
 45                 dfs2(v,u,dis2[u]+S[i].len);
 46             else
 47                 dfs2(v,u,dis1[u]+S[i].len);
 48         }
 49     }
 50 }
 51 int main()
 52 {
 53     int f,d;
 54     while(~scanf("%d%d",&n,&m))
 55     {
 56         for(int i=0;i<=n;i++)
 57         {
 58             head[i]=-1;
 59             dis1[i]=dis2[i]=0;
 60         }
 61         sn=0;
 62         for(int i=1;i<n;i++)
 63         {
 64             scanf("%d%d",&f,&d);
 65             add(i+1,f,d);
 66             add(f,i+1,d);
 67         }
 68         dfs1(1,0);
 69         dfs2(1,0,0);
 70         deque<int> qx,qd;//qx单调递增队列，qd单调递减队列  
 71         int ans=0;         //注意是双向队列，单向没有back() 
 72         for(int i=1,j=0;i<=n;i++)//j是左端，i是右端 
 73         {
 74             //维护单调性 
 75             while(qd.size()&&dis1[qd.back()]<dis1[i])
 76                 qd.pop_back();
 77             while(qx.size()&&dis1[qx.back()]>dis1[i])
 78                 qx.pop_back();
 79             qd.push_back(i);
 80             qx.push_back(i);
 81             while(qd.size()&&qx.size()&&dis1[qd.front()]-dis1[qx.front()]>m)
 82             {
 83                 //当最大值减最小值大于m时这时左端的位置就需要调整了
 84                 //左端肯定是要调整到最大值和最小值中较近的位置的前面 
 85                 if(qd.front()<qx.front())
 86                 {
 87                     j=qd.front();
 88                     qd.pop_front();
 89                 }
 90                 else
 91                 {
 92                     j=qx.front();
 93                     qx.pop_front();
 94                 }
 95                 //把区间外的数全部弹出 
 96                 while(qd.size()&&qd.front()<=j)
 97                     qd.pop_front();
 98                 while(qx.size()&&qd.front()<=j)
 99                     qx.pop_front();
100             }
101             ans=max(ans,i-j);//更新区间长度 
102         }
103         printf("%d\n",ans);
104     }
105     return 0;
106 }