T1:判断一个矩阵是否是X矩阵
这个题目的思路有很多种,看怎么去判断。但是最终的目的都是为了判断对角线上的元素是否都不为零,其他元素都为零。我的解法是:先算出整个矩阵的元素之和,再计算两条对角线的元素之和,在计算对角线元素之和时要判断元素是否为零,出现元素为零的情况就直接退出,返回false。然后比较两个和是否相等,若相等,则符合题意。
class Solution {
public:
bool checkXMatrix(vector<vector<int>>& grid) {
int sum=0;
int cnt=0;
int n=grid.size();
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
sum+=grid[i][j];
if(i==j||i+j==n-1){
if(grid[i][j]==0)return false;
else{
cnt+=grid[i][j];
}
}
}
}
if(sum==cnt)return true;
return false;
}
};
T2:统计放置房子的方式数
这个题目我在比赛的想法是因为两侧互不影响,只需要计算一边的放置情况就好了。但是一直没有想到用动态规划的方法,加上我看数据还比较大,就放弃了。现在来看,这个题真的是比较简单的动态规划问题了。
可能比较难考虑的就是DP方程。思路类似于斐波那契数列。两种情况:第i处放置房子,则i-1处不能放置,有DP[i-2]种情况;若第i处不放置房子,则i-1处可以放置。由DP[i-1]开始递推。
class Solution {
public:
int countHousePlacements(int n) {
vector<long long> dp(n + 1 , 0);
dp[0] = 1 ;
dp[1] = 2 ;
int mod = 1e9+7;
for(int i = 2 ; i <= n ; i ++){
dp[i] = (dp[i-1] + dp[i-2]) % mod ;
}
return (dp[n]*dp[n]) % mod ;
}
};