【ACM算法竞赛日常训练】DAY3题解与分析【旅游】【tokitsukaze and Soldier】

DAY3共2题:

  • 旅游

  • tokitsukaze and Soldier

🎈 作者:Eriktse
🎈 简介:19岁,211计算机在读,现役ACM银牌选手🏆力争以通俗易懂的方式讲解算法!❤️欢迎关注我,一起交流C++/Python算法。(优质好文持续更新中……)🚀
🎈 原文链接(阅读原文获得更好阅读体验):

旅游

题目传送门:https://ac.nowcoder.com/acm/problem/15748

该题主要考察对树的理解,以及简单的树上dp和贪心算法。

我们将会住的节点标记为1,其余不住的节点标记为0

我们可以发现,根节点(s)是一定会标记为1的,那么剩下的节点该怎么分配可以使得标记为1的节点数最多呢?

当我们在某个点x标记时,我们可以发现它的父亲、儿子们都不能再被标记了。但是点x的兄弟却不受影响,接下来考虑一下哪些节点的兄弟多呢?应该是叶子节点。

所以我们可以想到首先将根节点和叶子结点全部都标记为1,然后遍历整棵树,如果某个点的父亲和儿子们都没被标记,那么他也可以被标记为1

注意考虑特殊情况,比如只有一个点的树,只有两个点的树....

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxn = 5e5 + 9, inf = 8e18;

bitset<maxn> sel;

vector<int> g[maxn];
int n, s;

void dfs(int x, int pre)
{
    //如果到了叶子节点,就直接标记并返回
    //这里sel[x] = !sel[pre]是考虑到叶子的深度可能为2(即叶子的父亲就是根)
    //此时根一定被标记,那么叶子就不能被标记
    //如果是一般情况,那么父亲肯定不会被标记(因为父亲的标记需要儿子处理完成之后再决定)
    //自己就打上标记
    if(g[x].size() == 1 and x != s)return sel[x] = !sel[pre], void();
    
    bool tag = true;//tag == true表示当前点可以被标记
    
    if(x == s)sel[x] = true;//根一定被标记
    else if(sel[pre])tag = false;//如果父亲被标记了,那么当前点一定不能被标记
        
    //看看儿子们是否被标记
    for(auto &y : g[x])
    {
        if(y == pre)continue;
        dfs(y, x);
        if(sel[y])tag = false;
    }
    sel[x] = tag;
}

signed main()
{
    scanf("%lld %lld", &n, &s);
    
    for(int i = 1;i < n; ++ i)
    {
        int x, y;scanf("%lld %lld", &x, &y);
        g[x].push_back(y), g[y].push_back(x);
    }
    
    dfs(s, 0);
    int ans = 0;
    for(int i = 1;i <= n; ++ i)
        if(sel[i])ans ++;//统计标记的点的个数
    printf("%lld\n", ans);
    return 0;
}

做完这道题,我们可以总结一点点对于树上dp这一类题的经验技巧:

1.将特殊点作为根,建立一棵树,建树一般用双向边,边的条数严格等于点的个数-1。

2.优先考虑树中特殊的点,比如根、叶子

3.不要忘记考虑特殊情况,比如一条链状的树(此时注意根是否会被判定为叶子、注意复杂度是否会爆)、仅有1个点的树(可能需要特判)。

tokitsukaze and Soldier

题目传送门:https://ac.nowcoder.com/acm/problem/50439

这题主要考察贪心+优先队列维护区间k个最值。

我们观察题目可以发现,当我们枚举到一个士兵的要求是"队伍人数不能超过s[i] = k"时,那么此时军队的战斗力最大值应该是所有s >= k的军人中的最大的k个军人的战斗力之和。

我们可以考虑用一个优先队列维护最大的k个值之和(小根堆,每次弹出最小值,即可维护最大值之和),然后通过限制“遍历方式”使得当遍历到第i个人(s[i] = k)时,优先队列里的所有值对应的s都是>= k的,这样就只需要求出优先队列里最大的k个数字之和即可,这个遍历方式就是按照s[i]降序排列,然后从前往后遍历。

我们可以维护一个大小始终等于s[i]的优先队列,因为s[i]为降序,所以这个优先队列的元素个数的最大值是一直在减小的,减小的时候只需要弹出最小的元素即可,用一个变量sum维护优先队列里的所有元素之和(push时加上,pop时减去)。

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxn = 1e5 + 9;
struct Node
{
    int v, s;
}a[maxn];

signed main()
{
    int n;scanf("%lld", &n);
    
    for(int i = 1;i <= n; ++ i)scanf("%lld %lld", &a[i].v, &a[i].s);
    
    sort(a + 1,a + 1 + n, [](const Node &u, const Node &v)
         {
             return u.s > v.s;
         });
    
    priority_queue<int, vector<int>, greater<int> > pq;
    int ans = 0, sum = 0;
    for(int i = 1;i <= n; ++ i)
    {
        sum += a[i].v, pq.push(a[i].v);
        while(pq.size() > a[i].s)sum -= pq.top(), pq.pop();
        ans = max(ans, sum);
    }
    printf("%lld\n", ans);
    
    return 0;
}

经验总结:

1.用某种遍历方式来限制某个条件,比如本题用"降序"来限制在当前点之前的所有点的s[i]都比当前的大或相等。

2.优先队列可以维护区间的k个最值之和,只适用于连续的查询且k只能变小或不变,因为变大的话,不知道要将哪一个放进去。

最后

感谢大家的阅读,欢迎大家跟我一起刷题呀!

🎈 本文由eriktse原创,创作不易,如果对您有帮助,欢迎小伙伴们点赞👍、收藏⭐、留言💬

热门相关:仙城纪   大神你人设崩了   网游之逆天飞扬   刺客之王   第一神算:纨绔大小姐