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

XOR ของแบบสอบถามเมทริกซ์ย่อยใน C++


ในปัญหานี้ เราได้รับเมทริกซ์ N x N และเคียวรีบางรายการ แต่ละเคียวรีมีมุมบนซ้ายและมุมขวาล่างของเมทริกซ์ย่อยที่สร้างจากเมทริกซ์นี้ งานของเราคือค้นหา XOR ขององค์ประกอบทั้งหมดของเมทริกซ์ย่อยที่กำหนดโดยข้อความค้นหา

มาดูตัวอย่างเพื่อทำความเข้าใจปัญหากัน

อินพุต

arr[][] = {{1, 2, 3}
{4, 5, 6}
{7, 8, 9}}
Querries: {0,0, 1,2} , {1, 2, 2, 2}

ผลลัพธ์

1 15

คำอธิบาย

querry 1 : 1^2^3^4^5^6
querry 2 : 6^9

เพื่อแก้ปัญหานี้ เราจะหาเมทริกซ์ prefix-XOR เพื่อแก้ปัญหาการสืบค้น ค่าของเมทริกซ์ที่ตำแหน่ง (R, C) คือ XOR ของเมทริกซ์ย่อยจากมุมบนซ้าย (0, 0) และมุมล่างขวาที่ตำแหน่ง (R, C)

ก่อนอื่นเราจะพบ คำนำหน้า -XOR สำหรับเมทริกซ์ทุกแถวทีละแถว จากนั้นคำนวณคำนำหน้า XOR สำหรับแต่ละคอลัมน์ทีละรายการ

สำหรับการค้นหา XOR ของเมทริกซ์ย่อยของแบบสอบถามที่กำหนดโดย (r1, c1) ถึง (r2, c2) คำนวณโดยใช้ prefixXor[r2][c2] ^ prefixXor[r1-1][c2] ^ prefixXor[r2][c1 -1] ^ prefixXor[r1-1][c1-1].

โปรแกรมแสดงการใช้งานโซลูชันของเรา

ตัวอย่าง

#include <iostream>
using namespace std;
#define n 3
void preXOR(int arr[][n], int prefix_xor[][n]) {
   for (int i = 0; i < n; i++)
      for (int j = 0; j < n; j++) {
         if (j == 0)
            prefix_xor[i][j] = arr[i][j];
         else
         prefix_xor[i][j]
            = (prefix_xor[i][j - 1] ^ arr[i][j]);
      }
   for (int i = 0; i < n; i++)
      for (int j = 1; j < n; j++)
         prefix_xor[j][i]
            = (prefix_xor[j - 1][i] ^ prefix_xor[j][i]);
}
int XORSubMatrix(int prefix_xor[][n], int querry[2]) {
   int xor_1 = 0, xor_2 = 0, xor_3 = 0;
   if (querry[0] != 0)
      xor_1 = prefix_xor[querry[0] - 1][querry[3]];
   if (querry[1] != 0)
      xor_2 = prefix_xor[querry[2]][querry[1] - 1];
   if (querry[2] != 0 and querry[1] != 0)
      xor_3 = prefix_xor[querry[0] - 1][querry[1] - 1];
   return ((prefix_xor[querry[2]][querry[3]] ^ xor_1) ^ (xor_2 ^ xor_3));
}
int main() {
   int arr[][n] = { { 1, 2, 3 },
      { 4, 5, 6 },
      { 7, 8, 9 } };
   int prefix_xor[n][n];
   preXOR(arr, prefix_xor);
   int querry1[] = {0,0, 2,2} ;
   int querry2[] = {1,2, 2,2} ;
   cout<<"The XOR of submatrix created by querries :\n";
   cout<<"Querry 1 : "<<XORSubMatrix(prefix_xor, querry1)<<endl;
   cout<<"Querry 2 : "<<XORSubMatrix(prefix_xor, querry2)<<endl;
   return 0;
}

ผลลัพธ์

Querry 1 : 1
Querry 2 : 15