lightoj 1230 - Placing Lampposts 树形DP

给定一张无向图，有n个点，m条边，可以在有些点上放置灯，会照亮与其相连的所有边，问最少多少个灯照亮所有的边，并且用最少的灯的 情况下最多有多少条边可以同时被俩灯照亮..还要输出多少边被一灯照亮。

首先第一问就是普通树形DP...也就是如果父节点亮灯的话儿子节点可亮可不亮，否则儿子节点必须亮这样...

就是dp[i][j] j的01状态表示i这个点是否亮灯的最少用灯数量。和之前的树形DP入门题类似。

对于第二问来言，其实就是找出来最多的父节点亮灯同时子节点也亮灯那么有一条边就被照亮两次...

bs[i][j]，j的01状态表示i这个点是否亮灯的最少用灯数量情况下的最多被照亮两次的边。

所以对于父节点亮灯的情况下，看子节点亮灯用灯少还是不亮灯用灯少来转移DP数组的同时转移bs数组。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define ull unsigned long long
#define mod 1000000007
#define inf 0x3f3f3f3f
int dp[1234][2],bs[1234][2],used[1234];
int n,m,tot,head[1234];
struct edge
{
    int u,v,next;
}edge[3000];
void add(int u,int v)
{
    edge[tot].v=v;
    edge[tot].next=head[u];
    head[u]=tot++;
}
void init()
{
    memset(used,0,sizeof(used));
    memset(dp,0,sizeof(dp));
    memset(bs,0,sizeof(bs));
    tot=0;
    memset(head,-1,sizeof(head));
}
void dfs(int u)
{
    used[u]=1;
    dp[u][1]=1;
    dp[u][0]=0;
    bs[u][1]=0;
    bs[u][0]=0;
    for(int i=head[u];i!=-1;i=edge[i].next)
    {
        int v=edge[i].v;
        if(used[v]) continue;
        dfs(v);
        dp[u][0]+=dp[v][1];
        bs[u][0]+=bs[v][1];
        if(dp[v][0]<dp[v][1])
        {
            dp[u][1]+=dp[v][0];
            bs[u][1]+=bs[v][0];
        }
        else if(dp[v][0]>dp[v][1])
        {
            dp[u][1]+=dp[v][1];
            bs[u][1]+=bs[v][1]+1;
        }
        else
        {
            dp[u][1]+=dp[v][1];
            bs[u][1]+=max(bs[v][1]+1,bs[v][0]);
        }
    }
}
int main()
{
    int t;
    scanf("%d",&t);
    for(int cas=1;cas<=t;cas++)
    {
        scanf("%d %d",&n,&m);
        init();
        int u,v;
        for(int i=0;i<m;i++)
        {
            scanf("%d %d",&u,&v);
            add(u,v);
            add(v,u);
        }
        int ans1=0,ans2=0;
        for(int i=1;i<=n;i++)
        {
            if(used[i]) continue;
            dfs(i);
            if(dp[i][0]>dp[i][1])
            {
                ans1+=dp[i][1];
                ans2+=bs[i][1];
            }
            else if(dp[i][0]<dp[i][1])
            {
                ans1+=dp[i][0];
                ans2+=bs[i][0];
            }
            else
            {
                ans1+=dp[i][0];
                ans2+=max(bs[i][0],bs[i][1]);
            }
        }
        printf("Case %d: %d %d %d\n",cas,ans1,ans2,m-ans2);
    }
    return 0;
}