#include<bits/stdc++.h>
//#include "rainbow.h"
#define MAXN 300007
using namespace std;
int n,m,sx,sy,a,b,c,d,x,y,comps;
string w;
map< pair<int,int> , int > mp;
void init(int R, int C, int sr, int sc, int M, char *S) {
n=R; m=C;
sx=sr; sy=sc;
for(int i=0;i<M;i++){
w.push_back(S[i]);
}
}
bool in(int x,int y){
return x>=a and x<=c and y>=b and y<=d;
}
void check(int x,int y){
if(in(x-1,y) and mp[{x-1,y}]==0)mp[{x-1,y}]=1;
if(in(x+1,y) and mp[{x+1,y}]==0)mp[{x+1,y}]=1;
if(in(x,y-1) and mp[{x,y-1}]==0)mp[{x,y-1}]=1;
if(in(x,y+1) and mp[{x,y+1}]==0)mp[{x,y+1}]=1;
if(in(x-1,y-1) and mp[{x-1,y-1}]==0)mp[{x-1,y-1}]=1;
if(in(x+1,y+1) and mp[{x+1,y+1}]==0)mp[{x+1,y+1}]=1;
if(in(x+1,y-1) and mp[{x+1,y-1}]==0)mp[{x+1,y-1}]=1;
if(in(x-1,y+1) and mp[{x-1,y+1}]==0)mp[{x-1,y+1}]=1;
}
void dfs(int x,int y){
mp[{x,y}]=2;
if(mp[{x-1,y}]==1)dfs(x-1,y);
if(mp[{x+1,y}]==1)dfs(x+1,y);
if(mp[{x,y-1}]==1)dfs(x,y-1);
if(mp[{x,y+1}]==1)dfs(x,y+1);
}
void calc(int x,int y){
if(mp[{x-1,y}]==1){dfs(x-1,y);comps++;}
if(mp[{x+1,y}]==1){dfs(x+1,y);comps++;}
if(mp[{x,y-1}]==1){dfs(x,y-1);comps++;}
if(mp[{x,y+1}]==1){dfs(x,y+1);comps++;}
}
int colour(int ar, int ac, int br, int bc){
a=ar; b=ac;
c=br; d=bc;
comps=0; mp.clear();
for(int i=a;i<=c;i++){
mp[{i,b}]=mp[{i,d}]=1;
}
for(int i=b;i<=d;i++){
mp[{a,i}]=mp[{c,i}]=1;
}
x=sx; y=sy; mp[{x,y}]=-1;
for(int i=0;i<w.size();i++){
if(w[i]=='N')x--;
if(w[i]=='E')y++;
if(w[i]=='S')x++;
if(w[i]=='W')y--;
mp[{x,y}]=-1;
}
x=sx; y=sy; check(x,y);
for(int i=0;i<w.size();i++){
if(w[i]=='N')x--;
if(w[i]=='E')y++;
if(w[i]=='S')x++;
if(w[i]=='W')y--;
check(x,y);
}
x=sx; y=sy; calc(x,y);
for(int i=0;i<w.size();i++){
if(w[i]=='N')x--;
if(w[i]=='E')y++;
if(w[i]=='S')x++;
if(w[i]=='W')y--;
calc(x,y);
}
if(comps==0 and mp[{a,b}]==1)comps++;
return comps;
}
int main(){
init(6, 4,3, 3, 9,"NWESSWEWS");
cout<<colour(2,3, 2, 3)<<"\n";
cout<<colour(3,2, 4, 4)<<"\n";
cout<<colour(5,3, 6, 4)<<"\n";
cout<<colour(1,2, 5, 3)<<"\n";
cout<<colour(4,6,4,6)<<"\n";
return 0;
}
Compilation message
rainbow.cpp: In function 'int colour(int, int, int, int)':
rainbow.cpp:68:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
68 | for(int i=0;i<w.size();i++){
| ~^~~~~~~~~
rainbow.cpp:77:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
77 | for(int i=0;i<w.size();i++){
| ~^~~~~~~~~
rainbow.cpp:86:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
86 | for(int i=0;i<w.size();i++){
| ~^~~~~~~~~
rainbow.cpp: In function 'int main()':
rainbow.cpp:101:23: warning: ISO C++ forbids converting a string constant to 'char*' [-Wwrite-strings]
101 | init(6, 4,3, 3, 9,"NWESSWEWS");
| ^~~~~~~~~~~
/usr/bin/ld: /tmp/ccm729SJ.o: in function `main':
grader.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/ccKnkRxM.o:rainbow.cpp:(.text.startup+0x0): first defined here
collect2: error: ld returned 1 exit status
