码迷,mamicode.com
首页 > 其他好文 > 详细

[CSP-S模拟测试]:Race(数学+Trie树)

时间:2019-09-22 12:51:53      阅读:98      评论:0      收藏:0      [点我收藏+]

标签:har   std   模拟   san   怎么   输出   时间   需要   mic   

题目描述

一年一度的运动会开始了。有$N$个选手参赛,第$i$个选手有一个能力值(保证$A[i]$两两不同),比赛一共进行了天。在第$j$天($0\leqslant j\leqslant 2^{m-1}$)的比赛中,第$i$个选手的得分为$A[i]\ xor\ j$,然后从大到小排名,排名为$x$($x$从$0$开始)的同学会获得的积分,你需要求出每个同学最后总的积分和$q[i]$模${10}^9+7$的结果$p[i]$。为了避免输出文件过大,你只要输出$p[i]$的异或和即可。


输入格式

第一行两个整数,分别代表$N,M$。
接下来一行$N$个整数,第$i$个数代表$A[i]$。


输出格式

一个整数代表答案。


样例

样例输入:

3 2
0 1 2

样例输出:

8


数据范围与提示

对于$10\%$的数据,$M\leqslant 5$。
对于$30\%$的数据,$N\leqslant 100$。
对于$50\%$的数据,$N\leqslant 1,000$。
对于$100\%$的数据,$N\leqslant 200,000,M\leqslant 30,A[i]<2^m$。


题解

考虑怎么求出$x$的答案。平方相当于是枚举两个人(可以相同),把这两个人同时排在$x$前面的天数计入答案。那么对于$x$,如果我们求出$f[i]$也就是能力值的二进制中第$i+1$到$M-1$位都和他相等且第$i$位不同的人有多少个,那么这些人是否排在他前面只由第$i$位确定,一共有$2^{(M-1)}$天,而且不需要从这些人中枚举两个人了,因为直接平方即可。我们只要枚举$i$和$j$,$f[i]$这些人和$f[j]$这些人同时排在他前面的天数为$2^{(M-2)}$,所以把$2\times f[i]\times f[j]\times 2^{(M-2)}$计入答案。具体实现有很多方法,可以用$trie$树实现。

时间复杂度:$\Theta(N)$。

期望得分:$100$分。

实际得分:$100$分。


代码时刻

#include<bits/stdc++.h>
using namespace std;
int n,m;
int trie[10000000][2],cnt;
long long size[10000000];
long long ans;
void insert(int x)
{
	int p=0;
	for(int i=m-1;i>=0;i--)
	{
		int c=(x>>i)&1;
		if(!trie[p][c])trie[p][c]=++cnt;
		p=trie[p][c];
		size[p]++;
	}
}
void dfs(int x,long long sum,long long num)
{
	if(trie[x][0])dfs(trie[x][0],(sum+(size[trie[x][1]]*size[trie[x][1]]%1000000007*(1<<m-1)%1000000007+size[trie[x][1]]*num%1000000007*(1<<m-1)%1000000007)%1000000007)%1000000007,num+size[trie[x][1]]);
	if(trie[x][1])dfs(trie[x][1],(sum+(size[trie[x][0]]*size[trie[x][0]]%1000000007*(1<<m-1)%1000000007+size[trie[x][0]]*num%1000000007*(1<<m-1)%1000000007)%1000000007)%1000000007,num+size[trie[x][0]]);
	if(!trie[x][0]&&!trie[x][1])ans^=sum;
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		int a;scanf("%d",&a);
		insert(a);
	}
	dfs(0,0,0);
	printf("%lld",ans);
	return 0;
}

rp++

[CSP-S模拟测试]:Race(数学+Trie树)

标签:har   std   模拟   san   怎么   输出   时间   需要   mic   

原文地址:https://www.cnblogs.com/wzc521/p/11566898.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!