poj 3375 Network Connection

题目链接：http://poj.org/problem?id=3375

题目大意及思路：将n台电脑插到m个端口上，使距离最小，dp方程很容易想到，但直接计算会超时，所以需要优化，即对于一台电脑我们肯定想让它插到最近的端口上，但是为了总距离最短，可能这些端口被其它电脑占用了，又因为总电脑数为n,所以它的左右两端最多有n个端口被占用,所以只要计算它插到[pos-n,pos+n]的区间即可。

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<string>
#include<queue>
#include<math.h>
#include<algorithm>
#include<vector>
#include<stack>
#include<list>
#include<iostream>
#include<map>
using namespace std;
#define inf 0x3f3f3f3f
#define Max 110
int max(int a,int b)
{
	return a>b?a:b;
}
inline int min(int a,int b)
{
	return a<b?a:b;
}
int n,m;
int dp[2][100100];
int a[2010],b[101000];
int main()
{
    int i,j;
    scanf("%d%d",&m,&n);
        for(i=1;i<=m;i++)
            scanf("%d",&b[i]);
        for(i=1;i<=n;i++)
            scanf("%d",&a[i]);
        sort(b+1,b+m+1);
        sort(a+1,a+n+1);
        for(i=0;i<=m;i++)
            dp[0][i]=0;
       // dp[0][0]=0;
        int tmp=0;
        int pos=1;
        int st1,st2,ed1,ed2;
        st1=0;ed1=m;
        for(i=1;i<=n;i++)
        {
            for(;pos<=m;pos++)
            {
                if(b[pos]>a[i])
                    break;
            }
            st2=max(i,pos-n-10);
            ed2=min(m,pos+n+10);
            dp[1-tmp][st2-1]=inf;
            for(j=st2;j<=ed2;j++)
            {
               // dp[1-tmp][j]=inf;
                if(j>=st1&&j<=ed1)
                dp[1-tmp][j]=min(dp[tmp][j-1]+abs(a[i]-b[j]),dp[1-tmp][j-1]);
                else
                    dp[1-tmp][j]=min(dp[tmp][ed1]+abs(a[i]-b[j]),dp[1-tmp][j-1]);
               // printf("i %d j %d dp %d\n",i,j,dp[1-tmp][j]);
            }
            st1=st2;
            ed1=ed2;
            tmp=1-tmp;
        }
        int ans=inf;
        printf("%d\n",dp[tmp][ed1]);
}