【BZOJ4523】[Cqoi2016]路由表 Trie树模拟

【BZOJ4523】[Cqoi2016]路由表

Description

路由表查找是路由器在转发IP报文时的重要环节。通常路由表中的表项由目的地址、掩码、下一跳（Next Hop）地址和其他辅助信息组成。例如：

当路由器收到一个IP报文时，会将报文中的目的IP地址与路由表中的表项逐条进行比较，选择匹配且最明确的表项，将报文转发给该表项中指定的下一跳。

匹配的过程是将报文中的目的地址和表项中的目的地址分别转为二进制串，再查看表项中的掩码长度，若掩码长度为x，则将两个二进制串的前x位进行比较，如果相同则认为匹配。

所谓最明确是指在有多个表项匹配时，总是掩码长度最大的表项。也可以理解为匹配的二进制位最多的项。

IP地址转为二进制串的操作是把地址中4个整数（一定在y到255的范围内）分别转为8位二进制数，再顺序拼接起来，得到一个32位的二进制串。例如，192.168.1.253转为二进制串后为

11000000 10101000 00000001 11111101

我们以报文的目的地址为8.8.8.8为例，说明其在上述路由表的匹配过程。

上表将地址均转为二进制串，并用红色标记出待比较的位（由掩码长度决定）。将红色部分与报文中的目的地址比较，可知0.0.0.0/1、8.8.8.0/24、8.8.8.8、32均能够匹配。路由器从中选取掩码长度最长（/32）的表项8.8.8.8/32，将报文转发给其对应的下一跳地址192.168.1.253。

在实际的核心路由器中，路由表通常较大（现在互联网的全局路由表已经接60万条记录），并且会随着新接入设备不断扩张。为了分析路由表变化对转发产生的影响，网络工程师想要知道一段时间内某个IP地址的路由表项选择发生了多少次变化（变化是指由于最明确匹配等因素选择了不同的表项，不考虑下一跳地址）。

Input

第一行为整数M，表示共有M次操作。接下来M行，每行描述一次操作。操作有两种：

A D/L

其中.为一个IP地址，G为整数（1≤L≤32）。添加一条表项至路由表，其目的地址为D掩码长度为L。下一跳地址由于没有用到，故省略。

Q D a b

其中D为一个IP地址，a,b为正整数（a≤b）。查询从第a次至第b次添加表项期间（含a、b），目的地址D的路由表项选择发生了多少次变化。保证查询时表中至少有b个表项。

N<=10^6数据保证不会重复添加目的地址和掩码长度都相同的表项。

Output

包含若干行，每行仅有一个整数，依次对应每个查询操作。

题解：奇怪了，我怎么连这种模拟题都做了~

我们先离线，对询问串构建Trie树，然后依次将表项放到Trie树中查询。如何求一个询问串被更改的次数呢？由于每次更改，匹配的长度一定会变长，所以我们可以对Trie树的每个节点都开一个vector，在询问串经过的Trie树上的每个节点的vector中都加入这个询问。在查询时，我们将对应节点的vector中的询问都拿出来，依次判断是否产生更新，然后将这个vector清空。

结果第一发MLE了，后来知道数组开3500000就行。。。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
typedef unsigned int ui;
const int maxn=1000010;
int n,m,Q,tot;
int vis[maxn],first[maxn],last[maxn],mx[maxn],ans[maxn],len[maxn];
ui val[maxn],v[maxn];
char str[10];
struct node
{
	int ch[2];
	vector<int> v;
}p[maxn*5];
vector<int> q[maxn];
vector<int>::iterator it;
inline void insert(int x)
{
	register int i,u;
	register bool d;
	for(i=31,u=1;i>=0;i--)
	{
		d=(val[x]>>i)&1;
		if(!p[u].ch[d])	p[u].ch[d]=++tot;
		u=p[u].ch[d],p[u].v.push_back(x);
	}
}
inline void query(int x)
{
	register int i,u;
	register bool d;
	for(i=31,u=1;i>=31-len[x]+1;i--)
	{
		d=(v[x]>>i)&1;
		if(!p[u].ch[d])	return ;
		u=p[u].ch[d];
	}
	for(it=p[u].v.begin();it!=p[u].v.end();it++)
	{
		i=*it;
		if(x>=first[i]&&x<=last[i]&&mx[i]<len[x]&&vis[i]<x)	vis[i]=x,ans[i]++;
		mx[i]=max(mx[i],len[x]);
	}
	p[u].v.clear();
}
inline int rd()
{
	register int ret=0,f=1;	register char gc=getchar();
	while(gc<'0'||gc>'9')	{if(gc=='-')	f=-f;	gc=getchar();}
	while(gc>='0'&&gc<='9')	ret=ret*10+(gc^'0'),gc=getchar();
	return ret*f;
}
int main()
{
	//freopen("bz4523.in","r",stdin);
	//freopen("bz4523.out","w",stdout);
	m=rd(),tot=1;
	register int i;
	register ui a,b,c,d;
	for(i=1;i<=m;i++)
	{
		scanf("%s",str);
		if(str[0]=='Q')	a=rd(),b=rd(),c=rd(),d=rd(),val[++Q]=a<<24|b<<16|c<<8|d,insert(Q),first[Q]=rd(),last[Q]=rd();
		else	a=rd(),b=rd(),c=rd(),d=rd(),v[++n]=a<<24|b<<16|c<<8|d,len[n]=rd();
	}
	tot=1;
	for(i=1;i<=n;i++)	query(i);
	for(i=1;i<=Q;i++)	printf("%d\n",ans[i]);
	return 0;
}