BZOJ1138: [POI2009]Baj 最短回文路

开始想建个n^2个点的图跑最短路来着…因为显然会T，一直在想优化结果不会就弃疗了..

dp
f[i][j]表示i到j的最短回文路长度
转移：$f[i][j]=2+min_c(f[u][v]),u∈S,v∈T$，其中S是i走c边能到达的点集，T是走c边能到达j的点集
发现转移是$O(n^2)$的
但我们可以通过dp时顺便处理一个min平衡一下时空复杂度
g[i][j][c]表示$min(f[i][v])，v∈T$，T的定义同上
这样转移的复杂度就变成$O(n)$了
转移时f和g开两个队列互相更新

code：

#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<cmath>
#include<ctime>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<climits>
#include<complex>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;

const int maxn = 410;

int n,m;
vector<int>V1[maxn][26],V2[maxn][26];
int f[maxn][maxn],g[maxn][maxn][26];

char str[110];

struct node{int i,j,c;};
queue<node>q1,q2;
void updf(int i,int j,int c)
{
    if(f[i][j]==-1) f[i][j]=c,q1.push((node){i,j,0});
    else if(f[i][j]>c) f[i][j]=c;
}
void updg(int i,int j,int k,int c)
{
    if(g[i][j][k]==-1) g[i][j][k]=c,q2.push((node){i,j,k});
    else if(g[i][j][k]>c) g[i][j][k]=c;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        int x,y; scanf("%d%d",&x,&y); scanf("%s",str);
        int c=str[0]-'a';
        V1[x][c].push_back(y),V2[y][c].push_back(x);
    }
    memset(f,-1,sizeof f); memset(g,-1,sizeof g);
    for(int i=1;i<=n;i++)
    {
        updf(i,i,0);
        for(int c=0;c<26;c++) for(int j=0;j<V1[i][c].size();j++)
            updf(i,V1[i][c][j],1);
    }
    int ii=0;
    while(!q1.empty()||!q2.empty())
    {
        if(!ii)
        {
            while(!q1.empty())
            {
                const node now=q1.front(); q1.pop();
                int tmp=f[now.i][now.j];
                for(int c=0;c<26;c++) for(int j=0;j<V1[now.j][c].size();j++)
                    updg(now.i,V1[now.j][c][j],c,tmp+2);
            }
        }
        else
        {
            while(!q2.empty())
            {
                const node now=q2.front(); q2.pop();
                int tmp=g[now.i][now.j][now.c];
                for(int j=0;j<V2[now.i][now.c].size();j++)
                    updf(V2[now.i][now.c][j],now.j,tmp);
            }
        }
        ii=!ii;
    }
    scanf("%d",&m);
    for(int i=1,la=0;i<=m;i++)
    {
        int x; scanf("%d",&x);
        if(la) printf("%d\n",f[la][x]);
        la=x;
    }

    return 0;
}