Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> C++

โค้ด C++ หาผลรวมขององค์ประกอบพิเศษในเมทริกซ์

สมมติว่าเราได้รับเมทริกซ์สี่เหลี่ยมขนาด n * n ค่าต่อไปนี้ในเมทริกซ์เรียกว่า "องค์ประกอบพิเศษ":

  • ค่าที่อยู่บนเส้นทแยงมุมหลัก
  • ค่าที่อยู่บนเส้นทแยงมุมรอง
  • ค่าของแถวที่มีแถวด้านบนและด้านล่างเท่ากัน (n - 1 / 2)
  • ค่าของคอลัมน์ที่มีคอลัมน์ซ้ายและขวาเท่ากัน (n - 1 / 2)

เราต้องหาผลรวมของค่าพิเศษเหล่านี้ในเมทริกซ์

เช่น ถ้าค่าที่ป้อนเข้าเป็น n = 4, mat = {{1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10, 11, 12}, {13, 14, 15, 16}} ผลลัพธ์จะได้ 107

ขั้นตอนการแก้ปัญหา

เพื่อแก้ปัญหานี้ เราจะทำตามขั้นตอนดังนี้:

res := 0 สำหรับ i := 0 เมื่อ i < n เพิ่ม i ทีละ 1 ทำ:    สำหรับ j := 0 เมื่อ j < n เพิ่ม j ทีละ 1 ทำ:    ถ้า i เท่ากับ j หรือ i เท่ากับ n / 2 หรือ j เท่ากับ n / 2 หรือ i + j เท่ากับ n - 1 แล้ว:       res := res + mat[i, j] พิมพ์(res)

ตัวอย่างโค้ด

ดูการนำไปใช้จริงด้านล่างเพื่อความเข้าใจที่ดียิ่งขึ้น:

#include <bits/stdc++.h> using namespace std; #define N 100 void solve(int n, vector<vector<int>> mat) {    int res = 0;    for (int i = 0; i < n; i++)    for (int j = 0; j < n; j++){       if (i == j || i == n / 2 || j == n / 2 || i + j == n - 1)          res += mat[i][j];    }    cout << res << endl; } int main() {    int n = 4;    vector<vector<int>> mat = {{1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10, 11, 12}, {13, 14, 15, 16}};    solve(n, mat);    return 0; }

ข้อมูลนำเข้า

4, {{1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10, 11, 12}, {13, 14, 15, 16}}

ข้อมูลส่งออก

107