PAT乙级 1012.数字分类

93

只有100ms,Java目测几乎不可能拿到满分了。研究了一波怎么让Java更快

题目


原题链接

给定一系列正整数,请按要求对数字进行分类,并输出以下5个数字:

A1 = 能被5整除的数字中所有偶数的和;

A2 = 将被5除后余1的数字按给出顺序进行交错求和,即计算n1-n2+n3-n4...;

A3 = 被5除后余2的数字的个数;

A4 = 被5除后余3的数字的平均数,精确到小数点后1位;

A5 = 被5除后余4的数字中最大数字。

输入格式:
每个输入包含1个测试用例。每个测试用例先给出一个不超过1000的正整数N,随后给出N个不超过1000的待分类的正整数。数字间以空格分隔。

输出格式:
对给定的N个正整数,按题目要求计算A1~A5并在一行中顺序输出。数字间以空格分隔,但行末不得有多余空格。

若其中某一类数字不存在,则在相应位置输出“N”。

输入样例1:
13 1 2 3 4 5 6 7 8 9 10 20 16 18

输出样例1:
30 11 2 9.7 9

输入样例2:
8 1 2 4 5 6 7 9 16

输出样例2:
N 11 2 N 9



解析


Java有一组没有过,因为时限只有100ms,实在是太难为Java了
在PTA上测试了输出空语句的Java,平均耗时都有90ms以上,所以除去启动时间,Java基本上没有剩下的时间了。
这个时候,不需要虚拟机的C++优势就出来了,只用了3ms就跑完了程序,同样需要虚拟机的Python则用来37ms。

重点不是能不能拿到满分,而是探讨如何给Java加速。
能不能再快一点
首先可以知道Scanner是不能用了,这个实在是太慢了,用了Scanner肯定是全部超时的。
所以要换用BufferedReaderInputStreamReader
这样大概能过2、3组。
经过测试,可以发现import的文件越多,Java启动越慢。那么只留下import这两个文件的语句。
虽然需要IOEcpetion,不过也可以用Ecpetion来代替,这样可以再少引入一个模块。

再优化下代码结构,和细节实现应该可以做到只有一组过不了。

代码


C++解法


#include <cmath>
#include <iomanip>
#include <iostream>
using namespace std;

const double eps = 1e-12;
int sgn(const double &x) {
    if (fabs(x) < eps)
        return 0;
    return x > 0 ? 1 : -1;
}
int double_round(double num) { return sgn(num) > 0 ? num + 0.5 : num - 0.5; }

int main() {
    cin.tie(0);
    cin.sync_with_stdio(false);

    int n;
    cin >> n;
    int A[5] = {0, 0, 0, 0, -90000};
    bool has[5] = {0, 0, 0, 0, 0};
    int flag = 1;
    int cnt = 0;
    for (int i = 0; i < n; ++i) {
        int a, b;
        cin >> a;
        b = a % 5;

        if (b == 0 && a % 2 == 0) {
            has[0] = 1;
            A[0] += a;
        }
        if (b == 1) {
            has[1] = 1;
            A[1] += flag * a;
            flag *= -1;
        }
        if (b == 2) {
            has[2] = 1;
            A[2]++;
        }
        if (b == 3) {
            has[3] = 1;

            A[3] += a;
            cnt++;
        }
        if (b == 4) {
            has[4] = 1;
            A[4] = A[4] > a ? A[4] : a;
        }
    }
    bool isFirst = true;
    for (int i = 0; i < 5; ++i) {
        if (isFirst) {
            isFirst = false;
        } else {
            cout << " ";
        }
        if (has[i])
            if (i == 3)
                cout << fixed << setprecision(1)
                     << (double)double_round((double)A[3] / cnt * 10) / 10;
            else
                cout << A[i];
        else
            cout << "N";
    }
    cout << endl;
    return 0;
}


Python解法


read = [int(i) for i in input().split(" ")]
n = read[0]
del read[0]

calc = [[], [], [], [], []]
for i in read:
    calc[i % 5].append(i)

ans = []
calc[0] = [i for i in calc[0] if i % 2 == 0]
calc[1] = [i * (-((idx % 2) * 2 - 1)) for (idx, i) in enumerate(calc[1])]

ans.append(str(sum(calc[0])) if len(calc[0]) > 0 else "N")
ans.append(str(sum(calc[1])) if len(calc[1]) > 0 else "N")
ans.append(str(len(calc[2])) if len(calc[2]) > 0 else "N")
ans.append(str(round(sum(calc[3]) / len(calc[3])*10)/10) if len(calc[3]) > 0 else "N")
ans.append(str(max(calc[4])) if len(calc[4]) > 0 else "N")

print(" ".join(ans))



Java解法


import java.io.BufferedReader;
import java.io.InputStreamReader;

class Main {
    public static void main(String[] args) throws Exception {
        BufferedReader input = new BufferedReader(new InputStreamReader(System.in));
        String[] read = input.readLine().split(" ");
        int n = Integer.valueOf(read[0]);

        boolean[] has = { false, false, false, false, false };
        int[] A = { 0, 0, 0, 0, -1001 };

        int flag = 1;
        int cnt = 0;
        for (int i = 0; i < n; ++i) {
            int a = Integer.valueOf(read[i + 1]);
            int b = a % 5;
            if (b == 0 && a % 2 == 0) {
                has[0] = true;
                A[0] += a;
            }
            if (b == 1) {
                has[1] = true;
                A[1] += a * flag;
                flag *= -1;
            }
            if (b == 2) {
                has[2] = true;
                A[2]++;
            }
            if (b == 3) {
                has[3] = true;
                A[3] += a;
                cnt++;
            }
            if (b == 4) {
                has[4] = true;
                A[4] = A[4]> a?A[4]:a;
            }
        }

        for (int i = 0; i < 5; ++i) {
            if (i>0)
                System.out.print(" ");
            if (has[i])
                if (i == 3)
                    System.out.print(Math.round((float) A[3] / cnt * 10) / 10.0);
                else
                    System.out.print(A[i]);
            else
                System.out.print("N");
        }
        System.out.print("\n");
    }
}
发布评论
  • 点击查看/关闭被识别为广告的评论