L2-028 秀恩爱分得快 (25 分)（思维结构体 +1-CSDN博客

该程序读取输入的字符串集合，计算特定字符串出现的概率，并寻找具有最高概率的相关字符串组合。它处理负数并进行概率加权，输出可能的解决方案。主要涉及字符串操作、概率计算和集合操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<bits/stdc++.h>
using namespace std;
const int N=1e4+9;
set<string>s[N],in;
struct node
{
    string name;
    double p;
    bool operator <(const node &x)const{
        return p>x.p;
    }
}ta[N],tb[N];
int sexa,sexb;
int main(){
    int n,m;
    cin>>n>>m;
    for(int j=1;j<=m;j++){
        int k;
        cin>>k;
        for(int i=1;i<=k;i++){
            string p;
            cin>>p;
            in.insert(p);
            s[j].insert(p);
        }
    }
    string a,b;
    cin>>a>>b;
    int A = abs(stoi(a)), B = abs(stoi(b));
    if(a[0]=='-')sexa=1;
    if(b[0]=='-')sexb=1;
    double mxa=0,mxb=0;
    for(int i=1;i<=m;i++){
        
        if(s[i].find(a)!=s[i].end()||s[i].find(b)!=s[i].end()){
            int checka=0,checkb=0;
            if(s[i].find(a)!=s[i].end())checka=1;
            if(s[i].find(b)!=s[i].end())checkb=1;

            for(auto it:s[i]){
                int nea=0,neb=0;
                int p=abs(atoi(it.c_str()));
                if(it[0]=='-'){
                    nea=1;
                }
                if(nea!=sexa&&checka){
                    ta[p].name=it;//!!会默认初始化p
                    // cout<<ta[p].p<<"/*/\n";
                    // cout<<1.0/s[i].size()<<"\n";
                    ta[p].p+=1.0/s[i].size();
                    mxa=max(ta[p].p,mxa);
                    // cout<<ta[p].p<<"/*/\n";
                }

                
                if(nea!=sexb&&checkb){
                    tb[p].name=it;
                    tb[p].p+=1.0/s[i].size();
                     mxb=max(tb[p].p,mxb);
                    // cout<<p<<" ";
                    // cout<<tb[p].p<<"/*/\n";
                }
            }
        }
    }
    // cout<<"*--\n";
    // sort(ta,ta+1+1000);
    // sort(tb,tb+1+1000);
    vector<string>ansa,ansb;
    double pa=ta[0].p,pb=tb[0].p;
    // cout<<pa<<" "<<pb<<"\n";
    int oka=0,okb=0;
    if(ta[B].p==mxa&&tb[A].p==mxb){
        cout<<a<<" "<<b<<"\n";
        return 0;
    }

    for(auto it:ta){
        if(mxa==it.p&&mxa>0){
            ansa.push_back(it.name);
            if(it.name==b)oka=1;
        }
    }
    for(auto it:tb){
        if(mxb==it.p&&mxb>0){
            ansb.push_back(it.name);
            if(it.name==a)okb=1;
        }
    }
    
    // if(in.find(a)==in.end()&&in.find(b)==in.end()){
    //     cout<<a<<" "<<b<<"\n";
    //     return 0;
    // }
    // cout<<ansa.size();
//    if(ansa.size()==0)ansa.push_back(b);
//    if(ansb.size()==0)ansb.push_back(a);
    for(auto it:ansa){
        cout<<a<<" "<<it<<"\n";
    }
    for(auto it:ansb){
        cout<<b<<" "<<it<<"\n";
    }



    return 0;
}