题目大意
题目中给出一组点(横纵坐标均为整数,且范围很小),问其中存在的正多边形的个数。
解题思路
由于题目中并没有明确的指出横纵坐标都是整数(题目中明确只说了two numbers),比赛的时候并没有注意到这一点,以为给的都是浮点数,所以按照普适的方法做的。
先讲一下普遍的方法
任选一条边,从该边的一个端点出发,每次均寻找该端点连接到的边且边长与现有边长度相等,将该点入栈,直到某个点能够连到起始边的另一个端点结束,将cont数组中N边形的数目增加1,当以所有的边为起始点循环一次后,注意所有K边形的数量都被统计了K次,因为以其不同的边为起始点都统计了一次,故除以K后得到正确答案。
但是由于坐标都用double的话会出现很多的精度问题,今天二分查找的过程中lower_bound就踩了不少坑(由于乱加提高精度的小数),如果今后出了这种题估计精度会是个大坑吧。。。
在执行的过程中注意在几个地方剪枝可以少走很多弯路
1. 新加入的边长度必须等于起始边的长度
2. 当栈中点数大于等于3时,新加入的点与(top-2)位置点的距离等于dis(top-1,top-3),栈顶元素为top-1
3.可以将每个点连接的边长排序,二分的搜索所有满足条件的边长
只有条件一的话会将很多角度不对的边也入栈,增加第二个条件后可以将角度也确定,不太明白的同学可以手动画图模拟一下。这里不剪枝的话很有可能会TLE,剪过之后跑了873MS
然后讲一下正式的解法
由 NOI 2017的一篇论文里指出(144页)关于正多边形的结论,即只有正方形能满足其所有顶点坐标均为整数。所以只需要考虑能形成多少个正方形就可以了(=,=),有兴趣的同学可以去看一下国家集训队的论文。
小小的吐槽一下,6min的时候发现清华飞快的A了这个题,但是当我写出了160行的代码后我整个人都懵了(虽然我知道他们肯定没写我这种冗长的代码)。。6min连代码都写不完的我。。果然姿势水平有待提高阿。。
最后贴上(辣眼睛的)代码
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
const double eps=1e-6;
const ll mx=550;
double x[mx],y[mx],dis[mx][mx];
int num[mx][mx],vis[mx],cont[mx],flag=0,cnt=0,n;
bool ok=false;
struct edge
{
int l,r,id;
double len;
edge(){}
edge(int l1,int rr,int le,int ii)
{
l=l1;r=rr;len=le;id=ii;
}
bool operator <(const edge &aa)const
{
return len < aa.len+eps;
}
}ed[mx*mx];
int sta[mx<<2];
vector<edge> graph[mx];
bool cmp(const edge &aa,const edge &bb)
{
return aa.len<bb.len;
}
bool check2(int bb)
{
double len1=dis[sta[flag-1]][sta[flag-3]];
double len2=dis[sta[flag-2]][bb];
if(len1<len2)
swap(len1,len2);
if(len1-len2<=eps)
return true;
return false;
}
void dfs(int al,int now)
{
double len1=dis[sta[flag-1]][sta[flag-2]];
edge newedge=edge(0,0,len1,-1);
int pos=upper_bound(graph[now].begin(),graph[now].end(),newedge)-graph[now].begin();
int rr;
while(pos<graph[now].size())
{
rr=graph[now][pos].r;
if(fabs(graph[now][pos].len-newedge.len)>eps)
break;
if(vis[rr])
{
if(al>1&&rr==sta[0])
{
cont[al+1]++;
ok=true;
break;
}
else
{
pos++;
continue;
}
}
if(flag>=3)
{
if(check2(rr))
{
vis[rr]=1;
sta[flag++]=rr;
dfs(al+1,rr);
flag--;
}
}
else
{
vis[rr]=1;
sta[flag++]=rr;
dfs(al+1,rr);
flag--;
}
pos++;
}
}
void solve()
{
for(int i=0;i<n;i++)
{
for(int j=i+1;j<n;j++)
{
ok=false;
flag=0;
memset(vis,0,sizeof(vis));
vis[i]=vis[j]=1;
sta[flag++]=i;
sta[flag++]=j;
dfs(1,j);
}
}
}
int main()
{
while(~scanf("%d",&n))
{
cnt=0;
memset(dis,0,sizeof(dis));
memset(num,-1,sizeof(num));
memset(cont,0,sizeof(cont));
for(int i=0;i<n;i++)
graph[i].clear();
for(int i=0;i<n;i++)
{
scanf("%lf%lf",&x[i],&y[i]);
}
for(int i=0;i<n;i++)
{
for(int j=i+1;j<n;j++)
{
ed[cnt].l=i;
ed[cnt].r=j;
double tmp=(x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]);
ed[cnt].len=tmp;
dis[i][j]=dis[j][i]=tmp;
num[i][j]=num[j][i]=cnt;
ed[cnt].id=cnt;
edge new1=edge(i,j,tmp,cnt);
graph[i].push_back(new1);
edge new2=edge(j,i,tmp,cnt);
graph[j].push_back(new2);
cnt++;
}
}
for(int i=0;i<n;i++)
sort(graph[i].begin(),graph[i].end());
solve();
int ans=0;
for(int i=3;i<=n;i++)
{
if(cont[i])
ans+=cont[i]/i;
}
printf("%d\n",ans);
}
return 0;
}