제출 #1238891

#제출 시각아이디문제언어결과실행 시간메모리
1238891adriines06모자이크 (IOI24_mosaic)C++20
22 / 100
1010 ms2162688 KiB
#include "mosaic.h"
#include<bits/stdc++.h>

#include <vector>
using namespace std;
std::vector<long long> mosaic(std::vector<int> X, std::vector<int> Y,
                              std::vector<int> T, std::vector<int> B,
                              std::vector<int> L, std::vector<int> R) {
  int Q = (int)T.size();
  vector<long long> C;
  int N=(int)X.size();
  vector<vector<int>>grid(N,vector<int>(N));
  int sum=0;
  for(int i=0;i<N;i++){
    grid[i][0]=Y[i];
    grid[0][i]=X[i];
    sum+=Y[i]+X[i];
  }
  if(sum==0){
    vector<long long>v;
    for(int i=0;i<Q;i++){
      if(R[i]==0 || B[i]==0){
        v.push_back(0);
      }
      else{
        long long a=B[i]-T[i]+1;
        if(T[i]==0) a--;
        long long lo=R[i]-L[i]+1;
        if(L[i]==0) lo--;
        if((a*lo)%2==1){
          int u=T[i],d=B[i],l=L[i],r=R[i];
          if(u==0) u=1;
          if(l==0) l=1;
          int x=(l+r)/2;
          int y=(u+d)/2;
          //cout<<x<<" "<<y<<"\n";
          //cout<<a<<" "<<l<<"\n";
          if((x+y)%2==0){
            v.push_back((a*lo+1)/2);
          }
          else v.push_back((a*lo)/2);
        }
        else{
          v.push_back((a*lo)/2);
        }
        
      }
    }
    return v;
  }

  for(int i=1;i<N;i++){
    for(int j=1;j<N;j++){
      if(grid[i-1][j]==0 && grid[i][j-1]==0){
        grid[i][j]=1;
      }
      else{
        grid[i][j]=0;
      }
    }
  }

  vector<vector<int>>pref(N+1,vector<int>(N+1,0));

  for(int i=1;i<=N;i++){
    for(int j=1;j<=N;j++){
      pref[i][j]=pref[i][j-1]+grid[i-1][j-1];
      pref[i][j]+=pref[i-1][j]-pref[i-1][j-1];
    }
  }


  for(int i=0;i<Q;i++){
    int u=T[i]+1,d=B[i]+1,l=L[i]+1,r=R[i]+1;
    int t=pref[d][r],c1=pref[d][l-1],c2=pref[u-1][r],e=pref[u-1][l-1];
    long long ans=t-c1-c2+e;
    C.push_back(ans);
  }


  return C;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...