杭电5636 Shortest Path

题目链接：
http://acm.hdu.edu.cn/showproblem.php?pid=5636

题目大意：
有一条长度为n的链. 节点i和i+1之间有长度为1的边. 现在又新加了3条边, 每条边长度都是1. 给出m个询问, 每次询问两点之间的最短路.

解题思路：
将需要用的点的编号提取出来，最多8个（因为询问的点和和新增加边的点可能重复），然后将连边的点距离变为1，未连边的点就和下一个点连边，距离为编号之差，连完边后用floyd跑一遍，即可知道距离。

#include <vector>
#include <list>
#include <map>
#include <set>
#include <deque>
#include <queue>
#include <stack>
#include <bitset>
#include <algorithm>
#include <functional>
#include <numeric>
#include <utility>
#include <sstream>
#include <iostream>
#include <iomanip>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cstring>
#include <ctime>
#include <cassert>
#define RI(N) scanf("%d",&(N))
#define RII(N,M) scanf("%d %d",&(N),&(M))
#define RIII(N,M,K) scanf("%d %d %d",&(N),&(M),&(K))
#define Cl0(a) memset((a),0,sizeof(a))
using namespace std;
const int inf=1e9+7;
const int inf1=-1*1e9;
typedef long long LL;
int a[10];
long long int ans[100005];
int main()
{
    int t;
    RI(t);
    while(t--)
    {
        int n,m,s,t;
        RII(n,m);
        int a1,b1,a2,b2,a3,b3;
        RII(a1,b1);
        RII(a2,b2);
        RII(a3,b3);
        a[8]=-1;
        for(int sa=1; sa<=m; sa++)
        {
            RII(s,t);
            a[0]=a1;
            a[1]=b1;
            a[2]=a2;
            a[3]=b2;
            a[4]=a3;
            a[5]=b3;
            a[6]=s;
            a[7]=t;
            sort(a,a+8);
            int b[10],len=0,dp[8][8];
            for(int i=0; i<8; i++)
                for(int j=0; j<8; j++)
                {
                    if(i==j)  dp[i][j]=0;
                    else dp[i][j]=inf;
                }
            for(int i=0; i<8; i++)
            {
                if(a[i]!=a[i+1])
                {
                    b[len++]=a[i];
                }
            }

            for(int i=0; i<len-1; i++)
            {
                dp[i][i+1]=b[i+1]-b[i];
                dp[i+1][i]=b[i+1]-b[i];
            }
            int x=lower_bound(b,b+len,a1)-b;
            int y=lower_bound(b,b+len,b1)-b;
            dp[x][y]=1;
            dp[y][x]=1;
            x=lower_bound(b,b+len,a2)-b;
            y=lower_bound(b,b+len,b2)-b;
            dp[x][y]=1;
            dp[y][x]=1;
            x=lower_bound(b,b+len,a3)-b;
            y=lower_bound(b,b+len,b3)-b;
            dp[x][y]=1;
            dp[y][x]=1;
            x=lower_bound(b,b+len,s)-b;
            y=lower_bound(b,b+len,t)-b;
            for(int k=0; k<len; k++)
                for(int i=0; i<len; i++)
                    for(int j=0; j<len; j++)
                    {
                        dp[i][j]=min(dp[i][j],dp[i][k]+dp[k][j]);
                    }

            ans[sa]=dp[x][y];
        }
        long long int ans1=0;
        for(int i=1; i<=m; i++)
        {
            ans1=(ans1+ans[i]*i)%inf;
        }
        printf("%lld\n",ans1);
    }
    return 0;
}