小R的并集大小期望计算 | 蛮力

问题描述

小R有 n 个集合,她想通过随机选择两个集合,并计算它们的并集,来求出这个并集大小的期望值。每个集合中的元素都是唯一的且互不相同。她需要计算出随机选择两个集合并集大小的期望值,并且要求结果保留两位小数。

保证输入至少有两个集合。

测试样例

样例1:

输入:n = 2,st = [[1, 2], [1, 3, 5]]

输出:'4.00'

样例2:

输入:n = 3,st = [[1, 4], [2, 5], [3, 6, 7]]

输出:'4.67'

样例3:

输入:n = 2,st = [[10, 20, 30], [10, 30, 50, 70]]

输出:'5.00'

题解:

题目不是很难,双重遍历整个数组,选取其中两个,并再次遍历其中数据存入set中,就可方便计算出其个数,然后记录总个数与选取个数,相除即为期望。

但是本题的难点在于,怎么将一个double数据转换成string数据,我也不知道,后来查了才发现一个stringstream的类型,可以通过ss << fixed << d来将double转换成string,并使用setprecision来控制小数位数即可。

代码:

cpp 复制代码
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<iomanip>
#include<queue>
#include<stack>
#include<vector>
#include<unordered_set>
#include<unordered_map>
#include<map>
#include<set>
using namespace std;
typedef long long int ll;

string solution(int n, vector<vector<int>> st) {
    set<int> st1;
    int sum=0;
    int num=0;
    for(int i=0;i<n-1;i++){
        for(int j=i+1;j<n;j++){
            vector<int> vt1=st[i];
            vector<int> vt2=st[j];
            for(int k=0;k<vt1.size();k++){
                st1.insert(vt1[k]);
            }
            for(int k=0;k<vt2.size();k++){
                st1.insert(vt2[k]);
            }
            sum+=st1.size();
            num++;
            vt1.clear();vt2.clear();st1.clear();
        }
    }
    double as=0;
    string ans="";
    as=sum*1.0/num;
    //cout << as << "\n";
    stringstream ss;
    ss.setf(ios::fixed);
    ss << setprecision(2) << as;
    //cout << ss.str() << "\n";
    return ss.str();
}

int main() {
    cout << fixed << setprecision(2) << (solution(2, {{1, 2}, {1, 3, 5}}) == "4.00") << endl;
    cout << fixed << setprecision(2) << (solution(3, {{1, 4}, {2, 5}, {3, 6, 7}}) == "4.67") << endl;
    cout << fixed << setprecision(2) << (solution(2, {{10, 20, 30}, {10, 30, 50, 70}}) == "5.00") << endl;
    return 0;
}
相关推荐
Two_brushes.1 小时前
【算法】宽度优先遍历BFS
算法·leetcode·哈希算法·宽度优先
森焱森3 小时前
水下航行器外形分类详解
c语言·单片机·算法·架构·无人机
QuantumStack5 小时前
【C++ 真题】P1104 生日
开发语言·c++·算法
写个博客5 小时前
暑假算法日记第一天
算法
绿皮的猪猪侠5 小时前
算法笔记上机训练实战指南刷题
笔记·算法·pta·上机·浙大
hie988946 小时前
MATLAB锂离子电池伪二维(P2D)模型实现
人工智能·算法·matlab
杰克尼6 小时前
BM5 合并k个已排序的链表
数据结构·算法·链表
.30-06Springfield7 小时前
决策树(Decision tree)算法详解(ID3、C4.5、CART)
人工智能·python·算法·决策树·机器学习
我不是哆啦A梦7 小时前
破解风电运维“百模大战”困局,机械版ChatGPT诞生?
运维·人工智能·python·算法·chatgpt
xiaolang_8616_wjl7 小时前
c++文字游戏_闯关打怪
开发语言·数据结构·c++·算法·c++20