在行列都排好序的矩阵中找指定的数 - 数学计算

// C++
#include<iostream>
#include<vector>

using namespace std;

int main()
{
    int  n, m, k;
    cin>>n>>m>>k;
    vector<vector<int > > data(n, vector<int>(m));  //读入数组预先分配空间
    for(int i=0; i<n; ++i)
        for(int j=0; j<m; ++j)
            cin>>data[i][j];
    for(int i=0, j=m-1; i<n&&j>=0;)
    {
        if(data[i][j] > k) --j;
        else if(data[i][j] < k) ++i;
        else{
            cout<<"Yes"<<endl;
            return 0;
        }
    }
    cout<<"No"<<endl;
//    cout<<data.size()<<endl;
//    cout<<data[0].size()<<endl;
//    for(int i=0; i<n; ++i)
//    {
//        for(int j=0; j<m; ++j)
//        {
//            cout<<data[i][j]<<" ";
//        }
//        cout<<endl;
//    }
//    cout<<k<<endl;
    return 0;
}

# Python
def handle():
    n, m, k = map(int, input().split())
    data = []
    data_set = set()
    for _ in range(n):
        for one in map(int, input().split()):
            data_set.add(one)
    if k in data_set:
        print('Yes')
    else: print('No')


handle()