RMQ模板hdu 5452

 題意:有一棵生成樹,有n個點,給出m-n+1條邊,截斷一條生成樹上的邊後,再截斷至少多少條邊才能使圖不連通, 問截斷總邊數?

解題思路:

  因爲只能在生成樹上截斷一條邊(u, v),所以只需要統計以v爲根節點的子生成樹裏的節點與子生成樹外的節點的邊數就可以了。對於新加入的邊(u’, v’)來說,隻影響以LCA(u, v)爲根節點的子樹裏面的節點。統計所有答案,掃一遍輸出最小即可
  dp值記錄的是以v爲根節點的子樹與外界連的邊數,連到lca-2就可以了,然後樹形dp一遍統計最小值即可

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;
#include <queue>
#include <stack>
#include <vector>
#include <bitset>
#include <deque>
#include <set>
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <map>
typedef long long LL;
const int INF = 1<<29;
const LL mod = 530600414;
const int MAXN = 81000;//點數的最大值
const int MAXM = 604000;//邊數的最大值
int rmq[2*MAXN];//rmq數組,就是歐拉序列對應的深度序列
struct ST
{
    int mm[2*MAXN];
    int dp[2*MAXN][20];//最小值對應的下標
    void init(int n)
    {
        mm[0] = -1;
        for(int i = 1;i <= n;i++)
        {
            mm[i] = ((i&(i-1)) == 0)?mm[i-1]+1:mm[i-1];
            dp[i][0] = i;
        }
        for(int j = 1; j <= mm[n];j++)
            for(int i = 1; i + (1<<j) - 1 <= n; i++)
                dp[i][j] = rmq[dp[i][j-1]] < rmq[dp[i+(1<<(j-1))][j-1]]?dp[i][j-1]:dp[i+(1<<(j-1))][j-1];
    }
    int query(int a,int b)//查詢[a,b]之間最小值的下標
    {
        if(a > b)swap(a,b);
        int k = mm[b-a+1];
        return rmq[dp[a][k]] <= rmq[dp[b-(1<<k)+1][k]]?dp[a][k]:dp[b-(1<<k)+1][k];
    }
};
//邊的結構體定義
struct Edge{
    int from,to,next;
}edge[MAXM];

int edgenum,head[MAXN];
int F[MAXN*2];//歐拉序列,就是dfs遍歷的順序,長度爲2*n-1,下標從1開始
int P[MAXN];//P[i]表示點i在F中第一次出現的位置
int cnt;
ST st;

void addedge(int u,int v)//加邊,無向邊需要加兩次
{
    Edge E={u,v,head[u]};
    edge[edgenum] = E;
    head[u] = edgenum++;
}
int len[MAXN],fa[MAXN];
void dfs(int u,int pre,int dep)
{
    fa[u]=pre;
    F[++cnt] = u;
    rmq[cnt] = dep;
    P[u] = cnt;
    for(int i = head[u];i != -1;i = edge[i].next)
    {
        int v = edge[i].to;
        if(v == pre)continue;
        dfs(v,u,dep+1);
        F[++cnt] = u;
        rmq[cnt] = dep;
    }
}
void LCA_init(int root,int node_num)//查詢LCA前的初始化
{
    cnt = 0;
    dfs(root,root,0);
    st.init(2*node_num-1);
}
int query_lca(int u,int v)//查詢u,v的lca編號
{
    return F[st.query(P[u],P[v])];
}
int dp[MAXN];
void init(){
    memset(head,-1,sizeof(head)), edgenum = 0;
    memset(dp,0,sizeof(dp));
}
int gao(int u,int fa)
{
    for(int i=head[u];~i;i=edge[i].next)
    {
        int v=edge[i].to;
        if(v==fa) continue;
        gao(v,u);
        dp[u]+=dp[v];
    }
    return 0;
}
int main()
{
    int t,cas=1,n,m,a,b;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        init();
        for(int i=0;i<n-1;i++)
        {
            scanf("%d%d",&a,&b);
            addedge(a,b);
            addedge(b,a);
        }
        LCA_init(1,n);
        for(int i=0;i<=m-n;i++)
        {
            scanf("%d%d",&a,&b);
            int x=query_lca(a,b);
            dp[a]++;
            dp[b]++;
            dp[x]-=2;
//            dp[fa[x]]--;
        }
        gao(1,1);
        int ans=INF;
        for(int i=2;i<=n;i++)
            ans=min(ans,dp[i]+1);
        printf("Case #%d: ",cas++);
        printf("%d\n",ans);
    }
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章