【清华集训2017模拟12.10】大佬的难题

最新推荐文章于 2019-09-23 19:25:55 发布

Cold_Chair

最新推荐文章于 2019-09-23 19:25:55 发布

阅读量527

点赞数 1

分类专栏：数论杂集文章标签：信息学树状数组容斥原理

本文链接：https://blog.csdn.net/Cold_Chair/article/details/78759612

版权

数论杂集专栏收录该内容

49 篇文章 2 订阅

订阅专栏

题目大意：

这里写图片描述
1<=n<=2e6，时限：2.5s

题解：

容斥原理瞎搞。

设A,B,C分别为满足三个条件的集合。

根据容斥原理，有:
$A∪B∪C=A+B+C-A∩B-A∩C-B∩C+A∩B∩C$

而A∪B∪C同时会等同于总数 $n*(n-1)$ -都大于的个数（即都小于的个数 $A∩B∩C$ ）

所以：
$n*(n-1)-A∩B∩C=A+B+C-A∩B-A∩C-B∩C+A∩B∩C$
$2*A∩B∩C=n*(n-1)-A-B-C+A∩B+A∩C+B∩C$

$A=B=C=n*(n-1)/2$

所以:
$2*A∩B∩C=A∩B+A∩C+B∩C-n*(n-1)/2$
$A∩B∩C={A∩B+A∩C+B∩C-n*(n-1)/2\over 2}$

求 $A∩B+A∩C+B∩C$ 可以直接用树状数组搞三次，常数小。

Code:

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
#define fo(i, x, y) for(int i = x; i <= y; i ++)
#define low(a) ((a) & -(a))
using namespace std;

const int N = 2e6 + 5;

int a[3][N], n;

ll seed;
ll Rand() {
    return seed = ((seed * 19260817) ^ 233333) & ((1 << 24) - 1);
}
void gen(int *a) {
    fo(i, 1, n) a[i] = i;
    fo(i, 1, n) swap(a[i], a[Rand() % i + 1]);
}

ll ans;
int f[N], z[N];

void Gao(int p, int q) {
    fo(i, 1, n) z[a[p][i]] = a[q][i];
    memset(f, 0, sizeof(f));
    fo(i, 1, n) {
        int x = z[i];
        while(x) ans += f[x], x -= low(x);
        x = z[i];
        while(x <= n) f[x] ++, x += low(x);
    }
}

int main() {
    freopen("dalao.in", "r", stdin);
    freopen("dalao.out", "w", stdout);
    scanf("%d", &n);
    fo(i, 0, 2) scanf("%lld", &seed), gen(a[i]);
    ans = - (ll)n * (n - 1) / 2;
    Gao(0, 1); Gao(0, 2); Gao(1, 2);
    printf("%lld", ans / 2);
}