ZZULIOJ 2133 密室逃脱【思维+字典树】

2133: 密室逃脱

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 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

1

3 2

01

10

11

Sample Output

6

思路：

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);

    }
}