51nod 1920 空间统计学

空间统计学

Description

有个$m$维的空间，并且每一维的坐标$x$都满足$x$∈[$0$,$3$]并且$x$为整数。
这个空间有$n$个部落，每个部落都坐落在这片空间中的一个点上，可以用坐标($x_1$, $x_2$, …,$x_m$)来表示。
有些部落可能在在同一个点上面。
定义两个点的距离为它们的曼哈顿距离，即每一维坐标差的绝对值的和。
现在对$0$~$3m$之间的每一个数字$x$，统计有多少对部落之间的距离为$x$。
注意，一对部落是有序的，即部落($a$,$b$)和部落($b$,$a$)为不同的两对。
$n$<=$2*10^5$,$m$<=$9$

Solution

考虑动态规划。
设$f_{i,j,k}$表示只考虑在四进制下的的前$i$位，变成$j$花费了$k$步的部落有多少个。
转移也就很显然了。(实在不懂就看标程吧)
设$t_i$表示转成四进制后表示为i的部落的个数。
则对于距离为$k$的答案为$\sum$$f_{m,i,k}$*$t_i$

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>

#define fo(i,j,l) for(i=j;i<=l;i++)
#define fd(i,j,l) for(i=j;i>=l;i--)

using namespace std;
typedef long long ll;
const ll M=1<<19,N=11,K=3*N;
ll f[2][M][K],t[M];
ll m4[M];
int i,j,k,n,m,l,o,p,u,v;

int read()
{
    int o=0; char ch=' ';
    for(;ch<'0'||ch>'9';ch=getchar());
    for(;ch>='0'&&ch<='9';ch=getchar())o=o*10+ch-48;
    return o;
}

int main()
{
    m4[1]=1;
    fo(i,2,N+1)m4[i]=m4[i-1]*4;
    cin>>n>>m;
    fo(i,1,n)
    {
        ll k=0;int y;
        fo(l,1,m)y=read(),k=k*4+y;
        f[0][k][0]++;
        t[k]++;
    }
    ll zz=m4[m+1]-1;
    u=0;ll uu,kk;
    fo(i,0,m-1)
    {
        v=1-u;
        fo(l,0,zz)
        fo(j,0,3*(i-1))f[v][l][j]=0;
        fo(l,0,zz)
        fo(j,0,3*i)
        if(f[u][l][j])
        {
            uu=(l/m4[i+1])%4;
            kk=l-uu*m4[i+1];
            fo(k,0,3)
            f[v][kk+k*m4[i+1]][j+abs(uu-k)]+=f[u][l][j];
        }
        u=v;
    }
    fo(i,0,3*m)
    {
        kk=0;
        fo(l,0,zz)
        kk=kk+f[u][l][i]*t[l];
        printf("%lld ",kk);
    }
    return 0;
}