Submission #74773284

---

Source Code Expand

#include <bits/stdc++.h>
using ull = unsigned long long int;
using ll = long long int;
using lll = __int128;
using namespace std;
const int mod = 1e9 + 7;
int d[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
inline string sread() {
    string s;
    char ch = getchar();
    // 跳过空白字符
    while (ch == ' ' || ch == '\n' || ch == '\r') ch = getchar();
    // 读取非空白字符
    while (ch != ' ' && ch != '\n' && ch != '\r' && ch != EOF) {
        s += ch;
        ch = getchar();
    }
    return s;
}
ll read(){
    ll k = 0,f = 1;
    char c = getchar();
    while(c < '0' || c > '9')
    {
        if(c == '-')f = -1;
        c = getchar();
    }
    while(c >= '0' && c <= '9')k = k * 10 + c - '0',c = getchar();
    return k * f; // 别忘记标记的负数要乘进去
}
// 调用时用 n = in();
void print(ll x){
    if(x < 0)putchar('-'),x = -x;
    if(x < 10)putchar(x + '0');
    else print(x / 10),putchar(x % 10 + '0');
}
// 直接调用 out(n) 就行了
ll ksm(ll base, ll e){
    ll res = 1;
    while(e){
        if(e & 1){
            res = res * base % mod;
        }
        base = base * base % mod;
        e >>= 1;
    }
    return res;
}//ksm板子
vector<vector<ll>> matrix_mul (vector<vector<ll>>& a, vector<vector<ll>>& b) {
    int l1 = a.size(), l2 = b[0].size();
    vector<vector<ll>>res(l1, vector<ll>(l2, 0));
    for(int i = 0; i < l1; i++){
        for(int k = 0; k < a[0].size(); k++){
            if(a[i][k]){
                for(int j = 0; j < l2; j++){
                    res[i][j] = (res[i][j] + a[i][k] * b[k][j]);
                }
            }
        }
    }
    return res;
}// 矩阵a * 矩阵b(矩阵乘法)
void Solve(){
    int n = read(), m = read();
    auto cmp = [&](const auto& a, const auto & b) -> bool{
        return a[0] < b[0];
    };
    priority_queue<array<int, 2>, vector<array<int, 2>>, decltype(cmp)>pq(cmp);
    for(int i = 1; i <= n; i++){
        int p = read();
        pq.push({p, i});
    }
    vector<vector<int>>cnt(n + 1);
    for(int i = 0; i < m; i++){
        int u = read(), v = read();
        cnt[u].push_back(v);
        cnt[v].push_back(u);
    }
    for(int i = 1; i <= n; i++){
        sort(cnt[i].begin(), cnt[i].end());
    }
    vector<int>vis(n + 1, 0);
    while(!pq.empty()){
        auto can = pq.top()[1];
        pq.pop();
        if(vis[can])continue;
        vis[can] = 1; // 当前这个数字要标记
        for(int x : cnt[can]){
            if(!vis[x]){
                vis[x] = 1;
                if(x == 1){
                    cout << can << '\n';
                    return;
                }
                else if(can == 1){
                    cout << x << '\n';
                    return;
                }
            }
        }
    }
}
signed main(){
    int T = 1;
    while(T--){
        Solve();
    }
    return 0;
}

Submission Info

Compile Error

./Main.cpp: In function 'std::vector<std::vector<long long int> > matrix_mul(std::vector<std::vector<long long int> >&, std::vector<std::vector<long long int> >&)':
./Main.cpp:53:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   53 |         for(int k = 0; k < a[0].size(); k++){
      |                        ~~^~~~~~~~~~~~~

Judge Result