T3
计算几何妙哉
对于每个发射源,将激光塔和敌人按极角排序
答案用前缀和求出
有可能出现激光塔在发射源下方使得射不到敌人的情况
需要用到环形DP
选发射源
O
(
n
)
O(n)
O(n)
选激光塔
O
(
n
2
)
O(n^2)
O(n2)
排序大概
O
(
log
n
)
O(\log n)
O(logn)
考虑优化
发现激光塔可以优化
对于一个发射源,
a
i
a_i
ai表示射线往左敌人的个数
它对答案的贡献:
a
n
s
=
∑
i
=
2
n
[
(
i
−
1
)
a
i
−
∑
j
=
1
i
−
1
a
j
]
ans=\sum_{i=2}^n\big[(i-1)a_i-\sum_{j=1}^{i-1}a_j \big]
ans=i=2∑n[(i−1)ai−j=1∑i−1aj]
后面那一坨前缀和优化
总共 O ( n 2 log n ) O(n^2\log n) O(n2logn)
以上是考场想法
下面讲正解
坐标不需要double,但叉乘的时候会炸int,所以用long long
对于每个源,对塔和敌人排序
O
(
n
2
log
n
)
O(n^2\log n)
O(n2logn)
可能出现上述的头和尾构成方案,考虑环状处理
由于可能是头几个和尾几个构成方案,需要复制一份完整的数据
转起来统计: O ( n ) O(n) O(n)
- 塔:塔数++,加一遍目前的敌数;
- 敌:敌数++;
统计答案:
遇到塔统计
考虑使用双指针i,j(指针指向元素记录了前缀和)
要求:(1) 两塔向量夹角在
π
\pi
π以内;(2) 两个指针指向的向量需要定向
一通疯狂移动(j)之后得到的两根指针使用前缀和即可求出答案
O
(
n
)
O(n)
O(n)
容斥一下,发现还需要减去i以前的敌人数sd[i]*(st[j]-st[i])
总计 O ( n 2 log n ) O(n^2\log n) O(n2logn)
ACcode
注意数据类型
#include<bits/stdc++.h>
using namespace std;
#define in Read()
#define debug puts("qwq")
typedef long long ll;
ll in{
ll i=0,f=1;char ch=0;
while(!isdigit(ch)&&ch!='-') ch=getchar();
if(ch=='-') ch=getchar(),f=-1;
while(isdigit(ch)) i=(i<<1)+(i<<3)+ch-48,ch=getchar();
return i*f;
}
const int N=5e3+5;
ll D,S,T,ans;
struct point{
ll x,y;
point(){}
point(ll _x,ll _y){x=_x;y=_y;}
friend inline point operator + (const point &a,const point &b){return point(a.x+b.x,a.y+b.y);}
friend inline point operator - (const point &a,const point &b){return point(a.x-b.x,a.y-b.y);}
friend inline ll operator * (const point &a,const point &b){return a.x*b.y-a.y*b.x;}
}d[N],s[N],t[N];
#define vect point
struct ray{
vect vec;
bool type;
ray(){}
ray(vect _vec,bool _type){vec=_vec;type=_type;}
}v[N];
bool cmp(const ray &a,const ray &b){return a.vec.w>b.vec.w;}
int side(vect u,vect v){
if(u*v>0ll) return 1;
else if(u*v<0ll) return -1;
else return 0;
}
int sd[N],st[N],sum[N];
void solve(point p){
for(int i=1;i<=D;++i) v[i]=ray(d[i]-p,0);
for(int i=1;i<=T;++i) v[i+D]=ray(t[i]-p,1);
sort(v+1,v+D+T+1,cmp);
for(int i=1;i<=D+T;++i) v[i+D+T]=v[i];
for(int i=1;i<=((D+T)<<1);++i){
sd[i]=sd[i-1];
st[i]=st[i-1];
sum[i]=sum[i-1];
if(v[i].type){
++st[i];
sum[i]+=sd[i];
}else ++sd[i];
}
for(int i=1,j=1;i<=D+T;++i)
if(v[i].type){
j=max(i,j);
while(j+1<i+D+T&&side(v[i].vec,v[j+1].vec)<=0) ++j;
ans+=sum[j]-sum[i]-sd[i]*(st[j]-st[i]);
}
return;
}
int main(){
D=in;
for(int i=1;i<=D;++i){
ll x=in,y=in;
d[i]=point(x,y);
}
S=in;
for(int i=1;i<=S;++i){
ll x=in,y=in;
s[i]=point(x,y);
}
T=in;
for(int i=1;i<=T;++i){
ll x=in,y=in;
t[i]=point(x,y);
}
for(int i=1;i<=S;++i) solve(s[i]);
printf("%lld\n",ans);
return 0;
}