3318 Matrix Multiplication 随机算法

题意：给出n阶方阵A，B和C，判断C是否为A和B的乘积。

方阵规模最大为500×500，若直接将A和B相乘，算法复杂度为O(n^3)，肯定要超时。考虑采用随机算法，随机选取50000个项进行测试。

用马甲测试了一下，一次AC：3216K 1594MS，大号提交WA了3次……

代码如下：

1. #include  <iostream>
2. #include  <cstdlib>
3. #include  <ctime>
5. using  namespace  std;
7. int   A[500][500], B[500][500];
8. long  C[500][500];
10. int  main()
11. {
12.     srand( (unsigned)time(NULL) ); 
13.     
14.     int   k, j, n;
15.     int   r, c;
16.     int   sum, flag;
17.     
18.     cin >> n;
19.     
20.     for (k = 0; k < n; ++k)
21.     {
22.         for (j = 0; j < n; ++j)
23.         {
24.             scanf("%d", &A[k][j]);
25.         }
26.     } 
27.            
28.     for (k = 0; k < n; ++k)
29.     {
30.         for (j = 0; j < n; ++j)
31.         {
32.             scanf("%d", &B[k][j]);
33.         }
34.     } 
35.     
36.     for (k = 0; k < n; ++k)
37.     {
38.         for (j = 0; j < n; ++j)
39.         {
40.             scanf("%d", &C[k][j]);
41.         }
42.     } 
43.     
44.     flag = 0;
45.     
46.     for (j = 0; j < 50000; ++j)
47.     {
48.         r = rand() % n;
49.         c = rand() % n;
50.         
51.         sum = 0;
52.         
53.         for (k = 0; k < n; ++k)
54.         {
55.             sum += A[r][k] * B[k][c];
56.         }
57.         
58.         if (sum != C[r][c])
59.         {
60.             flag = 1;
61.             break;
62.         } 
63.     }    
64.     
65.     if (flag == 0)
66.     {
67.         cout << "YES" << endl;
68.     }
69.     else
70.     {
71.         cout << "NO" << endl;
72.     }        
74.     return  0;
75. }