AcWing 743. 数组中的行

AcWing 743. 数组中的行

写在前面：AcWing是由北大一神级人物——“大雪菜” 创办的算法交流社区, 里面除了正常 oj 网站的功能之外, 还提供单人训练、双人匹配、云端操作系统等模式，除此之外不定期举行各种打卡活动，而且可以分享题解和心得，完全可以当成空间来玩。欢迎大家入坑。
点击前往：AcWing

原题链接

题目描述
输入一个二维数组M[12][12]，根据输入的要求，求出二维数组中某一行的元素的平均值或元素的和。

输入格式
第一行输入整数L，表示所求的具体行数（行数从0开始计数）。
第二行包含一个大写字母，若为’S’，则表示需要求出第 L 行的元素的和，若为’M’，则表示需要求出第 L 行的元素的平均值。
接下来12行，每行包含12个用空格隔开的浮点数，表示这个二维数组，其中第 i+1 行的第 j+1 个数表示数组元素M[i][j]。

输出格式
输出一个数，表示所求的平均数或元素的和的值，保留一位小数。

数据范围
−100.0≤M[i][j]≤100.0

输入样例

7
S
8.3 -9.4 -5.6 9.9 2.2 3.2 6.9 7.7 -2.9 -6.3 3.9 -1.4
-0.5 4.3 -0.3 2.3 0.7 -3.1 -6.3 5.7 -7.7 -3.7 9.7 3.8
5.8 -9.1 3.5 7.0 2.5 -2.3 6.9 2.7 5.8 -4.6 -7.8 4.3
-1.7 -8.9 -2.6 7.5 -9.6 6.4 -4.7 -8.0 0.6 6.7 -1.8 -0.4
-9.8 8.9 -3.8 -2.1 5.8 -9.4 1.3 6.3 -1.3 -6.3 -6.6 8.8
-6.3 -7.2 -6.3 -5.9 6.2 9.3 2.0 -4.3 2.7 3.6 -4.6 8.7
3.8 7.4 8.7 -2.8 6.0 -9.2 1.9 3.1 -7.2 -6.0 9.7 -8.4
-9.8 6.6 8.6 2.3 1.1 -2.9 6.6 -1.5 2.1 -5.1 6.4 -7.5
-0.6 -1.6 9.5 -6.6 3.0 3.4 -7.5 -7.9 3.1 2.0 1.7 -7.6
-9.0 9.4 6.6 -0.2 1.4 6.5 8.6 2.0 -1.3 0.2 4.0 -2.4
-7.3 -6.3 -8.6 -7.5 8.5 1.4 -3.3 7.1 0.8 3.3 3.0 0.1
3.3 9.0 1.7 5.2 -3.7 8.9 3.8 -7.4 -2.8 0.6 -2.4 7.1

输出样例

6.9

解题思路
我怎么可能用数组呢, 果断直接判断输出啊

已过代码

# include <iostream>
using namespace std;
int main (){
    int n;cin>>n;
    string s;cin>>s;
    double ans=0.0, t;
    for (int i=0;i<12;i++){
        for (int j=0;j<12;j++){
            cin>>t;
            if (i==n) ans+=t;
        }
    }
    printf ("%.1f\n",s=="S"?ans:ans/12.0);
}