leetcode - 1706. 球会落何处

最新推荐文章于 2024-11-04 12:20:00 发布

zmm_mohua

最新推荐文章于 2024-11-04 12:20:00 发布

阅读量222

点赞数

分类专栏：刷题文章标签： leetcode 算法

本文链接：https://blog.csdn.net/qq_42250642/article/details/123115823

版权

刷题专栏收录该内容

312 篇文章 0 订阅

订阅专栏

leetcode - 1706. 球会落何处

题目

在这里插入图片描述

代码

#include <iostream>
#include <vector>
using namespace std;

vector<int> findBall(vector<vector<int> >& grid) {
	int n = grid[0].size();
	int m = grid.size();
	vector<int> res(n);
	for(int j = 0; j < n; j++){
		int col = j;  // 记录刚开始小球位于第几列
		for(int i = 0; i < m; i++){
			int dir = grid[i][col];
			col += dir;
			if(col < 0 || col == n || grid[i][col] != dir){
				col = -1;
				break;
			}
		}
		res[j] = col;
	}
	return res;
}

int main(){
	vector<int> res;
	int m, n;
	cin>>m>>n;
	vector<vector<int> > grid(m,vector<int>(n,0));
	for(int i = 0; i < m; i++){
		for(int  j = 0; j < n; j++){
			cin>>grid[i][j];
		}
	}
	res = findBall(grid);
	for(int i = 0; i < res.size(); i++){
		cout<<res[i]<<" ";
	}
	return 0;
}