CF587C Duff in the Army

题目:http://codeforces.com/problemset
分析:倍增求lca,合并操作比较神奇,注意边界写法,有时包含顶端节点有时不包含,比如往上走2^0,包不包含他的父亲.
代码:

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <iostream>
#include <vector>
using namespace std;
const int Tmax=100005,Tmax2=18;
struct node{
    int c[15],num;
    node(){num=0;}
    void add(int x)
    {
        int p=lower_bound(c+1,c+1+num,x)-c,i;
        if(p<=10){
            for(i=10;i>=p;i--)
              c[i]=c[i-1];
            c[p]=x;
            num++;
        }
    }
};
node g[Tmax2+5][Tmax];
int n,m,q,dep[Tmax],f[Tmax2+5][Tmax];
vector<int> G[Tmax];
node operator +(const node &a,const node &b)
{
    node tmp;
    int i=1,j=1;
    tmp.num=0;
    while((i<=a.num||j<=b.num)&&tmp.num<10)
    {
        tmp.num++;
        if(i>a.num) tmp.c[tmp.num]=b.c[j],j++;
        else if(j>b.num) tmp.c[tmp.num]=a.c[i],i++;
        else if(a.c[i]<b.c[j]) tmp.c[tmp.num]=a.c[i],i++;
        else tmp.c[tmp.num]=b.c[j],j++;
    }
    return tmp;
}
void dfs(int x,int d)
{
    dep[x]=d;
    int i,len=G[x].size();
    for(i=0;i<len;i++)
      if(dep[G[x][i]]==0)
      {
        f[0][G[x][i]]=x;
        dfs(G[x][i],d+1);
      }
    return;
}
void init()
{
    int i,j;
    for(i=1;i<=Tmax2;i++)
      for(j=1;j<=n;j++)
      {
        f[i][j]=f[i-1][f[i-1][j]];
        g[i][j]=g[i-1][j]+g[i-1][f[i-1][j]];
      }
    return;
}
void work()
{
    int i,j,u,v,a;
    node ans;
    for(i=1;i<=q;i++)
    {
        ans.num=0;
        scanf("%d%d%d",&u,&v,&a);
        if(dep[u]<dep[v]) swap(u,v);
        for(j=Tmax2;j>=0;j--)
          if(dep[f[j][u]]>=dep[v]){
            ans=ans+g[j][u];
            u=f[j][u];
          }
        for(j=Tmax2;j>=0;j--)
            if(f[j][u]!=f[j][v])
            {
            ans=ans+g[j][u]+g[j][v];
            u=f[j][u];
            v=f[j][v];
            }
        ans=ans+g[0][u];
        if(u!=v) ans=ans+g[0][f[0][u]]+g[0][v];
        a=min(a,ans.num);
        printf("%d ",a);
        for(j=1;j<=a;j++)
          printf("%d ",ans.c[j]);
        printf("\n");
    }
    return;
}
int main()
{
    int i,u,v;
    scanf("%d%d%d",&n,&m,&q);
    for(i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    for(i=1;i<=m;i++)
    {
        scanf("%d",&u);
        g[0][u].add(i);
    }
    dfs(1,1);
    init();
    work();
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值