#185669 (C++17) No.783 門松計画

• 問題: No.783 門松計画
• ユーザー: mai
• 提出日時: 2017-07-01 12:41:30
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 22 ms / 2,000 ms
• コード長: 4,934 bytes
• コンパイル時間: 3,370 ms
• コンパイル使用メモリ: 220,712 KB
• 実行使用メモリ: 5,248 KB
• 最終ジャッジ日時: 2024-11-15 04:45:52
• 合計ジャッジ時間: 4,594 ms
• ジャッジサーバーID （参考情報）: judge1 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	3 ms; 5,248 KB
testcase_02	AC	2 ms; 5,248 KB
testcase_03	AC	2 ms; 5,248 KB
testcase_04	AC	2 ms; 5,248 KB
testcase_05	AC	2 ms; 5,248 KB
testcase_06	AC	2 ms; 5,248 KB
testcase_07	AC	2 ms; 5,248 KB
testcase_08	AC	2 ms; 5,248 KB
testcase_09	AC	2 ms; 5,248 KB
testcase_10	AC	10 ms; 5,248 KB
testcase_11	AC	22 ms; 5,248 KB
testcase_12	AC	13 ms; 5,248 KB
testcase_13	AC	5 ms; 5,248 KB
testcase_14	AC	8 ms; 5,248 KB
testcase_15	AC	5 ms; 5,248 KB
testcase_16	AC	3 ms; 5,248 KB
testcase_17	AC	5 ms; 5,248 KB
testcase_18	AC	2 ms; 5,248 KB
testcase_19	AC	6 ms; 5,248 KB
testcase_20	AC	2 ms; 5,248 KB
testcase_21	AC	3 ms; 5,248 KB
testcase_22	AC	2 ms; 5,248 KB
testcase_23	AC	2 ms; 5,248 KB
testcase_24	AC	5 ms; 5,248 KB
testcase_25	AC	4 ms; 5,248 KB
testcase_26	AC	5 ms; 5,248 KB

コンパイルメッセージ

main.cpp: In member function '{anonymous}::MaiScanner& {anonymous}::MaiScanner::operator>>(std::string&)':
main.cpp:72:9: warning: no return statement in function returning non-void [-Wreturn-type]
   72 |         }
      |         ^

ソースコード

#pragma GCC optimize ("O3")
#pragma GCC target ("avx")
#include "bits/stdc++.h" // define macro "/D__MAI"

using namespace std;
typedef long long int ll;

#define debugv(v) printf("L%d %s => ",__LINE__,#v);for(auto e:v){cout<<e<<" ";}cout<<endl;
#define debugm(m) printf("L%d %s is..\n",__LINE__,#m);for(auto v:m){for(auto e:v){cout<<e<<" ";}cout<<endl;}
#define debuga(m,w) printf("L%d %s is => ",__LINE__,#m);for(int x=0;x<(w);x++){cout<<(m)[x]<<" ";}cout<<endl;
#define debugaa(m,w,h) printf("L%d %s is..\n",__LINE__,#m);for(int y=0;y<(h);y++){for(int x=0;x<(w);x++){cout<<(m)[x][y]<<" ";}cout<<endl;}
#define debugaar(m,w,h) printf("L%d %s is..\n",__LINE__,#m);for(int y=0;y<(h);y++){for(int x=0;x<(w);x++){cout<<(m)[y][x]<<" ";}cout<<endl;}
#define ALL(v) (v).begin(),(v).end()
#define repeat(l) for(auto cnt=0;cnt<(l);++cnt)
#define iterate(b,e) for(auto cnt=(b);cnt!=(e);++cnt)
#define MD 1000000007ll
#define PI 3.1415926535897932384626433832795
template<typename T1, typename T2>
ostream& operator <<(ostream &o, const pair<T1, T2> p) { o << "(" << p.first << ":" << p.second << ")"; return o; }

#define TIME chrono::system_clock::now()
#define MILLISEC(t) (chrono::duration_cast<chrono::milliseconds>(t).count())
namespace {
    std::chrono::system_clock::time_point ttt;
    void tic() { ttt = TIME; }
    void toc() { fprintf(stderr, "TIME : %lldms\n", MILLISEC(TIME - ttt)); }
    std::chrono::system_clock::time_point tle = TIME;
#ifdef __MAI
    void safe_tle(int msec) { assert(MILLISEC(TIME - tle) < msec); }
#else
#define safe_tle(k) ;
#endif
}

#ifdef __MAI
#define getchar_unlocked getchar
#define putchar_unlocked putchar
#endif
#ifdef __VSCC
#define getchar_unlocked _getchar_nolock
#define putchar_unlocked _putchar_nolock
#endif
namespace {
#define isvisiablechar(c) (0x21<=(c)&&(c)<=0x7E)
    class MaiScanner {
    public:
        template<typename T>
        void input_integer(T& var) {
            var = 0;
            T sign = 1;
            int cc = getchar_unlocked();
            for (; cc<'0' || '9'<cc; cc = getchar_unlocked())
                if (cc == '-') sign = -1;
            for (; '0' <= cc&&cc <= '9'; cc = getchar_unlocked())
                var = (var << 3) + (var << 1) + cc - '0';
            var = var*sign;
        }
        inline int c() { return getchar_unlocked(); }
        inline MaiScanner& operator>>(int& var) {
            input_integer<int>(var);
            return *this;
        }
        inline MaiScanner& operator>>(long long& var) {
            input_integer<long long>(var);
            return *this;
        }
        inline MaiScanner& operator>>(string& var) {
            int cc = getchar_unlocked();
            for (; !isvisiablechar(cc); cc = getchar_unlocked());
            for (; isvisiablechar(cc); cc = getchar_unlocked())
                var.push_back(cc);
        }
    };
}
MaiScanner scanner;


// -----------------------------------------------------------------------
// ここから！
// -----------------------------------------------------------------------


inline bool kadomatu(int a, int b, int c) {
    return (a != c) && ((a<b&&b>c) || (a>b&&b<c));
}



int n, cap;

int price[111], len[111];

// TODO: N<3を得てしまうので，個数を数えてごまかす
// TODO: 落とせるかも？
// 
int dfs(int cost = 0, int pre = 0, int prepre = 0) {
    static unordered_map<int, int> dp;

    int key = (cost << 24) | (pre << 8) | prepre;
    if (dp.count(key)) return dp[key];

    int best = 0;

    for (int i = 0; i < n; ++i) {
        if (cost + price[i] > cap) continue;
        if (prepre == 0 || kadomatu(len[i], pre, prepre)) {
            best = max(best, dfs(cost + price[i], len[i], pre) + len[i]);
        }
    }
    return best;
}

int dp[51][51][51];
int main() {

    scanner >> n >> cap;

    repeat(n) {
        scanner >> len[cnt];
    }

    repeat(n) {
        scanner >> price[cnt];
    }

    // cout << dfs() << endl;

    int result = 0;

    for (int i = 0; i < n; ++i) {
        if (price[i] <= cap)
            dp[price[i]][len[i]][0] = len[i];
    }

    for (int cost = 0; cost < cap; ++cost) {
        for (int pre = 0; pre <= 50; ++pre) {
            for (int prepre = 0; prepre <= 50; ++prepre) {
                for (int i = 0; i < n; ++i) {
                    if (cost + price[i] > cap) continue;
                    if (dp[cost][pre][prepre] == 0) continue;

                    if (prepre == 0 || kadomatu(len[i], pre, prepre)) {
                        dp[cost + price[i]][len[i]][pre]
                            = max(
                                dp[cost + price[i]][len[i]][pre],
                                dp[cost][pre][prepre] + len[i]);

                        if (prepre != 0)
                            result = max(result, dp[cost + price[i]][len[i]][pre]);
                    }
                }
            }
        }
    }

    cout << result << endl;


    return 0;
}