Born Slippy#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#include<iostream>
#include<algorithm>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<string>
using namespace std;
typedef long long ll;
int op;
ll ans;
ll w[100010];
int f[100010];
ll dp[300][300];
ll use[100010][300];
int vis[300];
char str[5];
vector<int>vec[100010];
const int mod=1e9+7;
ll opt(ll a,ll b)
{
if(op==1) return a^b;
if(op==2) return a&b;
if(op==3) return a|b;
}
void dfs(int u)
{
ll a=w[u]>>8;
ll b=w[u]&255;
ll sum=0;
for(int i=0;i<256;i++)
if(vis[i])
sum=max(sum,dp[i][b]+(opt(a,i)<<8));
ans=(ans+(u*(sum+w[u])%mod))%mod;
vis[a]++;
for(int i=0;i<256;i++)
{
use[u][i]=dp[a][i];
dp[a][i]=max(dp[a][i],sum+opt(b,i));
}
for(int i=0;i<vec[u].size();i++)
dfs(vec[u][i]);
for(int i=0;i<256;i++)
dp[a][i]=use[u][i];
vis[a]--;
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
int n;
scanf("%d%s",&n,str);
if(str[0]=='X') op=1;
if(str[0]=='A') op=2;
if(str[0]=='O') op=3;
for(int i=1;i<=n;i++)
vec[i].clear();
memset(vis,false,sizeof(vis));
memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++)
{
scanf("%lld",&w[i]);
}
for(int i=2;i<=n;i++)
{
scanf("%d",&f[i]);
vec[f[i]].push_back(i);
}
ans=0;
dfs(1);
printf("%lld\n",ans);
}
return 0;
}
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#include<iostream>
#include<algorithm>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<string>
using namespace std;
typedef long long ll;
int op;
ll ans;
ll w[100010];
int f[100010];
ll dp[300][300];
ll use[100010][300];
int vis[300];
char str[5];
vector<int>vec[100010];
const int mod=1e9+7;
ll opt(ll a,ll b)
{
if(op==1) return a^b;
if(op==2) return a&b;
if(op==3) return a|b;
}
void dfs(int u)
{
ll a=w[u]>>8;
ll b=w[u]&255;
ll sum=0;
for(int i=0;i<256;i++)
if(vis[i])
sum=max(sum,dp[i][b]+(opt(a,i)<<8));
ans=(ans+(u*(sum+w[u])%mod))%mod;
vis[a]++;
for(int i=0;i<256;i++)
{
use[u][i]=dp[a][i];
dp[a][i]=max(dp[a][i],sum+opt(b,i));
}
for(int i=0;i<vec[u].size();i++)
dfs(vec[u][i]);
for(int i=0;i<256;i++)
dp[a][i]=use[u][i];
vis[a]--;
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
int n;
scanf("%d%s",&n,str);
if(str[0]=='X') op=1;
if(str[0]=='A') op=2;
if(str[0]=='O') op=3;
for(int i=1;i<=n;i++)
vec[i].clear();
memset(vis,false,sizeof(vis));
memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++)
{
scanf("%lld",&w[i]);
}
for(int i=2;i<=n;i++)
{
scanf("%d",&f[i]);
vec[f[i]].push_back(i);
}
ans=0;
dfs(1);
printf("%lld\n",ans);
}
return 0;
}
感谢叉姐在ICPCCamp上出的这道题最初的原型 -- Data Structure You've Never Heard Of, 同样感谢Claris老师的教导.
由于and, or和xor方法都差不多, 这里仅考虑and操作. 不妨令dp(s)=f(s)−wsdp(s)=f(s)-w_sdp(s)=f(s)−ws, 我们大概要求的就是dp(i)=maxj is ancestor of i{dp(j)+wi and wj}dp(i)=\displaystyle\max_{j \text{ is ancestor of } i}{dp(j)+w_i\text{ and }w_j}dp(i)=j is ancestor of imax{dp(j)+wi and wj}. 然后, 显然dp(j)+wi and wjdp(j)+w_i\text{ and }w_jdp(j)+wi and wj这个式子可以拆成dp(j)dp(j)dp(j)+[wiw_iwi后8位] and [wjw_jwj后8位] + ([wiw_iwi前8位] and [wjw_jwj前8位]) << 8.
先考虑序列上应该如何做, 即求dp(i)=maxj<i{dp(j)+wi and wj}dp(i)=\displaystyle\max_{j < i}{dp(j)+w_i\text{ and }w_j}dp(i)=j<imax{dp(j)+wi and wj}. 考虑这样一个二维数组ds(x,y)ds(x,y)ds(x,y), 表示对于某个wiw_iwi的后8位为yyy, 对于某个wjw_jwj的前8位为xxx时, dp(j)dp(j)dp(j) + [wiw_iwi后8位] and [wjw_jwj后8位]的最值.
如果知道了上述数组, 那么对于某个iii, 计算dp(i)dp(i)dp(i)的值就十分方便, 不妨令wi=(a<<8)∣bw_i=(a << 8) | bwi=(a<<8)∣b, 即aaa和bbb分别是wiw_iwi前8位和后8位, 那么只需要枚举wjw_jwj的前8位xxx, 用ds(x,b)+((a and x)<<8)ds(x,b)+((a \text{ and } x) << 8)ds(x,b)+((a and x)<<8)更新dp(i)dp(i)dp(i). 把新的dpdpdp值更新到ds(x,y)ds(x,y)ds(x,y)也是类似的.
上述方法推广到树上也是十分简单, 由于每次更新ds(x,y)ds(x,y)ds(x,y)的时候只有数组的一维会变动(令aaa=wiw_iwi>>8, 那么只有ds(a,⋅)ds(a,\cdot)ds(a,⋅)会变化), 那么只要对数组的第一维做一个可持久化就好了(或者说边dfs边备份).