#include <iostream>
#include <vector>
#include<map>
using namespace std;
map<int,int>mX,mY;
long long max_weights(int N, int M, vector<int> X, vector<int> Y, vector<int> W)
{
for(int i=0;i<X.size();i++)
{
mX[X[i]]=1;
mY[Y[i]]=1;
}
if(N==1)
return 0;
bool check=true;
int maxX=0;
for(int i:X)
maxX=max(maxX,i);
for(int i:X)
if(i%2)
{
check=false;
break;
}
long long ans=0;
if(check)
{
for(int i=0;i<W.size();i++)
ans+=W[i];
return ans;
}
if(maxX==0)
{
for(int i=0;i<W.size();i++)
ans+=W[i];
return ans;
}
if(maxX==1)
{
long long ansX0=0,ansX1=0,ans=0;
for(int i=0;i<X.size();i++)
if(X[i]==1)
ansX1+=W[i];
ans=ansX1;
for(int i=0;i<N;i++)
{
if(mY[i]==1 && mX[1]==1)
ansX1-=W[i];
if(mY[i]==1 && mX[0]==1)
ansX0+=W[i];
ans=max(ans,ansX0+ansX1);
}
return ans;
}
return 0;
}
int main() {
int N, M;
cin>>N>>M;
vector<int> X(M), Y(M), W(M);
for (int i = 0; i < M; ++i)
cin>>X[i]>>Y[i]>>W[i];
long long result = max_weights(N, M, X, Y, W);
cout<<result;
return 0;
}
Compilation message
fish.cpp: In function 'long long int max_weights(int, int, std::vector<int>, std::vector<int>, std::vector<int>)':
fish.cpp:8:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
8 | for(int i=0;i<X.size();i++)
| ~^~~~~~~~~
fish.cpp:28:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
28 | for(int i=0;i<W.size();i++)
| ~^~~~~~~~~
fish.cpp:34:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
34 | for(int i=0;i<W.size();i++)
| ~^~~~~~~~~
fish.cpp:41:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
41 | for(int i=0;i<X.size();i++)
| ~^~~~~~~~~
/usr/bin/ld: /tmp/ccPSOlXY.o: in function `main':
grader.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/ccywIX9X.o:fish.cpp:(.text.startup+0x0): first defined here
collect2: error: ld returned 1 exit status
