hdu 4323 Magic Number 2012多校训练第三场1004题 BK树

今天下午进行了第三场的多校，表示被虐成了狗……

1004题当时感觉想暴力，毛估复杂度是1000*1500*10*10……这还是单case的，所以一直没敢敲……直到最后发现全场几十支队伍都过了，开始敲……结果没想清楚最终WA到死

晚上的时候回基地写了个暴力解法，375ms过了，感觉时间卡的也不紧……需要稍作优化，两个字符串长度之差大于要求的编辑距离时就不用再去dp了，dp的状态转移和求最长公共子序列非常像

for(int i=1; i<=l1; i++)
for(int j=1; j<=l2; j++)
if(s1[i-1]!=s2[j-1])
dp[i][j] = min(min(dp[i-1][j-1]+1,dp[i-1][j]+1),dp[i][j-1]+1);
else dp[i][j] = min(min(dp[i-1][j-1],dp[i-1][j]+1),dp[i][j-1]+1);dp[i][j]表示第一个的前i个字符和第二个的前j个字符的编辑距离
后来看了结题报告，知道了BKTree这种东西，然后去看了matrix67大神的博客，学了下这个数据结构……感觉和trie树基本雷同，具体的还是看看大神的博客吧

BK树解法代码如下：（我肯定写残了……只比我的暴力程序快了两倍）

#include <cstdio>
#include <cstring>
const int MAX = 1510;
const int LEN = 15;
char s2[LEN];
struct point
{
char s[LEN];
int next[LEN];
}p[MAX];
inline int min(int a, int b)
{
return (a<b)?a:b;
}
int dp[LEN][LEN];
int cal_levenshtein(char *s1,char *s2)
{
int l1 = strlen(s1);
int l2 = strlen(s2);
dp[l1][l2] = 1000;
dp[0][0] = 0;
for(int i=1; i<=l1; i++)
dp[i][0] = i;
for(int i=1; i<=l2; i++)
dp[0][i] = i;
for(int i=1; i<=l1; i++)
for(int j=1; j<=l2; j++)
if(s1[i-1]!=s2[j-1])
dp[i][j] = min(min(dp[i-1][j-1]+1,dp[i-1][j]+1),dp[i][j-1]+1);
else dp[i][j] = min(min(dp[i-1][j-1],dp[i-1][j]+1),dp[i][j-1]+1);
return dp[l1][l2];
}
int cnt = 0;
void insert(char *s,int x)
{
int tmp = cal_levenshtein(s,p[x].s);
if(p[x].next[tmp]==-1)
{
p[x].next[tmp] = ++cnt;
memset(p[cnt].next,-1,sizeof(p[cnt].next));
strcpy(p[cnt].s,s);
return;
}
insert(s,p[x].next[tmp]);
}
int query(char *s,int c,int x)
{
int ans = 0;
int tmp = cal_levenshtein(s,p[x].s);
if(tmp<=c) ans++;
int i,end;
if(tmp-c<0) i = 0;
else i = tmp-c;
if(tmp+c>LEN) end = LEN;
else end = tmp+c;
for(;i<=end; i++)
if(p[x].next[i]!=-1)
ans+=query(s,c,p[x].next[i]);
return ans;
}
int main()
{
int cas;
scanf("%d",&cas);
for(int t=1; t<=cas; t++)
{
int n,m;
scanf("%d%d",&n,&m);
scanf("%s",p[0].s);
memset(p[0].next,-1,sizeof(p[0].next));
cnt = 0;
char s[LEN];
for(int i=1; i<n; i++)
{
char s[LEN];
scanf("%s",s);
insert(s,0);
}
printf("Case #%d:\n",t);
while(m--)
{
int c;
scanf("%s%d",s,&c);
printf("%d\n",query(s,c,0));
}
}
return 0;
}


再附上自己的暴力代码：
#include <cstdio>
#include <cstring>
char s[1510][15];
char s2[15];
int dp[15][15];
int len[1510];
int n,m,l;
inline int min(int a,int b)
{
return (a<b)?a:b;
}
int cal(int x)
{
int cnt = 0,j,k;
l = strlen(s2);
for(int i=0; i<n; i++)
{
if(l-len[i]>x||l-len[i]<-x)
continue;
dp[0][0] = 0;
for(j=0; s[i][j]; j++)
dp[j+1][0] = j+1;
for(j=0; s2[j]; j++)
dp[0][j+1] = j+1;
for(j=0;s[i][j]; j++)
for(k=0; s2[k]; k++)
if(s[i][j]!=s2[k])
dp[j+1][k+1] = min(min(dp[j][k]+1,dp[j][k+1]+1),dp[j+1][k]+1);
else dp[j+1][k+1] = min(min(dp[j][k],dp[j][k+1]+1),dp[j+1][k]+1);
if(dp[j][k]<=x) cnt++;
}
return cnt;
}
int main()
{
int cas,x;
scanf("%d",&cas);
for(int t=1; t<=cas; t++)
{
scanf("%d%d",&n,&m);
for(int i=0; i<n; i++)
{
scanf("%s",s[i]);
len[i] = strlen(s[i]);
}
printf("Case #%d:\n",t);
while(m--)
{
scanf("%s%d",s2,&x);
printf("%d\n",cal(x));
}
}
return 0;
}