{ }codelper
2D listsnested loopsP0033

Row and column sums

The first line gives the number of rows r and columns c, followed by r lines of c integers each.
On the first output line print the r row sums from top to bottom, and on the second line the c column sums from left to right, separated by single spaces.

Input
1 ≤ r, c ≤ 50, -1000 ≤ each integer ≤ 1000
Output
two lines: r row sums, c column sums
Example 1 · stdin
2 3
1 2 3
4 5 6
stdout
6 15
5 7 9
main.py
Input (stdin)

When you run, these lines are fed to input()/stdin in order (the program doesn't stop to ask).

$ Ready

Open this on a computer to write and run code. On mobile you can read the problem and the solutions.

In other languagesShow solutions by language · 8 languages

A row sum just adds one row, but for column sums the outer loop is the column (j) and the inner loop the row (i).
Swapping r and c only works for squares, so always test a rectangle like 2×3.
Building a 2D list: in Python [[0] * c] * r points to one row r times, so write [[0] * c for _ in range(r)].
C uses int g[50][50], C++ vector<vector<int>>, C# has both int[,] (cells are g[i, j]) and int[][], Java int[r][c].
Craykon adds each row with grid.Add(Console.ReadNumbers()) and reads grid[i][j].

Pythonmain.py · 6 lines
r, c = map(int, input().split())grid = [list(map(int, input().split())) for _ in range(r)]row_sums = [sum(row) for row in grid]col_sums = [sum(grid[i][j] for i in range(r)) for j in range(c)]print(*row_sums)print(*col_sums)
JavaScriptmain.js · 10 lines
const lines = require("fs").readFileSync(0, "utf8").trim().split("\n");const [r, c] = lines[0].trim().split(/\s+/).map(Number);const grid = lines.slice(1, r + 1).map((line) => line.trim().split(/\s+/).map(Number));const rowSums = grid.map((row) => row.reduce((s, x) => s + x, 0));const colSums = new Array(c).fill(0);for (let i = 0; i < r; i++) {  for (let j = 0; j < c; j++) colSums[j] += grid[i][j];}console.log(rowSums.join(" "));console.log(colSums.join(" "));
TypeScriptmain.ts · 11 lines
import { readFileSync } from "node:fs";const lines: string[] = readFileSync(0, "utf8").trim().split("\n");const [r, c]: number[] = lines[0].trim().split(/\s+/).map(Number);const grid: number[][] = lines.slice(1, r + 1).map((line) => line.trim().split(/\s+/).map(Number));const rowSums: number[] = grid.map((row) => row.reduce((s, x) => s + x, 0));const colSums: number[] = new Array<number>(c).fill(0);for (let i = 0; i < r; i++) {  for (let j = 0; j < c; j++) colSums[j] += grid[i][j];}console.log(rowSums.join(" "));console.log(colSums.join(" "));
JavaMain.java · 24 lines
import java.util.Scanner;import java.util.StringJoiner; public class Main {    public static void main(String[] args) {        Scanner sc = new Scanner(System.in);        int r = sc.nextInt(), c = sc.nextInt();        int[] rowSums = new int[r];        int[] colSums = new int[c];        for (int i = 0; i < r; i++) {            for (int j = 0; j < c; j++) {                int x = sc.nextInt();                rowSums[i] += x;                colSums[j] += x;            }        }        StringJoiner rows = new StringJoiner(" ");        for (int s : rowSums) rows.add(String.valueOf(s));        StringJoiner cols = new StringJoiner(" ");        for (int s : colSums) cols.add(String.valueOf(s));        System.out.println(rows);        System.out.println(cols);    }}
C#Main.cs · 21 lines
using System;using System.Linq; class Program{    static void Main()    {        int[] rc = Console.ReadLine().Split(' ', StringSplitOptions.RemoveEmptyEntries).Select(int.Parse).ToArray();        int r = rc[0], c = rc[1];        int[,] grid = new int[r, c];        for (int i = 0; i < r; i++)        {            int[] row = Console.ReadLine().Split(' ', StringSplitOptions.RemoveEmptyEntries).Select(int.Parse).ToArray();            for (int j = 0; j < c; j++) grid[i, j] = row[j];        }        var rowSums = Enumerable.Range(0, r).Select(i => Enumerable.Range(0, c).Sum(j => grid[i, j]));        var colSums = Enumerable.Range(0, c).Select(j => Enumerable.Range(0, r).Sum(i => grid[i, j]));        Console.WriteLine(string.Join(" ", rowSums));        Console.WriteLine(string.Join(" ", colSums));    }}
C++main.cpp · 22 lines
#include <iostream>#include <vector> int main() {    int r, c;    std::cin >> r >> c;    std::vector<std::vector<int>> grid(r, std::vector<int>(c));    for (auto& row : grid)        for (int& x : row) std::cin >> x;    for (int i = 0; i < r; i++) {        int sum = 0;        for (int j = 0; j < c; j++) sum += grid[i][j];        std::cout << (i ? " " : "") << sum;    }    std::cout << "\n";    for (int j = 0; j < c; j++) {        int sum = 0;        for (int i = 0; i < r; i++) sum += grid[i][j];        std::cout << (j ? " " : "") << sum;    }    std::cout << "\n";}
Cmain.c · 21 lines
#include <stdio.h> int main(void) {    int r, c, grid[50][50];    scanf("%d %d", &r, &c);    for (int i = 0; i < r; i++)        for (int j = 0; j < c; j++) scanf("%d", &grid[i][j]);    for (int i = 0; i < r; i++) {        int sum = 0;        for (int j = 0; j < c; j++) sum += grid[i][j];        printf(i ? " %d" : "%d", sum);    }    putchar('\n');    for (int j = 0; j < c; j++) {        int sum = 0;        for (int i = 0; i < r; i++) sum += grid[i][j];        printf(j ? " %d" : "%d", sum);    }    putchar('\n');    return 0;}
Craykonmain.cray · 30 lines
rc = Console.ReadNumbers();r = rc[0];c = rc[1];grid = [];for (i = 0; i < r; i++){    grid.Add(Console.ReadNumbers());}rowSums = [];for (i = 0; i < r; i++){    sum = 0;    for (j = 0; j < c; j++)    {        sum += grid[i][j];    }    rowSums.Add(sum);}colSums = [];for (j = 0; j < c; j++){    sum = 0;    for (i = 0; i < r; i++)    {        sum += grid[i][j];    }    colSums.Add(sum);}Console.WriteLine(rowSums);Console.WriteLine(colSums);