P2607 [ZJOI2008]骑士

2020-11-01 10:01:32 阅读：183 来源： 互联网

标签：rt int long vis P2607 ZJOI2008 mathcal 骑士 dp

题目大意：有\(n\)个士兵，每个士兵有\(d_i,a_i\)，表示\(i\)和\(d_i\)士兵不能同时出征，\(i\)士兵战斗力为\(a_i\)，组一个士兵集合使其战斗力最大。

基环树\(\mathcal{dp}\)。

每次找到一个环并强行断开，进行两次\(\mathcal{dp}\)，\(\mathcal{dp}\)方法：

设\(dp_{i,0}\)为不选\(i\)的最大战斗力，\(dp_{i,1}\)为选\(i\)的最大战斗力。

\[dp_{i,0} = \sum_{v \in \texttt{son}} \max(dp_{v,0},dp_{v,1})\\ dp_{i,1} = \left(\sum_{v \in \texttt{son}} dp_{v,0}\right) + a_i\\ \]

\(\texttt{dfs}\)既珂。

找环：

有向图：

迭代，每次判断\(f_x\)是否被访问过，如果访问过，说明\(x\)为环上一点，否则\(x \leftarrow f_x\)继续迭代。

int findcircle(int x) 
{
    vis[x] = 1;
    while(!vis[f[x]]) x = f[x],vis[f[x]] = 1;
    return x;
}

无向图：

拓扑排序，结束后度数\(\ge 2\)的节点为环上的点。

bool bfs(void)
{
    queue <int> q;
    int cnt = 0;
    for(int i = 1; i <= n; i++) if(du[i] == 1) q.push(i);
    while(!q.empty())
    {
        int head = q.front();q.pop();
        cnt++;
        for(int i = 0; i < g[head].size(); i++)
        {
            int v = g[head][i];
            if(du[v])
            {
                if(--du[v] == 1) q.push(v);
            }
        }
    }
    return cnt == n; // 判断是否有环
}

这里是有向图，所以用第一个方法。先断\(x\)，进行\(\mathcal{dp}\)，再断\(f_x\)，进行\(\mathcal{dp}\)，取最大值相加（因为是森林）。

# include <bits/stdc++.h>
using namespace std;
const int N = 1e6 + 5,inf = 0x3f3f3f3f;

int n;
vector <int> g[N];
int a[N],du[N];
long long dp[N][2]; // 0 not choose 1 choose
int h[N],htot;
bool vis[N];
int f[N];
int rt;

long long ans;

void DP(int x)
{
    vis[x] = 1;
    dp[x][0] = 0,dp[x][1] = 1ll * a[x];
    for(int i = 0; i < g[x].size(); i++)
    {
        int v = g[x][i];
        if(v != rt)
        {
            DP(v);
            dp[x][1] += dp[v][0];
            dp[x][0] += max(dp[v][0],dp[v][1]);
        }
        else dp[v][1] = -inf; //强制不选
    }
    return;
}

void find_circle(int x)
{
    vis[x] = 1;
    while(!vis[f[x]]) x = f[x],vis[x] = 1;
    rt = x;
    // printf("rt = %d\n",rt);
    // for(int i = 1; i <= n; i++) dp[i][0] = 0,dp[i][1] = a[i];
    DP(rt);
    long long dpx0 = dp[x][0],dpx1 = dp[x][1];
    x = f[x];
    rt = x;
    // for(int i = 1; i <= n; i++) dp[i][0] = 0,dp[i][1] = a[i];
    DP(rt);
    long long _dpx0 = dp[x][0],_dpx1 = dp[x][1];
    ans += max(max(dpx0,_dpx1),max(dpx1,_dpx0));
    return;
}

int main(void)
{
    scanf("%d",&n);
    for(int i = 1; i <= n; i++)
    {
        int v;
        scanf("%d%d",&a[i],&v);
        f[i] = v;
        g[v].push_back(i);
    }
    for(int i = 1; i <= n; i++)
    {
        if(!vis[i]) 
        {
            find_circle(i);
        }
    }
    printf("%lld\n",ans);
    return 0;
}

标签：rt,int,long,vis,P2607,ZJOI2008,mathcal,骑士,dp
来源： https://www.cnblogs.com/luyiming123blog/p/P2607.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

P2607 [ZJOI2008]骑士