PTA-L2-028 秀恩爱分得快 (25分)
这道题纯粹的数据结构
我们特别注意0和-0的情况,所以应该字符串读入。
如果把所有异性的亲密度值都算出来,会tle
我们只需要算出关于那一对情侣和异性之间的值即可。
然后注意特殊情况的考虑,如果所给出的一对情侣没有出现在这些照片中,我们只需要输出这对情侣即可
所以我们初始化找最大的亲密度值的时候只能初始化为0,不能是负数(我在这wa了一次)
我们用sex[]来标记男女,女生标记为1.
注意题目中说到的彼此,只有彼此都是对方最亲密的,才输出自己本身,否则其余情况都得输出最高的,比如男方是女方最亲密的,但是男方有更亲密的女生对象,这种,也需要输出其他的(zha nan)
代码部分:
#include <bits/stdc++.h>
#define mst(a, n) memset(a, n, sizeof(a))
using namespace std;
const int N = 1e3 + 10;
const int M = 55;
const int INF = 1e6 + 10;
const double eps = 0.05;
typedef long long ll;
int n, k, m;
vector<int> v[N];
int sex[N];
double ga[N];
double gb[N];
int main()
{
cin >> n >> m;
int i, j, z;
for (i = 0; i < m; i++)
{
scanf ("%d", &k);
for (j = 0; j < k; j++)
{
int siz;
string s;
cin >> s;
if (s[0] == '-')
{
siz = s.size();
string t = "";
for (z = 1; z < siz; z++)
{
t += s[z];
}
v[i].push_back(atoi(t.c_str()));
sex[v[i][j]] = 1;
}
else
{
v[i].push_back(atoi(s.c_str()));
}
}
}
int a, b;
string s1, s2;
cin >> s1 >> s2;
int siz1, siz2;
siz1 = s1.size();
siz2 = s2.size();
if (s1[0] == '-')
{
string t = "";
for (i = 1; i < siz1; i++)
{
t += s1[i];
}
a = atoi(t.c_str());
sex[a] = 1;
}
else
{
a = atoi(s1.c_str());
}
if (s2[0] == '-')
{
string t = "";
for (i = 1; i < siz2; i++)
{
t += s2[i];
}
b = atoi(t.c_str());
sex[b] = 1;
}
else
{
b = atoi(s2.c_str());
}
double maxa = 0;
double maxb = 0;
for (i = 0; i < m; i++)
{
int k1 = find(v[i].begin(), v[i].end(), a) != v[i].end();
int k2 = find(v[i].begin(), v[i].end(), b) != v[i].end();
if (k1 || k2)
{
int siz = v[i].size();
for (j = 0; j < siz; j++)
{
if (k1 && sex[a] != sex[v[i][j]])
{
ga[v[i][j]] += 1.0 / siz;
if (ga[v[i][j]] > maxa)
{
maxa = ga[v[i][j]];
}
}
if (k2 && sex[b] != sex[v[i][j]])
{
gb[v[i][j]] += 1.0 / siz;
if (gb[v[i][j]] > maxb)
{
maxb = gb[v[i][j]];
}
}
}
}
}
if (maxa == ga[b] && maxb == gb[a])
{
if (sex[a])
{
cout << "-";
}
cout << a << " ";
if (sex[b])
{
cout << "-";
}
cout << b << endl;
}
else
{
for (i = 0; i < n; i++)
{
if (maxa == ga[i])
{
if (sex[a])
{
cout << "-";
}
cout << a << " ";
if (sex[i])
{
cout << "-";
}
cout << i << endl;
}
}
for (i = 0; i < n; i++)
{
if (maxb == gb[i])
{
if (sex[b])
{
cout << "-";
}
cout << b << " ";
if (sex[i])
{
cout << "-";
}
cout << i << endl;
}
}
}
return 0;
}