【BZOJ3616】War
Description
小x所在的世界正在经历一场在k个阵营之间的战争。每个阵营有若干个炮塔，每个炮塔由攻击系统和防御系统组成。第i个炮塔可以攻击到离它欧几里德距离小于等于ri 或者曼哈顿距离小于等于ai的炮塔，被攻击到的炮塔防御系统就会崩溃，同一联盟的炮塔不会被攻击到。每次会随机选择一个炮塔攻击它能打到的所有炮塔，问进行m轮后期望剩下多少个阵营，使得这些阵营拥有的炮塔的防御系统全部完好。防御系统崩溃的炮塔还是会被打到的。值得注意的是，如果一个联盟没有任何炮塔，那么不管怎样它都是完好的。
Input
输入文件第一行有三个整数n、m、k，分别表示炮塔数目、攻击轮数以及联盟个数。 接下去n行每行有五个正整数xi、yi、ri、ai、pi，其中xi、yi表示炮塔坐标，p表示炮塔所属于阵营。
Output
输出一行一个实数表示答案，你的输出与标准输出的误差在1e-3以内会被认为是正确的。
Sample Input
2 2 3
0 0 2 2 1
1 1 2 2 2
Sample Output
1.500
HINT
100%的数据， 0 <= n、m、k <= 35000，1 <= pi <= k，其余所有数字均非负且小于10000，保证在数据生成时不考虑一个点的具体位置，而将它的横纵坐标分开考虑。
Source
2014年国家集训队十五人互测 鸣谢佚名上传
考虑使用bitset,在kdtree的每个节点维护一个,然后连边,
然后发现其实概率论才是这个题考察的主要内容(雾
注意bitset不能写结构体里面..因为每个bitset都是35000的,然后如果一坨放一起做swap…场面太美,直接TLE
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<bitset>
#define MAXN 35002
#define Dnum 2
#define GET (ch>='0'&&ch<='9')
using namespace std;
inline void in(int &x)
{
char ch=getchar();x=0;
while (!GET) ch=getchar();
while (GET) x=x*10+ch-'0',ch=getchar();
}
double ans;
int N,M,K,n,root,top,tp;
int r[MAXN],a[MAXN],p[MAXN],id[MAXN];
bool cmp_d;
struct edge { int to; edge *next; }e[10000000],*prev[MAXN],E[MAXN<<1],*Prev[MAXN];
inline void insert(int u,int v) { e[++top].to=v;e[top].next=prev[u];prev[u]=&e[top]; }
inline void insert1(int u,int v) { E[++tp].to=v;e[tp].next=Prev[u];Prev[u]=&E[tp]; }
struct KDtree
{
int d[Dnum],ch[2],minn[Dnum],maxn[Dnum],fa;
inline void init() { for (int i=0;i<Dnum;++i) minn[i]=maxn[i]=d[i]; }
inline bool operator < (const KDtree& a)const { return d[cmp_d]<a.d[cmp_d]; }
}tree[MAXN];
bitset<MAXN> P[MAXN];
inline void push_up(int rt)
{
for (int i=0;i<2;++i)
if (tree[rt].ch[i])
{
int x=tree[rt].ch[i];
for (int j=0;j<Dnum;++j)
tree[rt].minn[j]=min(tree[rt].minn[j],tree[x].minn[j]),
tree[rt].maxn[j]=max(tree[rt].maxn[j],tree[x].maxn[j]);
}
}
int rebuild(int l=1,int r=n,bool d=0,int f=0)
{
cmp_d=d;int mid=(l+r)>>1;nth_element(tree+l,tree+mid,tree+r+1);
id[tree[mid].fa]=mid;tree[mid].fa=f;tree[mid].init();
if (l!=mid) tree[mid].ch[0]=rebuild(l,mid-1,d^1,mid);
if (r!=mid) tree[mid].ch[1]=rebuild(mid+1,r,d^1,mid);
return push_up(mid),mid;
}
inline int sqr(int x) { return x*x; }
void calc(int rt,int id,int x,int y)
{
if (sqr(max(max(x-tree[rt].maxn[0],tree[rt].minn[0]-x),0))+sqr(max(max(y-tree[rt].maxn[1],tree[rt].minn[1]-y),0))>r[id]
&&
max(tree[rt].minn[0]-x,0)+max(x-tree[rt].maxn[0],0)+max(tree[rt].minn[1]-y,0)+max(y-tree[rt].maxn[1],0)>a[id]) return;
if (max(sqr(x-tree[rt].minn[0]),sqr(x-tree[rt].maxn[0]))+max(sqr(y-tree[x].minn[1]),sqr(y-tree[x].maxn[1]))<=r[id]
||
max(abs(x-tree[rt].maxn[0]),abs(tree[x].minn[0]-x))+max(abs(y-tree[x].maxn[1]),abs(tree[x].maxn[1]-y))<=a[id])
{ P[rt][id]=1;return; }
for (int i=0;i<2;++i) if (tree[rt].ch[i]) calc(tree[rt].ch[i],id,x,y);
}
void solve(int rt=root)
{
for (int i=0,x=0;i<2;++i) if ((x=tree[rt].ch[i])) P[x]|=P[rt],solve(x);
for (edge *i=prev[rt];i;i=i->next) P[rt][i->to]=1;
}
int main()
{
in(N);in(M);in(K);n=N;int x;
for (int i=1;i<=N;i++)
{
tree[i].fa=i;in(tree[i].d[0]);in(tree[i].d[1]);
in(r[i]);in(a[i]);in(p[i]);
insert1(p[i],i);
}
root=rebuild();
for (int i=1;i<=N;i++) calc(root,i,tree[id[i]].d[0],tree[id[i]].d[1]);
for (solve(),x=1;x<=K;x++)
{
bitset<MAXN> tmp=0;
for (edge *i=Prev[x];i;i=i->next) tmp|=P[i->to];
for (edge *i=Prev[x];i;i=i->next) tmp[i->to]=0;
int cnt=tmp.count();ans+=pow(1-1.0*cnt/N,M);
}
printf("%.6f\n",ans);
}