Submission #19676965

Source Code Expand

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
#include <bits/stdc++.h>
#define ALL(x) std::begin(x), std::end(x)
using namespace std;
// @title 行列
#include <cassert>
#include <iostream>
#include <vector>
using namespace std;
template< class T >
struct Matrix {
  size_t H, W;
  vector< vector< T > > A;
  Matrix() {}
  Matrix(size_t n, size_t m) : H(n), W(m), A(H, vector< T >(W, 0)) {}
  Matrix(size_t n) : H(n), W(n), A(H, vector< T >(W, 0)) {};
  Matrix(const vector<vector<T>>& a) : A(a) {
      H = a.size();
      assert(H > 0);
      W = a[0].size();
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX

#include <bits/stdc++.h>
#define ALL(x) std::begin(x), std::end(x)
using namespace std;

// @title 行列
#include <cassert>
#include <iostream>
#include <vector>
using namespace std;

template< class T >
struct Matrix {
  size_t H, W;
  vector< vector< T > > A;
  Matrix() {}
  Matrix(size_t n, size_t m) : H(n), W(m), A(H, vector< T >(W, 0)) {}
  Matrix(size_t n) : H(n), W(n), A(H, vector< T >(W, 0)) {};
  Matrix(const vector<vector<T>>& a) : A(a) {
      H = a.size();
      assert(H > 0);
      W = a[0].size();
  }
  size_t height() const { return H; }
  size_t width() const { return W; }
  inline const vector< T > &operator[](int k) const { return (A.at(k)); }
  inline vector< T > &operator[](int k) { return (A.at(k)); }
  static Matrix I(size_t n) {
    Matrix mat(n);
    for(int i = 0; i < n; i++) mat[i][i] = 1;
    return (mat);
  }

  Matrix &operator+=(const Matrix &B) {
    assert(H == B.H && W == B.W);
    for(int i = 0; i < H; i++)
      for(int j = 0; j < W; j++)
        (*this)[i][j] += B[i][j];
    return (*this);
  }

  Matrix &operator-=(const Matrix &B) {
    assert(H == B.H && W == B.W);
    for(int i = 0; i < H; i++)
      for(int j = 0; j < W; j++)
        (*this)[i][j] -= B[i][j];
    return (*this);
  }

  Matrix &operator*=(const Matrix &B) {
    assert(W == B.H);
    vector< vector< T > > C(H, vector< T >(B.W, 0));
    for(int i = 0; i < H; i++)
      for(int j = 0; j < B.W; j++)
        for(int k = 0; k < W; k++)
          C[i][j] += ((*this)[i][k] * B[k][j]);
    A.swap(C);
    return (*this);
  }

  Matrix &operator^=(long long k) {
    Matrix B = Matrix::I(H);
    while(k > 0) {
      if(k & 1) B *= *this;
      *this *= *this;
      k >>= 1LL;
    }
    A.swap(B.A);
    return (*this);
  }

  vector<T> operator*(const vector<T> &v) {
    assert(W == v.size());
    vector<T> ret(H, 0);
    for(int i = 0; i < H; i++)
      for(int j = 0; j < W; j++)
        ret[i] += ((*this)[i][j] * v[j]);
    return ret;
  }

  Matrix operator+(const Matrix &B) const { return (Matrix(*this) += B); }
  Matrix operator-(const Matrix &B) const { return (Matrix(*this) -= B); }
  Matrix operator*(const Matrix &B) const { return (Matrix(*this) *= B); }
  Matrix operator^(const long long k) const { return (Matrix(*this) ^= k); }

  friend ostream &operator<<(ostream &os, Matrix &p) {
    for(int i = 0; i < p.H; i++) {
      for(int j = 0; j < p.W; j++) {
        os << p[i][j] << (j + 1 == p.W ? "\n" : " ");
      }
    }
    return (os);
  }
};

using ll = long long;
using matrix = Matrix<ll>;
matrix l1({
    {0, 1, 0},
    {-1, 0, 0},
    {0, 0, 1}
});
matrix l2({
    {0, -1, 0},
    {1, 0, 0},
    {0, 0, 1}
});

struct Pos { int x, y; };
int main() {
    int n; cin >> n;
    vector<Pos> pos;
    for (int i = 0; i < n; ++i) {
        int x, y; cin >> x >> y;
        pos.push_back((Pos){x, y});
    }
    vector<matrix> mat;
    mat.push_back(matrix({
        {1, 0, 0},
        {0, 1, 0},
        {0, 0, 1}
    }));
    int m; cin >> m;
    for (int i = 0; i < m; ++i) {
        int op; cin >> op;
        int p;
        if (op >= 3) cin >> p;
        matrix &b = mat.back();
        if (op == 1) {
            mat.push_back(l1*b);
        }
        else if (op == 2) {
            mat.push_back(l2*b);
        }
        else if (op == 3) {
            matrix l3({
                {-1, 0, 2*p},
                {0, 1, 0},
                {0, 0, 1}
            });
            mat.push_back(l3*b);
        }
        else {
            matrix l4({
                {1, 0, 0},
                {0, -1, 2*p},
                {0, 0, 1}
            });
            mat.push_back(l4*b);
        }
    }
    int q; cin >> q;
    for (int i = 0; i < q; ++i) {
        int a, b; cin >> a >> b;
        Pos p = pos[b-1];
        vector<ll> l = {p.x, p.y, 1};
        vector<ll> ans = mat[a]*l;
        cout << ans[0] << " " << ans[1] << endl;
    }

    return 0;
}

Submission Info

Compile Error

./Main.cpp: In instantiation of ‘std::vector<_Tp> Matrix<T>::operator*(const std::vector<_Tp>&) [with T = long long int]’:
./Main.cpp:156:33:   required from here
./Main.cpp:74:22: warning: comparison of integer expressions of different signedness: ‘int’ and ‘size_t’ {aka ‘long unsigned int’} [-Wsign-compare]
   74 |     for(int i = 0; i < H; i++)
      |                    ~~^~~
./Main.cpp:75:24: warning: comparison of integer expressions of different signedness: ‘int’ and ‘size_t’ {aka ‘long unsigned int’} [-Wsign-compare]
   75 |       for(int j = 0; j < W; j++)
      |                      ~~^~~
./Main.cpp: In instantiation of ‘Matrix<T>& Matrix<T>::operator*=(const Matrix<T>&) [with T = long long int]’:
./Main.cpp:82:67:   required from ‘Matrix<T> Matrix<T>::operator*(const Matrix<T>&) const [with T = long long int]’
./Main.cpp:129:30:   required from here
./Main.cpp:52:22: warning: comparison of integer expressions of different signedness: ‘int’ and ‘size_t’ {aka ‘long unsigned int’} [-Wsign-compare]
   52 |     for(int i = 0; i < H; i++)
      |                    ~~^~~
./Main.cpp:53:24: warning: comparison of integer expressions of different signedness: ‘int’ and ‘const size_t’ {aka ‘const long unsigned int’} [-Wsign-compare]
   53 |       for(int j = 0; j < B.W; j++)
      |                      ~~^~~~~
./Main.cpp:54:26: warning: comparison of integer expressions of different signedness: ‘int’ and ‘size_t’ {aka ‘long unsigned int’} [-Wsign-compare]
   54 |         for(int k = 0; k < W; k++)
      |                        ~~^~~

Judge Result