[BZOJ5404]party

Description

给定一棵以1为根的树，第i个点有第ai种商品。
有c个人要从b1~bc点出发，到其lca处集合，每个人只能带其经过的点的商品，且需要满足：
1：每个人带的商品的数量相同
2：所有人带的商品必须两两不同
3：最大化所有人带的商品的数量。
q次询问，设商品种类为m
n<=300000,q<=50000,m<=1000,c<=5

Solution

看错题以为是神仙题
考虑暴力，可以网络流+二分，每个点向其路径上的颜色连边
发现中间的边的容量都为1，且左边的点数<=5，可以转化为二分图最大匹配，每个点拆成x个相同的点。
现在我们要找到一个最大的x使得这个二分图有完美匹配，考虑hall定理，对于左边的每个集合S，其出边的并集大小sz>=|S|，因为每个点拆成x个相同的点，最坏情况一定是每个点拆出来的点同时都被选。
只需要枚举2^5种集合，根据hall定理，设枚举的集合大小为k，有x<=sz/k
问题变成某条祖先后代链上的颜色数，观察到颜色不多，考虑使用bitset，用线段树+树链剖分维护是log^2的，观察算法过程可以维护一个前缀bitset表示某个点到其重链顶的颜色集合，复杂度变成单次log n
然后便可通过此题

Code

#include <bitset>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
#define rep(i,a) for(int i=lst[a];i;i=nxt[i])
using namespace std;

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

void write(int x) {
    if (!x) {puts("0");return;}
    char ch[20];int tot=0;
    for(;x;x/=10) ch[++tot]=x%10+'0';
    fd(i,tot,1) putchar(ch[i]);
    puts("");
}

const int N=3e5+5,M=1e3+5;

int t[N],nxt[N],lst[N],l;
void add(int x,int y) {t[++l]=y;nxt[l]=lst[x];lst[x]=l;}

int n,m,q,dep[N],col[N],sz[N],son[N],top[N],w[N],fa[N],dfn[N],tot,c,b[6];

bitset<M> tr[N<<1],pre[N],bit[6];

void dfs(int x) {
    sz[x]=1;int k=0;
    rep(i,x) {
        dfs(t[i]);
        if (sz[t[i]]>k) k=sz[t[i]],son[x]=t[i];
        sz[x]+=sz[t[i]];
    }
}

void make(int x,int y) {
    pre[x][col[x]]=1;
    if (x!=y) pre[x]|=pre[fa[x]];
    top[x]=y;w[++tot]=x;dfn[x]=tot;
    if (!son[x]) return;
    make(son[x],y);
    rep(i,x) if (t[i]!=son[x]) make(t[i],t[i]);
}

int lca(int x,int y) {
    while (top[x]!=top[y]) {
        if (dep[top[x]]<dep[top[y]]) swap(x,y);
        x=fa[top[x]];
    }
    return dep[x]<dep[y]?x:y;
}

int Id(int l,int r) {return l+r|l!=r;}

void build(int l,int r) {
    int v=Id(l,r);
    if (l==r) {tr[v][col[w[l]]]=1;return;}
    int mid=l+r>>1,ls=Id(l,mid),rs=Id(mid+1,r);
    build(l,mid);build(mid+1,r);
    tr[v]=tr[ls]|tr[rs];
}

bitset<M> find(int l,int r,int x,int y) {
    int v=Id(l,r);
    if (l==x&&r==y) return tr[v];
    int mid=l+r>>1;
    if (y<=mid) return find(l,mid,x,y);
    else if (x>mid) return find(mid+1,r,x,y);
    else return find(l,mid,x,mid)|find(mid+1,r,mid+1,y);
}

bitset<M> query(int x,int y) {
    bitset<M> res;
    while (top[x]!=top[y]) {
        res|=pre[x];
        x=fa[top[x]];
    }
    res|=find(1,n,dfn[y],dfn[x]);
    return res;
}

int main() {
    freopen("party.in","r",stdin);
    freopen("party.out","w",stdout);
    n=read();m=read();q=read();dep[1]=1;
    fo(i,2,n) {
        fa[i]=read();
        dep[i]=dep[fa[i]]+1;
        add(fa[i],i);
    }
    fo(i,1,n) col[i]=read();
    dfs(1);make(1,1);
    build(1,n);
    for(;q;q--) {
        int c=read();
        fo(i,1,c) b[i]=read();
        int z=b[1];fo(i,2,c) z=lca(z,b[i]);
        fo(i,1,c) bit[i]=query(b[i],z);
        int ans=n+1;
        fo(s,1,(1<<c)-1) {
            bitset<M> tmp;int cnt=0;
            fo(i,1,c) if (s&(1<<i-1)) tmp|=bit[i],cnt++;
            ans=min(ans,(int)tmp.count()/cnt);
        }
        write(ans*c);
    }
    return 0;
}