清北暑假模拟day1 生活

最新推荐文章于 2024-07-11 22:19:45 发布

weixin_30611509

最新推荐文章于 2024-07-11 22:19:45 发布

阅读量54

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/hyfer/p/5978501.html

本文介绍了一种利用动态规划解决数字三角形中寻找第K大数值的方法。通过预处理和逐步更新策略，确保了算法的高效性和正确性。文章提供了完整的C++代码实现，并解释了核心思想。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

/*
数字三角形，要求第K大的值，可以推知，如果得知k的范围，那么一定是在上一行可转移状态的对应范围内（反证法可以证明），这个在背包九讲里也有提及
*/
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>

using namespace std;

int n,k,f[110][110][15],z[110][110],y[1010];

int main()
{
    freopen("live.in","r",stdin);
    freopen("live.out","w",stdout);

    scanf("%d%d",&n,&k);
    for (int a=1;a<=n;a++)
        for (int b=1;b<=a;b++)
            scanf("%d",&z[a][b]);
    for (int a=1;a<=n;a++)
        for (int b=1;b<=a;b++)
            for (int c=1;c<=k;c++)
                f[a][b][c]=-12345;
    f[1][1][1]=z[1][1];
    for (int a=1;a<n;a++)
        for (int b=1;b<=a;b++)
            for (int c=1;c<=k;c++)
                if (f[a][b][c]!=-12345)
                {
                    int v=f[a][b][c]+z[a+1][b];
                    for (int d=1;d<=k;d++)
                        if (v>=f[a+1][b][d])
                        {
                            for (int e=k;e>d;e--)
                                f[a+1][b][e]=f[a+1][b][e-1];
                            f[a+1][b][d]=v;
                            break;
                        }
                    v=f[a][b][c]+z[a+1][b+1];
                    for (int d=1;d<=k;d++)
                        if (v>=f[a+1][b+1][d])
                        {
                            for (int e=k;e>d;e--)
                                f[a+1][b+1][e]=f[a+1][b+1][e-1];
                            f[a+1][b+1][d]=v;
                            break;
                        }
                }
    int cnt=0;
    for (int a=1;a<=n;a++)
        for (int b=1;b<=k;b++)
            y[++cnt]=f[n][a][b];
    sort(y+1,y+cnt+1);
    printf("%d\n",y[cnt-k+1]);

    return 0;
}