HDU-2017 多校训练赛3-1004-Kanade's trio

ACM模版

描述

题解

这个题思路十分巧妙，没想到竟然可以用字母树解。



题解不难理解，就是有些出人意料了。看了题解后，我拿着官方题解，参考着写了一份，习惯性的将一些东西改成了自己的习惯……比如说，初始化能用 memset() 的都用 memset()，然后我就超时了……，很纳闷儿，再三斟酌，我尝试将初始化改成了循环初始化，发现一下子竟然 AC 了。这是最让我十分意外的，我一直以来都十分认可 memset()，以为他会快很多，但是，他并没有想象中那么快。之所以会超时是因为循环初始化人家用多少初始化多少，而 memset() 我则是习惯性的将整个数组初始化，导致时间消耗一下子上去了，题中有个很重要的条件，1≤∑n≤5∗105，所以，这里并没有直接给 n 的范围，为了肯定可行，我们都是开了足够大，但是我们绝大多数空间都是用不到的，如果每次都是 memset() 整个数组，自然是浪费太严重，时间消耗太高，TLE 也就正常了。

所以呢，以后一定要注意 memset() 函数的用处，并不能因为他效率高，而一味的简易处理，代码虽然简单了，TLE 也是没商量的事。

代码

#include <iostream>
#include <cstdio>

using namespace std;

typedef long long ll;

const int MAXN = 5e5 + 10;
const int MAGIC = 30;

template <class T>
inline void scan_d(T &ret)
{
char c;
ret = 0;
while ((c = getchar()) < '0' || c > '9');
while (c >= '0' && c <= '9')
{
ret = ret * 10 + (c - '0'), c = getchar();
}
}

int n, tot;
int a[MAXN];
int go[MAXN << 5][2];
int num[MAXN << 5];
int tree[MAXN][MAGIC][2];
ll ss[MAXN << 5];

int main()
{
int T;
scan_d(T);
while (T--)
{
for (int i = 1; i <= tot; i++)
{
go[i][0] = go[i][1] = 0;
ss[i] = num[i] = 0;
}
for (int i = 1; i <= n; i++)
{
for (int j = 0; j < MAGIC; j++)
{
tree[i][j][0] = tree[i][j][1] = 0;
}
}

tot = 1;
scan_d(n);
for (int i = 1; i <= n; i++)
{
scan_d(a[i]);
}

for (int i = 1; i <= n; i++)
{
for (int j = 0; j < MAGIC; j++)
{
tree[i][j][0] = tree[i - 1][j][0];
tree[i][j][1] = tree[i - 1][j][1];
}
for (int j = 0; j < MAGIC; j++)
{
int k = (a[i] & (1 << j)) > 0;
tree[i][j][k]++;
}
}

ll ans = 0;
for (int i = n; i >= 1; i--)
{
int now = 1;
for (int j = MAGIC - 1; j >= 0; j--)
{
int k = (a[i] & (1 << j)) > 0;
if (go[now][k ^ 1])
{
ans += ss[go[now][k ^ 1]] - num[go[now][k ^ 1]] * 1ll * tree[i][j][k];
}
if (!go[now][k])
{
break;
}
now = go[now][k];
}
now = 1;
for (int j = MAGIC - 1; j >= 0; j--)
{
int k = (a[i] & (1 << j)) > 0;
if (!go[now][k])
{
go[now][k] = ++tot;
}
now = go[now][k];
ss[now] += tree[i - 1][j][k ^ 1];
num[now]++;
}
}

cout << ans << endl;
}

return 0;
}