AC代码如下
#include <iostream>
#include <algorithm>
#include <map>
#include <set>
using namespace std;
multimap<int,int> f;
set<int>ss;
int main()
{
int n=0;
cin>>n;
while(n--)
{
int n1=0;
cin>>n1;
ss.clear();f.clear();
while(n1--)
{
int x,y;
cin>>x>>y;
f.insert(make_pair(y,x));ss.insert(y);
}
multimap<int,int>::iterator m;
int first=1;double judge1=0;double judge2=0;int gg=0;
for(set<int,int>::iterator it=ss.begin();it!=ss.end();it++)
{
int b=int(f.count(*it));int all=0;
m=f.find(*it);
for(int k=0;k!=b;k++,m++)
{int bb=m->second;all=all+bb;}
if(first) {first=0;judge1=(1.0*all)/(1.0*b);}
judge2=(1.0*all)/(1.0*b);
if(judge1==judge2) gg=1;
else {gg=0;cout<<"NO"<<endl;break;}
}
if(gg) cout<<"YES"<<endl;
}
return 0;
}
算法:
1、先是输入部分,输入每组坐标,然后把纵坐标存在集合中,用make_pair构造一个二元组存储坐标到multimap中(multimap是一个一对多的映射关系)。
2、设置一个集合迭代器it和一个映射迭代器m。
3、从集合中提取出纵坐标,统计纵坐标相等的横坐标的个数,然后将这些横坐标提取相加,除以对应横坐标的个数得到对称轴的x值(注意这个x可能是浮点数,要用double声明)。
4、设置first变量,judge1存储第一组的对称轴x值,judge2存储每一组的对称轴x值,如果不相等break输出NO,如果一直都是相等,输出YES
学到的东西:
1、multimap的使用:
首先设置迭代器
multimap<int,int>::iterator m;
然后对特定元素个数进行统计:
int b=int(f.count(*it));
这里的*it是集合中的元素,需要统计出这一元素对应的个数作为循环条件使用。
然后对特定元素进行查找
m=f.find(*it);
最后提取出该元素对应的second量,即横坐标
for(int k=0;k!=b;k++,m++)
{int bb=m->second;all=all+bb;}
具体细节可以参考:
https://zhidao.baidu.com/question/918049459459983899.html
以及:
https://blog.csdn.net/xy_cpp/article/details/80426852?depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-3&utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-3
这些是multimap用法比较好的参考资料
总结:算法是要经得起推敲的,要确定算法的可实现性,需要考虑多种特殊情况,多拿纸和笔算一算,确定了可行性后再去实现,然后能简化的部分尽量简化,比如用set代替vector,避免了重复元素的运算,加快了速度。