L2-028秀恩爱分得快(25 分) atoi()和stoi()、vector

题目

L2-028 秀恩爱分得快 (25 分)
给出大量照片，可分析人与人之间的亲密度。一张照片上出现了 K 个人，这些人两两间的亲密度就是 1/K。任意两个人同时出现在若干张照片里，他们的亲密度就是所有这些同框照片对应的亲密度之和。分析一对情侣有没有亲密度更高的异性

输入：
第一行有2 个正整数：N（<1000，总人数 0 到 N-1 编号。负号表示女性）和 M（<1000是照片总数）， M 行给出每一张照片的信息
K P[1] … P[K]
K（≤ 500）是该照片中出现的人数，P[1] ~ P[K] 是编号。最后一行给出一对异性情侣编号 A 和 B。

输出：
先输出 A PA，PA 是与 A 最亲密的异性。PA 不唯一的话，就按他们编号的绝对值递增输出；然后输出 B PB。若A 、 B 是彼此亲密度最高的一对，则只输出他们的编号，不管是否有人并列。

10 4
4 -1 2 -3 4
4 2 -3 -5 -6
3 2 4 -5
3 -6 0 2
-3 2

-3 2
2 -5
2 -6

atoi()和stoi()

atoi()的参数是 const char* ,因此对于一个字符串str我们必须调用 c_str()的方法把这个string转换成 const char类型的
stoi()的参数是const string,不需要转化为 const char*；直接把string转化为int

代码

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
int n,m,a,b,t,x,y;
string s;//字符串读入,用于判断"-0".用int无法判断-0的情况
vector<int> v[1010];//记录照片信息
double sum[1010][1010];//亲密度总和
int sex[1010];//1->男 -1->女

void O(int x){//因为输出"-0" 要特判
    if(x==0 && sex[0]==-1) cout<<'-';
    cout<<sex[abs(x)]*abs(x);//这样子输入普通i,也能输出正确的性别
}
int work(int i,int a){//有无该情侣,有的话计算亲密度总和
    auto q = lower_bound(v[i].begin(),v[i].end(),a);
    int tmp = 0;
    if(*q==a){//在照片中有a这个人
            x = sex[abs(a)];
        for(int j=0;j<v[i].size();j++){
            y = sex[abs(v[i][j])];
            if(x*y<0){//异性才计算亲密度
                sum[abs(a)][abs(v[i][j])]+=1.0/(v[i].size()*1.0);
                sum[abs(v[i][j])][abs(a)]+=1.0/(v[i].size()*1.0);
            }
        }
        return 1;
    }
    return 0;
}

int main(){
    cin>>n>>m;
    for(int i=1;i<=m;i++){
        cin>>a;
        for(int j=1;j<=a;j++){
            cin>>s;
            t = stoi(s);//string 转 int
            v[i].push_back(t);
            sex[abs(t)]=s[0]=='-'?-1:1;//判断男女 
        }
        sort(v[i].begin(),v[i].end());
    }//坑点,性别信息可能不在照片里,在给出的情侣里
    cin>>s; a=stoi(s); sex[abs(a)]=s[0]=='-'?-1:1; 
    cin>>s; b=stoi(s); sex[abs(b)]=s[0]=='-'?-1:1;
    for(int i=1;i<=m;i++){
        int tmp = 0;
        tmp += work(i,a);
        tmp += work(i,b);
        if(tmp==2){//去掉重复相加的
            sum[abs(a)][abs(b)]-=1.0/(v[i].size()*1.0);
            sum[abs(b)][abs(a)]-=1.0/(v[i].size()*1.0);
        }
    }
    double Max1=0,Max2=0;
    for(int i=0;i<n;i++) 
        Max1=max(Max1,sum[abs(a)][i]);
    for(int i=0;i<n;i++) 
        Max2=max(Max2,sum[abs(b)][i]);
    if(Max1 == Max2 && sum[abs(a)][abs(b)]==Max1) 
        O(a),cout<<" ",O(b),cout<<endl;
    else{
        for(int i=0;i<n;i++)
            if(sex[abs(a)]*sex[i]<0 && sum[abs(a)][i]==Max1) 
                O(a),cout<<" ",O(i),cout<<endl;
        for(int i=0;i<n;i++)
            if(sex[abs(b)]*sex[i]<0 && sum[abs(b)][i]==Max2) 
                O(b),cout<<" ",O(i),cout<<endl;
    }
}