Codechef July Challenge 2018 : Subway Ride

传送门

首先（想了很久之后）注意到一个性质：同一条边有多种颜色的话保留3种就可以了，这是因为假如最优解要求当前位置与相邻两条边都不相同，那么只要有3条边，就肯定可以满足这一点。

完事就做一个nlogn*3^4的倍增dp就行了……实际肯定是跑不满的（而且cc机子快）。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define MN 510000
using namespace std;

int n,m,l[MN],F[MN][],_F[MN][],v[MN][][][],num=,x,y,w,C[MN][],de[MN];
struct na{
    int y,w,ne;
}b[MN<<];
inline void add(int x,int y,int w){b[++num].y=y;b[num].w=w;b[num].ne=l[x];l[x]=num;}
void dfs(int x,int f){
    if (F[x][]) return;
    F[x][]=x;F[x][]=f;
    for (int i=;i<;i++)
    if (!(F[x][i]=F[F[F[x][i-]][i-]][])) break;
    _F[x][]=f;
    for (int i=;i<;i++)
    if (!(_F[x][i]=_F[_F[x][i-]][i-])) break;

    for (int i=l[x];i;i=b[i].ne)
    if (b[i].y!=f){
        dfs(b[i].y,x);
        if (!C[b[i].y][]) C[b[i].y][]=b[i].w;else
        if (!C[b[i].y][]) C[b[i].y][]=b[i].w;else
        if (!C[b[i].y][]) C[b[i].y][]=b[i].w;
    }
}
void DFS(int x,int f){
    if (de[x]) return;
    de[x]=de[f]+;

    for (int a=;a<;a++)
    for (int b=;b<;b++)
    if (a!=b) v[x][][a][b]=-1e8;

    int tmp1,tmp2,tmp3;
    for (int i=;i<;i++)
    if (_F[x][i])
    for (int a=;a<;a++)
    if (C[x][a])
    for (int b=;b<;b++)
    if (C[F[x][i]][b])
    for (int A=;A<;A++)
    if (C[F[x][i-]][A])
    if ((tmp1=v[x][i-][a][A])>=)
    for (int B=;B<;B++)
    if (C[_F[x][i-]][B])
    if ((tmp2=v[_F[x][i-]][i-][B][b])>=)
    if (v[x][i][a][b]<(tmp3=tmp1+tmp2+(C[F[x][i-]][A]!=C[_F[x][i-]][B]))) v[x][i][a][b]=tmp3;

    for (int i=l[x];i;i=b[i].ne)
    if (b[i].y!=x) DFS(b[i].y,x);
}
inline int ask(int x,int y){
    if (de[x]>de[y]) swap(x,y);
    int ans[][],la[];
    memset(ans,,sizeof(ans));
    memset(la,,sizeof(la));
    for (int i=;i>=;i--)
    if (de[_F[y][i]]>=de[x]){
        if (la[]==){
            for (int a=;a<;a++)
            for (int b=;b<;b++)
            if (ans[][b]<v[y][i][a][b])
            ans[][b]=v[y][i][a][b];
            la[]=F[y][i];
            y=_F[y][i];
        }else{
            int tmp[]={,,},TMP;
            for (int a=;a<;a++)
            if (C[la[]][a])
            for (int b=;b<;b++)
            if (C[y][b])
            for (int c=;c<;c++)
            if (C[F[y][i]][c])
            if (tmp[c]<(TMP=ans[][a]+v[y][i][b][c]+(C[la[]][a]!=C[y][b]))) tmp[c]=TMP;
            for (int a=;a<;a++) ans[][a]=tmp[a];

            la[]=F[y][i];
            y=_F[y][i];
        }
    }
    if (x==y){
        int tmp=;
        for (int a=;a<;a++) if (ans[][a]>tmp) tmp=ans[][a];
        return tmp;
    }
    int st[]={x,y};
    for (int i=;i>=;i--)
    if (_F[st[]][i]!=_F[st[]][i]){
        for (int j=;j<;j++){
            if (la[j]==){
                for (int a=;a<;a++)
                for (int b=;b<;b++)
                if (ans[j][b]<v[st[j]][i][a][b])
                ans[j][b]=v[st[j]][i][a][b];
                la[j]=F[st[j]][i];
                st[j]=_F[st[j]][i];
            }else{
                int tmp[]={-,-,-},TMP;
                for (int a=;a<;a++)
                if (C[la[j]][a])
                for (int b=;b<;b++)
                if (C[st[j]][b])
                for (int c=;c<;c++)
                if (C[F[st[j]][i]][c])
                if (tmp[c]<(TMP=ans[j][a]+v[st[j]][i][b][c]+(C[la[j]][a]!=C[st[j]][b]))) tmp[c]=TMP;
                for (int a=;a<;a++) ans[j][a]=tmp[a];

                la[j]=F[st[j]][i];
                st[j]=_F[st[j]][i];
            }
        }
    }
    int i=;
        for (int j=;j<;j++){
            if (la[j]==){
                for (int a=;a<;a++)
                for (int b=;b<;b++)
                if (ans[j][b]<v[st[j]][i][a][b])
                ans[j][b]=v[st[j]][i][a][b];
                la[j]=F[st[j]][i];
                st[j]=_F[st[j]][i];
            }else{
                int tmp[]={-,-,-},TMP;
                for (int a=;a<;a++)
                if (C[la[j]][a])
                for (int b=;b<;b++)
                if (C[st[j]][b])
                for (int c=;c<;c++)
                if (C[F[st[j]][i]][c])
                if (tmp[c]<(TMP=ans[j][a]+v[st[j]][i][b][c]+(C[la[j]][a]!=C[st[j]][b]))) tmp[c]=TMP;
                for (int a=;a<;a++) ans[j][a]=tmp[a];

                la[j]=F[st[j]][i];
                st[j]=_F[st[j]][i];
            }
        }
    int tmp=;
    for (int a=;a<;a++)
    for (int b=;b<;b++)
    if (tmp<ans[][a]+ans[][b]+(C[la[]][a]!=C[la[]][b])) tmp=ans[][a]+ans[][b]+(C[la[]][a]!=C[la[]][b]);
    return tmp;
}
int main(){
    scanf("%d%d",&n,&m);
    for (int i=;i<=m;i++){
        scanf("%d%d%d",&x,&y,&w);
        add(x,y,w);add(y,x,w);
    }
    dfs(,);DFS(,);
    //printf(">_<%d %d %d\n",C[3][0],C[3][1],C[3][2]);
    /*for (int i=1;i<=n;i++,puts("")){
        printf("%d:",i);
        for (int j=0;j<3;j++) printf("%d ",C[i][j]);
    }*/
    //printf("%d %d\n",v[6][2][0][0],_F[6][2]);
    scanf("%d",&m);
    while(m--){
        scanf("%d%d",&x,&y);
        printf("%d\n",ask(x,y));
    }
}