LeetCode Weekly Contest 162 2問目 - Reconstruct a 2-Row Binary Matrix [1253]

Source

LeetCode Weekly Contest 162
問題文

問題概要

省略

解法

省略

cLay（version 20191108-1）のコード

C++に変換後のコードはこちら

#define main dummy_main
{}
#undef main

class Solution {
public:
  vector<vector<int>> reconstructMatrix(int upper, int lower, vector<int>& colsum) {
    int N = colsum.size();
    vector<int> tmp(N);
    vector<vector<int>> res{tmp, tmp}, empty;

    rep(i,N) if(colsum[i]==2){
      res[0][i] = res[1][i] = 1;
      upper--;
      lower--;
    }

    rep(i,N) if(colsum[i]==1){
      if(upper > 0) upper--, res[0][i] = 1, continue;
      lower--, res[1][i] = 1;
    }

    if(upper || lower) return empty;
    return res;
  }
};

Current time: 2024年04月26日07時40分56秒
Last modified: 2019年11月10日13時46分28秒 (by laycrs)
Tags: Competitive_Programming_Incomplete LeetCode
トップページに戻る

Logged in as: unknown user (not login)