首先我们观察收费表格,对于非叶子节点,如果n0< n1,我们把这个节点的状态记作0,否则记作1.则不难发现,对于一个叶子节点x,如果和它的祖先i的状态一样,他就要付出在这个祖先处的所有费用。因此我们可以分开来计算每一个叶子节点的贡献。预处理出cost[x][i]数组,表示叶子节点x在祖先i处的所有花费。(
cost[x][i]=∑[lca(x,y)==i]∗f[x][y]
)。可以
O(n2logn)
处理出。然后我们只需要知道叶子x的所有祖先的状态就可以知道叶子x变或不变的总贡献。我们可以dfs来枚举一下祖先的状态。然后做一个dp,dp[x][i]表示在目前的祖先状态下,以x为根的子树中有i个1的最小花费。 然后类似背包转移即可。
考虑一共
2n
个叶子,每个叶子有n个祖先,枚举祖先的状态,所以一个叶子会被搜到
2n
,总的就是
22n
不会分析复杂度了啦gg
因为我给整棵满二叉树编了号,根为1,所以一最多共会有2047个点!
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 2060
inline char gc(){
static char now[1<<16],*S,*T;
if (T==S){T=(S=now)+fread(now,1,1<<16,stdin);if (T==S) return EOF;}
return *S++;
}
inline int read(){
int x=0,f=1;char ch=gc();
while(ch<'0'||ch>'9') {if (ch=='-') f=-1;ch=gc();}
while(ch<='9'&&ch>='0') x=x*10+ch-'0',ch=gc();
return x*f;
}
int n,bin[13],m,cost[N][N],c[N],dp[N<<1][N];
bool a[N],s[N];//dp[x][i],在目前的祖先状态下,以x为根的子树中有i个1的最小花费。
inline int lca(int x,int y){
while(x!=y) x>>=1,y>>=1;return x;
}
inline void dfs(int x,int d){
if(!d){
dp[x][a[x]]=0;dp[x][a[x]^1]=c[x];
for(int i=x>>1;i;i>>=1) dp[x][s[i]]+=cost[x][i];return;
}memset(dp[x],inf,sizeof(dp[x][0])*(bin[d]+1));//每次进来都要清inf,因为祖先状态发生了改变
s[x]=0;//n0<n1
dfs(x<<1,d-1);dfs(x<<1|1,d-1);
for(int i=1;i<=bin[d-1];++i)
for(int j=bin[d-1]+1-i;j<=bin[d-1];++j)
dp[x][i+j]=min(dp[x][i+j],dp[x<<1][i]+dp[x<<1|1][j]);
s[x]=1;//n0>=n1
dfs(x<<1,d-1);dfs(x<<1|1,d-1);
for(int i=0;i<=bin[d-1];++i)
for(int j=0;i+j<=bin[d-1];++j)
dp[x][i+j]=min(dp[x][i+j],dp[x<<1][i]+dp[x<<1|1][j]);
}
int main(){
// freopen("a.in","r",stdin);
m=read();bin[0]=1;
for(int i=1;i<=m;++i) bin[i]=bin[i-1]<<1;
n=bin[m];int ans=inf;
for(int i=1;i<=n;++i) a[n-1+i]=read();
for(int i=1;i<=n;++i) c[n-1+i]=read();
for(int i=1;i<=n;++i)
for(int j=i+1;j<=n;++j){
int x=n-1+i,y=n-1+j,t=lca(x,y),val=read();
cost[x][t]+=val;cost[y][t]+=val;
}
dfs(1,m);
for(int i=0;i<=n;++i) ans=min(ans,dp[1][i]);
printf("%d\n",ans);
return 0;
}