URAL 1303. Minimal Coverage(DP)

题目链接

又是输出路径。。。这题完全受上题影响，感觉两个题差不多。。用了基本上一样的算法写了，这题比较纠结，就是卡内存啊。。。5000*5000的数组开不了。。然后没办法，水了好几次MLE，看了一下虎哥的思路，完全不是一个套路，我写复杂了。。我啪啪按他的思路来了一次，就是过不了第三组，貌似得交了一二十次了。。。我把5000*5000降了降，因为只要i<=j的时候有用，乱搞了搞，2500*5000就过了。。。真心不容易啊。。。

只要把flag[i][j]记录 那一段，然后 pre记录i，sz记录上一次的结尾，写的很麻烦。。。O(N^2)的复杂度。

#include <cstring>
#include <cstdio>
#include <string>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <map>
using namespace std;
#define INF 100000000
int flag[2601][5111];
int dp[5001];
int pre[5001];
int sz[5001];
struct node
{
    int x,y;
    int ax,ay;
}p[99999],s[1001];
bool cmp(node a,node b)
{
    if(a.ax == b.ax)
        return a.ay < b.ay;
    else
        return a.ax < b.ax;
}
int main()
{
    int i,j,n,m,x,y,minz,key;
    scanf("%d",&m);
    for(n = 0;; n ++)
    {
        scanf("%d%d",&x,&y);
        if(x == 0&&y == 0)
            break;
        p[n].x = x;
        p[n].y = y;
        if(p[n].x < 0)
            p[n].ax = 0;
        else
            p[n].ax = p[n].x;
        if(p[n].y > m)
            p[n].ay = m;
        else
            p[n].ay = p[n].y;
        if(p[n].ax > p[n].ay) continue;
        if(p[n].ax > 2500)
        flag[5010-p[n].ax][5010-p[n].ay] = n+1;
        else
        flag[p[n].ax][p[n].ay] = n+1;
    }
    for(i = 0; i <= m; i ++)
        dp[i] = INF;
    for(i = 0;i <= m;i ++)
    {
        if(flag[0][i])
        {
            pre[i] = 0;
            dp[i] = 1;
        }
    }
    for(i = 0; i <= m; i ++)
    {
        minz = INF;
        for(j = i - 1; j >= 0; j --)
        {
            if(minz > dp[j])
            {
                minz = dp[j];
                key = j;
            }
            int ti,tj;
            if(j > 2500)
            {
                ti = 5010 - i;
                tj = 5010 - j;
            }
            else
            {
                ti = i;
                tj = j;
            }
            if(flag[tj][ti])
            {
                if(dp[i] > minz + 1)
                {
                    pre[i] = j;
                    sz[i] = key;
                    dp[i] = minz + 1;
                }
            }
        }
    }
    if(dp[m] == INF)
        printf("No solution
");
    else
    {
        printf("%d
",dp[m]);
        int num = 0;
        while(m)
        {
            int ti,tj;
            if(pre[m] > 2500)
            {
                ti = 5010 - pre[m];
                tj = 5010 - m;
            }
            else
            {
                ti = pre[m];
                tj = m;
            }
            s[num++] = p[flag[ti][tj] - 1];
            m = sz[m];
        }
        sort(s,s+num,cmp);
        for(i = 0; i < num; i ++)
            printf("%d %d
",s[i].x,s[i].y);
    }
}