CodeForce J. Jeopardy of Dropped Balls

粗略统计每一列有多少个除了2以外的数，当这一列全是2的时候直接输出

k个球的输入存入数组会增大复杂度

直接暴力写也可以过 

#include<iostream>
using namespace std;
int n,m,k;
int a[1005][1005];
int q;
int d[1005]; 
bool flag;
int x,y;
int main(){
	scanf("%d %d %d",&n,&m,&k);
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			scanf("%d",&a[i][j]);
		}
	}
	for(int i=1;i<=m;i++){
		for(int j=1;j<=n;j++){
			if(a[j][i]!=2) d[i]++;
		}
	}
	for(int i=1;i<=k;i++){
		scanf("%d",&q);
		x=1,y=q;
		if(d[y]==0) printf("%d ",y);
		else{
			flag=1;
			while(x<=n){
			if(d[y]==0){
				printf("%d ",y);flag=0;
				break;
			}
			if(a[x][y]==1){
				a[x][y]=2;
				if(d[y]>0) d[y]--;y++;
			}
			else if(a[x][y]==2){
				x++;
			}
			else if(a[x][y]==3){
				a[x][y]=2;if(d[y]>0) d[y]--;y--;
			}
		}
		if(flag) printf("%d ",y);;	
		} 
	}	
	return 0;
}