PAT乙级真题 1092 最好吃的月饼 C++实现

题目

月饼是久负盛名的中国传统糕点之一，自唐朝以来，已经发展出几百品种。
若想评比出一种“最好吃”的月饼，那势必在吃货界引发一场腥风血雨…… 在这里我们用数字说话，给出全国各地各种月饼的销量，要求你从中找出销量冠军，认定为最好吃的月饼。
输入格式：
输入首先给出两个正整数 N（≤1000）和 M（≤100），分别为月饼的种类数（于是默认月饼种类从 1 到 N 编号）和参与统计的城市数量。
接下来 M 行，每行给出 N 个非负整数（均不超过 1 百万），其中第 i 个整数为第 i 种月饼的销量（块）。数字间以空格分隔。
输出格式：
在第一行中输出最大销量，第二行输出销量最大的月饼的种类编号。如果冠军不唯一，则按编号递增顺序输出并列冠军。数字间以 1 个空格分隔，行首尾不得有多余空格。
输入样例：
5 3
1001 992 0 233 6
8 0 2018 0 2008
36 18 0 1024 4
输出样例：
2018
3 5

思路

用一个数组记录总销量，累加M行数据，记录最大值。遍历数组，输出等于最大值的编号。

总量最大值不超过10 ^ 2 * 10 ^ 6，可以用int存储。

代码

#include <iostream>
#include <vector>
using namespace std;

int main(){
    int n, m;
    cin >> n >> m;
    vector<int> a(n);
    int maxNum = -1;
    for (int i=0; i<m; i++){
        for (int j=0; j<n; j++){
            int temp;
            cin >> temp;
            a[j] += temp;
            if (a[j]>maxNum){
                maxNum = a[j];
            }
        }
    }
    cout << maxNum << endl;
    vector<int> b;  //记录最大值id
    for (int i=0; i<n; i++){
        if (a[i]==maxNum){
            b.push_back(i+1);
        }
    }
    for (int i=0; i<b.size(); i++){
        if (i==0){
            cout << b[i];
        }
        else {
            cout << " " << b[i];
        }
    }
    cout << endl;
    return 0;
}