牛客编程巅峰赛S1第9场 - 王者 C 牛牛摆木棒

牛牛摆木棒

https://ac.nowcoder.com/acm/contest/6781/C

题意:求n的排列中第k个波浪形的排列

牛客巅峰赛系列里面少有的,还算有一点意思的题目(没错我就是在喷),可惜考场上时间不太够,想出来没码完(好吧确实是因为我一开始把找第k个的部分写麻烦了)。

首先来考虑求满足条件序列数量的问题,记表示长度为n,以k开头的先上升或先下降的序列数量。


预处理出来这个dp数组之后,按照跟dp相同的思路逐位确定最终答案即可。

#define ll long long 
class Solution {
public:
    /**
     * 
     * @param n int整型 木棒的个数
     * @param k long长整型 第k个排列
     * @return int整型vector
     */
    ll dp[30][30][2];
    vector<int> ask(int n, int now, ll k, int flag){
        vector<int> res, _res;
        if (n>1){
            if (flag == 0){
                for (int j = 1; j < now; j++){
                    long long tmp = dp[n-1][j][1];
                    if (k > tmp) k-=tmp;else{
                        res = ask(n-1, j, k, 1);
                        break;
                    }
                }
            }else{
                for (int j = now+1; j <= n; j++){
                    long long tmp = dp[n-1][j-1][0];
                    if (k > tmp) k-=tmp;else{
                        res = ask(n-1, j-1, k, 0);
                        break;
                    }
                }
            }
        }
        for (int i = 0; i < n-1; i++) res[i] += res[i] >= now;
        _res.push_back(now);
        for (int i = 0; i < n-1; i++) _res.push_back(res[i]);
        return _res;
    }
    vector<int> stick(int n, ll k) {
        memset(dp, 0, sizeof(dp));
        dp[1][1][0] = dp[1][1][1] = 1;
        for (int i = 2; i <= n; i++){
            for (int j = 1; j <= i; j++){
                for (int k = 1; k < j; k++){
                    dp[i][j][0] += dp[i-1][k][1];
                }
                for (int k = j+1; k <= i; k++){
                    dp[i][j][1] += dp[i-1][k-1][0];
                }
            }
        }
        for (int a = 1; a <= n; a++){
            if (dp[n][a][0] >= k){
                return ask(n, a, k, 0);
            }else k-=dp[n][a][0];
            if (dp[n][a][1] >= k){
                return ask(n, a, k, 1);
            }else k-=dp[n][a][1];
        }
    }
};
全部评论
萌新对这个dp感到疑惑,有几个问题想要请教大佬,希望能得到答复 1.这样的转移方程不会产生重复的情况吗。比如dp[4][2][1]的答案里有一组解是(2,3,1,4) 那么在更新dp[5][3][0]的时候为啥不会产生(3,2,3,1,4)这种解呀。。 2.还有就是这里的转移方程为什么是k-1,而不是k。 dp[i][j][1] += dp[i-1][k-1][0];
点赞 回复 分享
发布于 2020-08-07 00:36

相关推荐

07-03 11:02
中山大学 C++
字节刚oc,但距离九月秋招很近了有两段互联网实习,非腾讯字节。不敢赌转正,现在在纠结去还是不去如果实习俩月离职会有什么后果吗
阿城我会做到的:不去后悔一辈子,能否转正取决于ld的态度,只要他不卡,答辩就是走流程,个人觉得可以冲一把
投递字节跳动等公司8个岗位
点赞 评论 收藏
分享
06-13 10:15
门头沟学院 Java
想去夏威夷的大西瓜在...:我也是27届,但是我现在研一下了啥项目都没有呀咋办,哎,简历不知道咋写
点赞 评论 收藏
分享
牛客83700679...:简历抄别人的,然后再投,有反馈就是简历不行,没反馈就是学历不行,多投多改只要技术不差机会总会有的
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
07-07 13:47
机械打工仔:你自己匿名可以,这么好的公司就别给它匿名了
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务