前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >挑战程序竞赛系列(48):4.2 推理与动态规划算法(1)

挑战程序竞赛系列(48):4.2 推理与动态规划算法(1)

作者头像
用户1147447
发布2019-05-26 09:42:02
2750
发布2019-05-26 09:42:02
举报
文章被收录于专栏:机器学习入门机器学习入门

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://cloud.tencent.com/developer/article/1434719

挑战程序竞赛系列(48):4.2 推理与动态规划算法(1)

详细代码可以fork下Github上leetcode项目,不定期更新。

练习题如下:

POJ 1082: Calendar Game

思路:

可以由2001年11月4号推出1,2,3号的输赢态,因为1,2,3没有任何选择,只能+1,那么10月份该如何?

对于10月份有两种情况,选择同月+1和下个月同天,它的策略就是找下一状态的必输态,找到任何一种必输态,对于当前状态就必赢。

所以递推式为:

代码语言:javascript
复制
dp[year][month][day]: 表示在某天的输赢态

dp[year][month][day] = {选择同月下一天} || {选择下月同一天}

注意一些边界条件,比如一年的最后一天,一个月的最后一天,以及闰年,大小月等。

代码如下:

代码语言:javascript
复制
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.util.StringTokenizer;

public class Main{

    String INPUT = "./data/judge/201709/P1082.txt";

    public static void main(String[] args) throws IOException {
        new Main().run();
    }

    static int[] leap_month    = {0, 31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
    static int[] nonleap_month = {0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};

    boolean[][][] dp = new boolean[200][13][32];

    void solve() {
        int T = ni();

        DP();
        while (T --> 0) {
            int year  = ni();
            int month = ni();
            int day   = ni();
            out.println(dp[year - 1900][month][day] ? "YES" : "NO");
        }
    }

    void DP() {
        dp[2001 - 1900][11][3] = true;
        dp[2001 - 1900][11][2] = false;
        dp[2001 - 1900][11][1] = true;
        for (int j = 4; j <= 30; ++j) dp[2001 - 1900][11][j] = true;

        // 2001
        for (int mon = 10; mon >= 1; --mon) {
            for (int j = nonleap_month[mon]; j >= 1; --j) {
                // 当前月是否为最后一天
                if (j == nonleap_month[mon]) {
                    dp[101][mon][j] |= !dp[101][mon + 1][1];
                }
                else {
                    dp[101][mon][j] |= !dp[101][mon][j + 1];
                }

                if (j <= nonleap_month[mon + 1]) {
                    dp[101][mon][j] |= !dp[101][mon + 1][j];
                }
            }
        }

        for (int year = 2000 - 1900; year >= 0; --year) {
            int[] month = leap(year) ? leap_month : nonleap_month;
            for (int mon = 12; mon >= 1; --mon) {
                for (int day = month[mon]; day >= 1; --day) {
                    if (mon == 12 && day == month[mon]) { // 1 年中的最后一天
                        dp[year][mon][day] = !dp[year + 1][1][1] || !dp[year + 1][1][day];
                    }
                    else if (day == month[mon]) { // 1个月中的最后一天
                        dp[year][mon][day] |= !dp[year][mon + 1][1];
                        if (day <= month[mon + 1]) {
                            dp[year][mon][day] |= !dp[year][mon + 1][day];
                        }
                    }
                    else {  // 其他
                        dp[year][mon][day] |= !dp[year][mon][day + 1];
                        if (day <= month[mon % 12 + 1]) {
                            dp[year][mon][day] |= !dp[year][mon % 12 + 1][day];
                        }
                    }
                }
            }
        }
    }

    boolean leap(int year){
        return (year % 4 == 0 && year % 100 != 0 || year % 400 == 0);
    }

    FastScanner in;
    PrintWriter out;

    void run() throws IOException {
        boolean oj;
        try {
            oj = ! System.getProperty("user.dir").equals("F:\\java_workspace\\leetcode");
        } catch (Exception e) {
            oj = System.getProperty("ONLINE_JUDGE") != null;
        }

        InputStream is = oj ? System.in : new FileInputStream(new File(INPUT));
        in = new FastScanner(is);
        out = new PrintWriter(System.out);
        long s = System.currentTimeMillis();
        solve();
        out.flush();
        if (!oj){
            System.out.println("[" + (System.currentTimeMillis() - s) + "ms]");
        }
    }

    public boolean more(){
        return in.hasNext();
    }

    public int ni(){
        return in.nextInt();
    }

    public long nl(){
        return in.nextLong();
    }

    public double nd(){
        return in.nextDouble();
    }

    public String ns(){
        return in.nextString();
    }

    public char nc(){
        return in.nextChar();
    }

    class FastScanner {
        BufferedReader br;
        StringTokenizer st;
        boolean hasNext;

        public FastScanner(InputStream is) throws IOException {
            br = new BufferedReader(new InputStreamReader(is));
            hasNext = true;
        }

        public String nextToken() {
            while (st == null || !st.hasMoreTokens()) {
                try {
                    st = new StringTokenizer(br.readLine());
                } catch (Exception e) {
                    hasNext = false;
                    return "##";
                }
            }
            return st.nextToken();
        }

        String next = null;
        public boolean hasNext(){
            next = nextToken();
            return hasNext;
        }

        public int nextInt() {
            if (next == null){
                hasNext();
            }
            String more = next;
            next = null;
            return Integer.parseInt(more);
        }

        public long nextLong() {
            if (next == null){
                hasNext();
            }
            String more = next;
            next = null;
            return Long.parseLong(more);
        }

        public double nextDouble() {
            if (next == null){
                hasNext();
            }
            String more = next;
            next = null;
            return Double.parseDouble(more);
        }

        public String nextString(){
            if (next == null){
                hasNext();
            }
            String more = next;
            next = null;
            return more;
        }

        public char nextChar(){
            if (next == null){
                hasNext();
            }
            String more = next;
            next = null;
            return more.charAt(0);
        }
    }
}
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2017年09月05日,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 挑战程序竞赛系列(48):4.2 推理与动态规划算法(1)
    • POJ 1082: Calendar Game
    领券
    问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档