CodeForces 10 B.Cinema Cashier（dp）

Description

一个电影院有$n$排座位，每排有$n$个座位，其中$n$为奇数，用$(i,j)$表示第$i$排第$j$个座位，令$x_c=y_c=\frac{n+1}{2}$表示中间一排的中间座位，假如当前来了一个公司的$M$人，那么这$M$人需要坐在同一排的连续位置，假设坐在第$x$排的第$y_l$个位置到第$y_r$个位置，那么需要使得$\sum\limits_{i+y_l}^{y_r}|x-x_c|+|i-y_c|$最小，对于该值最小的多种选择，首先选择$x$较小的，然后选择$y_l$较小的，现在有$q$个公司的人以此来到电影院看电影，第$i$个来的公司有$M_i$个人，问每个公司的人做的位置$x,y_l,y_r$是多少，如果坐不下这个公司的人就不看电影了，输出$-1$

Input

第一行两个整数$q,n$，之后$q$行每行一整数$M_i$$(1\le q\le 1000,1\le n\le 99,1\le M_i\le n)$

Output

如果当前剩余位置可以坐下则输出该公司人坐的位置，否则输出$-1$

Sample Input

4 3
1 2 3 1

Sample Output

2 2 2
1 1 2
3 1 3
2 1 1

Solution

以$dp[i][j][k]$表示$k$个人坐在第$i$排第$j$个位置到第$j+k-1$个位置时离中心的距离之和，即$dp[i][j][k]=\sum\limits_{l=j}^{{j+k-1}}|i-x_c|+|l-y_c|$，轻易得到转移$dp[i][j][k]=dp[i][j][k-1]+|i-x_c|+|j+k-1-y_c|$，$O(n^3)$得到$dp$数组，对于一个查询$M$，枚举$i,j$找到$dp[i][j][M]$的最小值即为答案，如果找不到就输出$-1$，注意找到位置后，这些位置就被占了，所以要更新被影响的$dp$值，假设这$M$个人坐在第$x$排第$y$个位置到第$y+M-1$个位置，那么对于$dp[x][i][j]$，如果区间$[i,i+j-1]$和区间$[y,y+M-1]$有交，则$dp[x][i][j]$就没有意义了，将其变成不合法值，单次更新时间复杂度$O(n^2)$，总时间复杂度$O(n^3+q\cdot n^2)$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100;
int q,n,dp[maxn][maxn][maxn];
int main()
{
    scanf("%d%d",&q,&n);
    int m=(n+1)/2;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            for(int k=1;k<=n;k++)
                dp[i][j][k]=INF;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        {
            dp[i][j][1]=abs(i-m)+abs(j-m);
            for(int k=2;k<=n-j+1;k++)
                dp[i][j][k]=dp[i][j][k-1]+abs(i-m)+abs(j+k-1-m);
        }
    while(q--)
    {
        int num;
        scanf("%d",&num);
        int ans=INF,x,y;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                if(dp[i][j][num]<ans)
                    ans=dp[i][j][num],x=i,y=j;
        if(ans==INF)printf("-1\n");
        else 
        {
            printf("%d %d %d\n",x,y,y+num-1);
            for(int i=1;i<=n;i++)
                for(int j=1;j<=n-i+1;j++)
                    if(!(i+j-1<y||i>y+num-1))
                        dp[x][i][j]=INF;
        }
    }
    return 0;
}