洛谷P1220 关路灯(区间dp)

题目链接


先记录一下功耗前缀和，每次关灯我们只在乎是哪个区间被关了，关完时刻人在哪个端点。定义dp[i][j][0/1] 为人站在i/j 时关闭区间i j的灯所产生的最小花费。那么就有

ll a[111111];
ll b[111111];
ll sum[111111];
ll dp[1001][1001][3];
signed main()
{


    ll n,pos;
    read(n);
    read(pos);
    for(int i=1; i<=n; i++)
    {
        read(a[i]);
        read(b[i]);
        sum[i]=sum[i-1]+b[i];
    }
    memset(dp,inf,sizeof(dp));
    dp[pos][pos][0]=dp[pos][pos][1]=0;


    for(int len=1; len<=n-1; len++)
    {

        for(int i=1; i<=n; i++)
        {

            ll j=i+len;
            if(j>n)
                break;
            dp[i][j][0]=min(dp[i+1][j][0]+(a[i+1]-a[i])*(sum[i]+sum[n]-sum[j]),dp[i+1][j][1]+(a[j]-a[i])*(sum[i]+sum[n]-sum[j]));
            dp[i][j][1]=min(dp[i][j-1][0]+(a[j]-a[i])*(sum[i-1]+sum[n]-sum[j-1]),dp[i][j-1][1]+(a[j]-a[j-1])*(sum[i-1]+sum[n]-sum[j-1]));

        }

    }
    printf("%lld\n",min(dp[1][n][0],dp[1][n][1]));

}