結果

問題 No.1141 田グリッド
コンテスト
ユーザー shiomusubi496
提出日時 2020-07-31 23:17:43
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 3,283 bytes
コンパイル時間 1,656 ms
コンパイル使用メモリ 185,572 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-07-06 21:26:42
合計ジャッジ時間 5,367 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 29 WA * 2
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function 'int main()':
main.cpp:55:29: warning: 'm' may be used uninitialized [-Wmaybe-uninitialized]
   55 |                 if(i!=n && j!=m)ans=ans*A[i][j]%mod;
      |                            ~^~~
main.cpp:41:27: note: 'm' was declared here
   41 |         int mx1=0,mx2=0,n,m;
      |                           ^
main.cpp:61:13: warning: 'n' may be used uninitialized [-Wmaybe-uninitialized]
   61 |             if(r!=n || c!=m)puts("0");
      |             ^~
main.cpp:41:25: note: 'n' was declared here
   41 |         int mx1=0,mx2=0,n,m;
      |                         ^

ソースコード

diff # 

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int mod=1000000007;
int p(int A,int B){
    if(B==0)return 1;
    if(B%2)return p(A,B-1)*A%mod;
    int C=p(A,B/2);
    return C*C%mod;
}
signed main(){
    int H,W,ans=1;
    cin>>H>>W;
    vector<vector<int>> A(H,vector<int>(W));
    vector<int> B(H,1),C(W,1);
    for(int i=0;i<H;i++){
        for(int j=0;j<W;j++){
            cin>>A[i][j];
            ans=ans*A[i][j]%mod;
            B[i]=B[i]*A[i][j]%mod;
            C[j]=C[j]*A[i][j]%mod;
        }
    }
    vector<int> D(0),E(0);
    for(int i=0;i<H;i++){
        if(B[i]==0)D.push_back(i);
    }
    for(int i=0;i<W;i++){
        if(C[i]==0)E.push_back(i);
    }
    int Q;
    cin>>Q;
    if(D.size()>1 && E.size()>1){
        int cnt=0;
        map<int,int> F,G;
        for(int i=0;i<H;i++){
            for(int j=0;j<W;j++){
                if(A[i][j]==0)F[i]++,G[j]++,cnt++;
            }
        }
        int mx1=0,mx2=0,n,m;
        for(pair<int,int> p:F){
            if(mx1<p.second)mx1=p.second,n=p.first;
        }
        for(pair<int,int> p:G){
            if(mx2<p.second)mx2=p.second,m=p.first;
        }
        if(mx1+mx2-1!=cnt){
            while(Q--)puts("0");
            return 0;
        }
        int ans=1;
        for(int i=0;i<H;i++){
            for(int j=0;j<W;j++){
                if(i!=n && j!=m)ans=ans*A[i][j]%mod;
            }
        }
        while(Q--){
            int r,c;
            cin>>r>>c;r--,c--;
            if(r!=n || c!=m)puts("0");
            else cout<<ans<<endl;
        }
        return 0;
    }
    if(D.size()>1){
        int ans=1;
        vector<int> F(H,1);
        for(int i=0;i<H;i++){
            for(int j=0;j<W;j++){
                if(E[0]==j)continue;
                ans=ans*A[i][j]%mod;
                F[i]=F[i]*A[i][j]%mod;
            }
        }
        while(Q--){
            int r,c;
            cin>>r>>c;r--,c--;
            if(c!=E[0])puts("0");
            else cout<<ans*p(F[r],mod-2)%mod<<endl;
        }
        return 0;
    }
    if(E.size()>1){
        int ans=1;
        vector<int> F(W,1);
        for(int i=0;i<H;i++){
            if(i==D[0])continue;
            for(int j=0;j<W;j++){
                ans=ans*A[i][j]%mod;
                F[j]=F[j]*A[i][j]%mod;
            }
        }
        while(Q--){
            int r,c;
            cin>>r>>c;r--,c--;
            if(r!=D[0])puts("0");
            else cout<<ans*p(F[c],mod-2)%mod<<endl;
        }
        return 0;
    }
    if(D.size()==1 && E.size()==1){
        int ans=1;
        vector<int> F(H,1),G(W,1);
        for(int i=0;i<H;i++){
            for(int j=0;j<W;j++){
                if(D[0]==i && E[0]==j)continue;
                ans=ans*A[i][j]%mod;
                F[i]=F[i]*A[i][j]%mod;
                G[j]=G[j]*A[i][j]%mod;
            }
        }
        while(Q--){
            int r,c;
            cin>>r>>c;r--,c--;
            if(c!=E[0] && r!=D[0])puts("0");
            else if(c==E[0] && r==D[0])cout<<ans*p(F[r]*G[c]%mod,mod-2)%mod<<endl;
            else cout<<ans*A[r][c]%mod*p(F[r]*G[c]%mod,mod-2)%mod<<endl;
        }
        return 0;
    }
    while(Q--){
        int r,c;
        cin>>r>>c;r--,c--;
        cout<<ans*A[r][c]%mod*p(B[r]*C[c]%mod,mod-2)%mod<<endl;
    }
}
0