[C/C++ 백준 2036번] 수열의 점수 (Silver 1)

https://www.acmicpc.net/problem/2036

2036번: 수열의 점수

도서관 문제와 매우 비슷한 문제. 양수에서의 1과, 음수에서 남는 수들을 0과 조합해야하는 것을 조심하면 쉽게 풀 수 있다.

 

#include <cstdio>
#include <algorithm>
using namespace std;
bool comp(long long int a, long long int b){
	return a>b;
}
int main(void){
	long long int n, ans=0, num, plus[100001]={}, minus[100001]={}, pcnt=0, mcnt=0;
	scanf("%lld", &n);
	for(int i=0; i<n; i++){
		scanf("%lld", &num);
		if(num>0)
			plus[pcnt++]=num;
		else
			minus[mcnt++]=num*(-1);
	}
	sort(plus, plus+pcnt, comp);
	sort(minus, minus+mcnt, comp);
	if(mcnt!=0){
		for(int i=0; i<mcnt; i+=2){
			if(i==mcnt-1)
				ans -= minus[i];
			else
				ans += minus[i]*minus[i+1];
			}
	}
	if(pcnt!=0){
		for(int i=0; i<pcnt; i+=2){
			if(i==pcnt-1)
				ans += plus[i];
			else if(plus[i]<=1 ||plus[i+1]<=1)
				ans += (plus[i]+plus[i+1]);
			else
				ans += plus[i]*plus[i+1];
		}	
	}
	printf("%lld",ans);
}