codeforces590E Birthday【AC自动机+Floyd+匈牙利算法】

因为没有重复串,所以把有包含关系的串连边之后是个DAG,也就是二分图,就变成求二分图的最大独立集=n-最小点覆盖=n-最大匹配
关于包含关系,建出AC自动机,然后把串放上去找子串,但是如果每次都一路找到根就会T,所以每次只找最近的一个,并且对于没有结尾id的点承接father的id,这样就O(1)的找到最近子串了
然后再用floyd传递闭包把关系建出图来
然后跑匈牙利,输出方案就是把一个匹配环里同一侧的都dfs标记一下,最后输出没有被标记的

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
#include<string>
#include<bitset>
using namespace std;
const int N=2005;
int n,ch[10000005][2],tot=1,h[N],cnt,fa[10000005],id[10000005],ans,lk[N],to[N],v[N],ti,mx[N],my[N];
string s[755];
bitset<N>a[N];
struct qwe
{
    int ne,to;
}e[N*N];
void add(int u,int v)
{//cerr<<u<<" "<<v<<endl;
    cnt++;
    e[cnt].ne=h[u];
    e[cnt].to=v;
    h[u]=cnt;
}
bool dfs(int u)
{
    for(int i=h[u];i;i=e[i].ne)
        if(v[e[i].to]!=ti)
        {
            v[e[i].to]=ti;
            if(!lk[e[i].to]||dfs(lk[e[i].to]))
            {
                lk[e[i].to]=u,to[u]=e[i].to;
                return 1;
            }
        }
    return 0;
}
void wk(int u)
{
    if(mx[u])
        return;
    mx[u]=1;
    for(int i=h[u];i;i=e[i].ne)
        if(!my[e[i].to])
        {
            my[e[i].to]=1;
            wk(lk[e[i].to]);
        }
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        cin>>s[i];
        int nw=1;
        for(int j=0;j<s[i].length();j++)
        {
            if(!ch[nw][s[i][j]-'a'])
                ch[nw][s[i][j]-'a']=++tot;
            nw=ch[nw][s[i][j]-'a'];
        }
        id[nw]=i;
    }
    queue<int>q;
    for(int i=0;i<2;i++)
    {
        if(ch[1][i])
            fa[ch[1][i]]=1,q.push(ch[1][i]);
        else
            ch[1][i]=1;
    }
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(int i=0;i<2;i++)
        {
            if(ch[u][i])
            {
                fa[ch[u][i]]=ch[fa[u]][i];
                q.push(ch[u][i]);
                if(!id[ch[u][i]])
                    id[ch[u][i]]=id[fa[ch[u][i]]];
            }
            else
                ch[u][i]=ch[fa[u]][i];
        }
    }
    for(int i=1;i<=n;i++)
    {
        int nw=1;
        for(int j=0;j<s[i].length();j++)
        {
            nw=ch[nw][s[i][j]-'a'];
            if(id[nw]&&id[nw]!=i)
                a[i][id[nw]]=1;//,cerr<<" "<<i<<" "<<id[nw]<<endl;
            if(id[fa[nw]]&&id[fa[nw]])
                a[i][id[fa[nw]]]=1;//,cerr<<" "<<i<<" "<<id[fa[nw]]<<endl;;
        }
    }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            if(a[j][i])
                a[j]|=a[i];
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            if(i!=j&&a[i][j])
                add(i,j);
    for(int i=1;i<=n;i++)
    {
        ti++;
        if(dfs(i))
            ans++;
    }
    printf("%d\n",n-ans);
    for(int i=1;i<=n;i++)
        if(!to[i])
            wk(i);
    for(int i=1;i<=n;i++)
        if(mx[i]&&!my[i])
            printf("%d ",i);
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/lokiii/p/10961486.html