2133: 密室逃脱
Time Limit: 1 Sec Memory Limit: 128 MBSubmit: 391 Solved: 79
Submit Status Web Board
Description
XOR在玩密室逃脱,在某一关中,桌上有一个一张纸,上面写着“请根据所给例子求解答案从而获得密码”,下面写了几个字符串“01 10 11”,而答案为“6”,聪明的XOR立马就知道了这是给出一些二进制数字S,求存在多少对有序二元组(i,j)使得S[i]^S[j]<S[i],现在还有T组数据,每组是n个长度为m的二进制数字,聪明的XOR立马开始动手求解答案。
Input
第一行一个整数T,表示数据组数。
对于每组数据,首先读入两个整数n,m(n*m<=1000000),接下来为n行,每行为一个长度为m的01串,表示一个二进制数字
Output
对于每个数据,输出一个整数x,表示二元组数目
Sample Input
Sample Output
思路:
1、位运算操作肯定我们要从位运算下手。2进制亦或运算对于一位来讲有四种情况:
①1^1==0 会将当前值变小。
②1^0==1 当前值没有变。
③0^1==1 当前值变大了。
④0^0==0 当前值没有变。
那么其中只有①和③会影响到s【i】^s【j】和s【i】比较的结果。
很显然,如果从高位到低位进行分析的话,如果第1位发生了①,那么之后无论多少位子发生了③都是使得s【i】^s【j】<s【i】;
同理,如果第1位发生了③ ,那么之后无论多少位子发生了①都是使得s【i】^s【j】>s【i】;
所以我们不妨将问题转化成字典树问题。
2、每一次输入进来一个01串我们都入树。
那么对于查询,我们每个字符串都要遍历到,每个字符串在find统计的过程中,遵循下列两个原则即可:
①如果当前是1.统计前缀到此处为1的字符串数量,然后接下来去遍历0.
②如果当前是0.我们肯定是接下来去遍历0.
3、过程维护一下即可,注意细节以及优化点,950+ms过的,出题人数据真的多。
Ac代码:
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<iostream>
using namespace std;
#define ll long long int
#define maxn 2
typedef struct tree
{
tree *nex[maxn];
ll val;
}tree;
ll n,m;
tree root;
void init()
{
for(ll i=0;i<maxn;i++)
{
root.nex[i]=NULL;
root.val=0;
}
}
void creat(char *str)
{
ll len=m;
tree *p=&root,*q;
for(ll i=0;i<len;i++)
{
ll id=str[i]-'0';
if(p->nex[id]==NULL)
{
q=(tree *)malloc(sizeof(root));
q->val=1;
for(ll j=0;j<2;j++)
{
q->nex[j]=NULL;
}
p->nex[id]=q;
}
else
{
p->nex[id]->val++;
}
p=p->nex[id];
}
}
ll find(char *str)
{
ll ans=0;
ll len=m;
tree *p=&root;
for(ll i=0;i<len;i++)
{
ll id=str[i]-'0';
if(id==0)
{
if(p->nex[id]!=NULL)
{
p=p->nex[id];
}
else return ans;
}
if(id==1)
{
if(p->nex[id]!=NULL)ans+=p->nex[id]->val;
if(p->nex[1-id]!=NULL)
{
p=p->nex[1-id];
}
else return ans;
}
}
return ans;
}
int main()
{
ll t;
scanf("%lld",&t);
while(t--)
{
init();
scanf("%lld%lld",&n,&m);
char a[n+2][m+2];
for(ll i=0;i<n;i++)
{
scanf("%s",a[i]);
creat(a[i]);
}
ll ans=0;
for(ll i=0;i<n;i++)
{
ans+=find(a[i]);
}
printf("%lld\n",ans);
}
}