Java题解 dfs做法 P1018 [NOIP2000 提高组] 乘积最大

题目描述

今年是国际数学联盟确定的“ 2000 ――世界数学年”，又恰逢我国著名数学家华罗庚先生诞辰 90 周年。在华罗庚先生的家乡江苏金坛，组织了一场别开生面的数学智力竞赛的活动，你的一个好朋友 XZ 也有幸得以参加。活动中，主持人给所有参加活动的选手出了这样一道题目：

设有一个长度为 NN 的数字串，要求选手使用 KK 个乘号将它分成 K+1K+1 个部分，找出一种分法，使得这 K+1K+1 个部分的乘积能够为最大。

同时，为了帮助选手能够正确理解题意，主持人还举了如下的一个例子：

有一个数字串：312312， 当 N=3,K=1N=3,K=1 时会有以下两种分法：

1. 3 \times 12=363×12=36
2. 31 \times 2=6231×2=62

这时，符合题目要求的结果是: 31 \times 2 = 6231×2=62

现在，请你帮助你的好朋友 XZ 设计一个程序，求得正确的答案。

输入格式

程序的输入共有两行：

第一行共有 22 个自然数 N,KN,K（6≤N≤40,1≤K≤66≤N≤40,1≤K≤6）

第二行是一个长度为 NN 的数字串。

输出格式

结果显示在屏幕上，相对于输入，应输出所求得的最大乘积（一个自然数）。

输入输出样例

输入 #1复制

4  2
1231

输出 #1复制

62

说明/提示

NOIp2000提高组第二题

思路：数据量不大 直接dfs剪枝可过

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.math.BigInteger;
import java.util.HashSet;
import java.util.StringTokenizer;

public class Main {
    static int n, k;
    static StringBuffer sb;
    static HashSet<String> set = new HashSet<>();

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        k = sc.nextInt();
        sb = new StringBuffer(sc.next());
        dfs(0);
        System.out.println(max);

    }

    static BigInteger max = BigInteger.ZERO;

    static void dfs(int m) {
        //剪枝，没有这步就TLE 80分
        if (set.contains(sb.toString())) {
            return;
        }
        if (m > k) return;
        if (m == k) {
            print();
            return;
        }
        set.add(sb.toString());
        for (int i = 0; i < sb.length() - 1; i++) {
            if (sb.charAt(i + 1) == ' ' || sb.charAt(i) == ' ') {
                continue;
            }
            //添加分隔
            sb.insert(i + 1, " ");
            dfs(m + 1);
            //回溯
            sb.delete(i + 1, i + 2);
        }
    }

    static void print() {
//        System.out.println(sb);
//        System.out.println(set);
        String arr[] = sb.toString().split(" ");
        BigInteger sum = new BigInteger(arr[0]);
        for (int i = 1; i < arr.length; i++) {
            sum = sum.multiply(new BigInteger(arr[i]));
        }
        max = max.max(sum);
    }
    //快读
    static class Scanner {

        public BufferedReader reader;
        public StringTokenizer tokenizer;

        public Scanner(InputStream stream) {
            reader = new BufferedReader(new InputStreamReader(stream), 32768);
            tokenizer = null;
        }

        public String next() {
            while (tokenizer == null || !tokenizer.hasMoreTokens()) {
                try {
                    tokenizer = new StringTokenizer(reader.readLine());
                } catch (IOException e) {
                    throw new RuntimeException(e);
                }
            }
            return tokenizer.nextToken();
        }

        public String nextLine() {
            String str = null;
            try {
                str = reader.readLine();
            } catch (IOException e) {
                // TODO 自动生成的 catch 块
                e.printStackTrace();
            }
            return str;
        }

        public int nextInt() {
            return Integer.parseInt(next());
        }

        public long nextLong() {
            return Long.parseLong(next());
        }

        public Double nextDouble() {
            return Double.parseDouble(next());
        }

        public BigInteger nextBigInteger() {
            return new BigInteger(next());
        }
    }
}