pat::b::b1050 命名空间参考

1050 螺旋矩阵 更多...

函数
int	main (istream &cin, ostream &cout)
	TEST (b1050, case1)

详细描述

1050 螺旋矩阵

函数说明

main()

int pat::b::b1050::main	(	istream &	cin,
		ostream &	cout
	)

在文件 pat.cpp 第 1635 行定义.

                                                  {
                int N;
                cin >> N;
                int m = N;
                int n = 1;
                for(int i = 1; i * i <= N; ++i) {
                    if(N % i == 0) {
                        n = i;
                        m = N / i;
                    }
                }
                vector matrix(m, vector(n, 0));
                vector<int> vec(N);
                for(int i = 0; i < N; i++) {
                    cin >> vec[i];
                }
                sort(vec.rbegin(), vec.rend());
                int direction = 0;//0 右  1 下  2 左  3 上
                int x         = 0;
                int y         = 0;
                int current   = 0;
                matrix[0][0]  = vec[current];
                --N;
                while(N != 0) {
                    int next_x = x;
                    int next_y = y;
                    switch(direction) {
                        case 0:
                            next_y += 1;
                            break;
                        case 1:
                            next_x += 1;
                            break;
                        case 2:
                            next_y -= 1;
                            break;
                        case 3:
                            next_x -= 1;
                            break;
                    }
                    if(next_x < 0 || next_x >= m || next_y < 0 || next_y >= n || matrix[next_x][next_y] != 0) {
                        direction += 1;
                        direction %= 4;
                    } else {
                        matrix[next_x][next_y] = vec[++current];
                        x                      = next_x;
                        y                      = next_y;
                        N--;
                    }
                }
                for(int i = 0; i < m; i++) {
                    for(int j = 0; j < n; j++) {
                        cout << matrix[i][j];
                        if(j != n - 1) {
                            cout << ' ';
                        }
                    }
                    if(i != m - 1) {
                        cout << endl;
                    }
                }
                return 0;
            }

引用了 acwing::acwing2019::N , 以及 pat::a::a7_2::vec.

被这些函数引用 TEST().

TEST()

pat::b::b1050::TEST	(	b1050	,
		case1
	)

在文件 pat_test.cpp 第 765 行定义.

                               {
                istringstream in("12\n37 76 20 98 76 42 53 95 60 81 58 93\n");
                auto out = ostringstream();
                main(in, out);
                const auto ans = out.str();
                ASSERT_EQ("98 95 93\n42 37 81\n53 20 76\n58 60 76", out.str());
            }

引用了 main().