【CF】Day66——Edu 168.D + CF 853 (Div. 2).C (树 + 二分 + 贪心 | 组合数学)

D. Maximize the Root

题目:

思路:

树上二分,中下题

我们可以发现如果 x 可以,那么 x - 1 肯定也可以,所以可以直接二分答案

具体的,我们每次二分能增加的值 mid ,如果 a[i] < mid,那么子树就要 a[i] + a[i] - mid 个,否则直接递归子树即可,以此类推,具体实现看代码

代码:

cpp 复制代码
#include <iostream>
#include <algorithm>
#include<cstring>
#include<cctype>
#include<string>
#include <set>
#include <vector>
#include <cmath>
#include <queue>
#include <unordered_set>
#include <map>
#include <unordered_map>
#include <stack>
#include <memory>
using namespace std;
#define int long long
#define yes cout << "Yes\n"
#define no cout << "No\n"

void solve()
{
    int n;
    cin >> n;
    vector<int> a(n + 1);
    vector<vector<int>> g(n + 1);
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
    }
    for (int i = 2; i <= n; i++)
    {
        int p; cin >> p;
        g[p].push_back(i);
    }
    if (n == 1)
    {
        cout << a[1] << endl;
        return;
    }
    auto check = [&](int need) ->bool{
        int flag = 1;
        auto dfs = [&](auto self,int fa,int needval) ->void{
            if (!flag)
            {
                return;
            }
            if (fa != 1)
                needval += max(needval - a[fa], 0LL);
            if (needval > a[fa] && g[fa].empty() || needval > 1e9)
            {
                flag = 0;
                return;
            }
            for (auto& son : g[fa])
            {
                self(self, son, needval);
            }
            };
        dfs(dfs, 1, need);
        return flag;
        };
    int l = 0, r = 1e9;
    while (l + 1 < r)
    {
        int mid = l + r >> 1;
        if (check(mid))
        {
            l = mid;
        }
        else
        {
            r = mid;
        }
    }
    if (check(r))
    {
        cout << a[1] + r << endl;
        return;
    }
    cout << a[1] + l << endl;
}

signed main()
{
    cin.tie(0)->sync_with_stdio(false);
    int t = 1;
    cin >> t;
    while (t--)
    {
        solve();
    }
    return 0;
}

C. Serval and Toxel's Arrays

题目:

思路:

很考验实现方式的一题

遇到这种题,我们要知道拆分,即把每个数的奉献算出来再累加即可

对于一个数 x,如果 我们选了一个 没有 x 的数组,那么奉献就是 1,取有 x 的数列,奉献也是 1,但是这是两种不同的取法,所以我们可以分开计算(特别的一共有 m + 1个数组,因为还有初始数组)

对于第一种取法,那么就是 cnt += xhas * (m + 1 - xhas),其中 xhas 为 x 的数量

对于第二种取法,有 cnt += xhas * (xhas-1) / 2

由于 n + m 不是很大,所以枚举 n+m 的每一个数是可行的,那么如何快速计算 xhas 成了问题

我们可以这样想,由于每次只改变一个数,那么也就是说如果某个数一直没改变,那么最后肯定有 m + 1 个,而如果中间改变过,那么肯定在改变之前这个数就一直存在了,也就是连续的某一段全含有 x,所以我们可以存储 last[i] 代表上一个数是否存在,以及如果存在它的位置在哪里

这样我们就解决了 xhas 的问题,具体实现看代码

代码:

cpp 复制代码
#include <iostream>
#include <algorithm>
#include<cstring>
#include<cctype>
#include<string>
#include <set>
#include <vector>
#include <cmath>
#include <queue>
#include <unordered_set>
#include <map>
#include <unordered_map>
#include <stack>
#include <memory>
using namespace std;
#define int long long
#define yes cout << "Yes\n"
#define no cout << "No\n"

void solve()
{
    int n, m;
    cin >> n >> m;
    vector<int> a(n+1),last(n+m+1,-1),cnt(n+m+1,0);
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
        last[a[i]] = 0;
    }
    for (int i = 1; i <= m; i++)
    {
        int p, v;
        cin >> p >> v;
        if (a[p] != v)
        {
            cnt[a[p]] += i - last[a[p]];
            last[a[p]] = -1;
            last[v] = i;
            a[p] = v;
        }
    }
    int res = 0;
    for (int i = 1;i <= n+m;i++)
    {
        if (last[i] != -1)
        {
            cnt[i] += m - last[i] + 1;
        }
    }
    for (int i = 1; i <= n + m; i++)
    {
        res += cnt[i] * (m + 1 - cnt[i]);
        res += cnt[i] * (cnt[i] - 1) / 2;
    }
    cout << res << endl;
}

signed main()
{
    cin.tie(0)->sync_with_stdio(false);
    int t = 1;
    cin >> t;
    while (t--)
    {
        solve();
    }
    return 0;
}
相关推荐
DIY机器人工房16 分钟前
一个可以检测本机的字节顺序,并对任意数据进行字节顺序的反转操作的代码。
嵌入式硬件·算法·嵌入式·diy机器人工房
杰克尼2 小时前
11. 盛最多水的容器
算法·leetcode·职场和发展
程序员编程指南3 小时前
Qt 嵌入式界面优化技术
c语言·开发语言·c++·qt
程序员Xu4 小时前
【OD机试题解法笔记】查找接口成功率最优时间段
笔记·算法
云泽8084 小时前
数据结构前篇 - 深入解析数据结构之复杂度
c语言·开发语言·数据结构
技术思考者4 小时前
Leetcode - 反转字符串
数据结构·算法·leetcode
程序员编程指南5 小时前
Qt 开发 IDE 插件开发指南
c语言·c++·ide·qt·elasticsearch
SKYDROID云卓小助手5 小时前
无人设备遥控器之多设备协同技术篇
网络·人工智能·嵌入式硬件·算法·信号处理
熬了夜的程序员6 小时前
【华为机试】34. 在排序数组中查找元素的第一个和最后一个位置
数据结构·算法·华为od·华为·面试·golang
laocooon5238578866 小时前
C语言画表格
c语言·开发语言