BZOJ4481: [Jsoi2015]非诚勿扰【概率期望+树状数组】

Description

【故事背景】
JYY赶上了互联网创业的大潮，为非常勿扰开发了最新的手机App实现单身
大龄青年之间的“速配”。然而随着用户数量的增长，JYY发现现有速配的算法似
乎很难满足大家的要求，因此JYY决定请你来调查一下其中的原因。
【问题描述】
应用的后台一共有N个女性和M个男性，他们每个人都希望能够找到自己的
合适伴侣。为了方便，每个男性都被编上了1到N之间的一个号码，并且任意两
个人的号码不一样。每个女性也被如此编号。
JYY应用的最大特点是赋予女性较高的选择权，让每个女性指定自己的“如
意郎君列表”。每个女性的如意郎君列表都是所有男性的一个子集，并且可能为
空。如果列表非空，她们会在其中选择一个男性作为自己最终接受的对象。
JYY用如下算法来为每个女性速配最终接受的男性：将“如意郎君列表”中的
男性按照编号从小到大的顺序呈现给她。对于每次呈现，她将独立地以P的概率
接受这个男性（换言之，会以1−P的概率拒绝这个男性）。如果她选择了拒绝，
App就会呈现列表中下一个男性，以此类推。如果列表中所有的男性都已经呈现，
那么中介所会重新按照列表的顺序来呈现这些男性，直到她接受了某个男性为止。
显然，在这种规则下，每个女性只能选择接受一个男性，而一个男性可能被多个
女性所接受。当然，也可能有部分男性不被任何一个女性接受。
这样，每个女性就有了自己接受的男性（“如意郎君列表”为空的除外）。现
在考虑任意两个不同的、如意郎君列表非空的女性a和b，如果a的编号比b的编
号小，而a选择的男性的编号比b选择的编号大，那么女性a和女性b就叫做一对
不稳定因素。
由于每个女性选择的男性是有一定的随机性的，所以不稳定因素的数目也是
有一定随机性的。JYY希望你能够求得不稳定因素的期望个数（即平均数目），
从而进一步研究为什么速配算法不能满足大家的需求。

Input

输入第一行包含2个自然数N,M，表示有N个女性和N个男性，以及所有女
性的“如意郎君列表”长度之和是M。
接下来一行一个实数P，为女性接受男性的概率。
接下来M行，每行包含两个整数a,b，表示男性b在女性a的“如意郎君列表”
中。
输入保证每个女性的“如意郎君列表”中的男性出现切仅出现一次。
1≤N,M≤500,000，0.4≤P<0.6

Output

输出1行，包含一个实数，四舍五入后保留到小数点后2位，表示不稳定因素的期望数目。

Sample Input

5 5
0.5
5 1
3 2
2 2
2 1
3 1

Sample Output

0.89

---

思路

直接考虑每个男人的期望贡献就可以了

用树状数组来维护

直接把每个男人的被选概率算出来统计答案

这东西推一下等比数列求和就可以了

水题

不过要卡精度，勇士请使用long double

---

#include<bits/stdc++.h>

using namespace std;

typedef long double lb;
const int N = 1e6 + 10;

int n, m;
lb bit[N], p;
vector<int> g[N];

lb fast_pow(lb a, int b) {
  lb res = 1.0;
  while (b) {
    if (b & 1) res *= a;
    b >>= 1;
    a *= a; 
  } 
  return res;
}

void add(int x, lb val) {
  while (x < N) {
    bit[x] += val;
    x += x & (-x);
  }
}

lb query(int x) {
  lb res = 0;
  while (x) {
    res += bit[x];
    x -= x & (-x);
  }
  return res;
}

lb query(int l, int r) {
  return query(r) - query(l - 1);
}

int main() {
  scanf("%d %d", &n, &m);
  scanf("%Lf", &p);
  for (int i = 1; i <= m; i++) {
    int u, v; scanf("%d %d", &u, &v);
    g[u].push_back(v);
  }
  for (int i = 1; i <= n; i++) {
    sort(g[i].begin(), g[i].end());
  }
  lb ans = 0.0;
  for (int i = 1; i <= n; i++) {
    int len = g[i].size();
    for (int j = 0; j < len; j++) {
      lb cur = p * fast_pow(1.0 - p, j) / (1.0 - fast_pow(1.0 - p, len));
      ans += query(g[i][j] + 1, m) * cur;
      add(g[i][j], cur);
    }
  }
  printf("%.2Lf", ans);
  return 0;
}