글 읽기 - 간단한 실수같은데... 맞은경우와 틀린경우의 다른점이 뭘까요?

s7d8f9 6년 전 0

맞은경우와 틀린경우의 다른점이 뭘까요...... 눈에 잘 안들어오네요 ㅠㅠ

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <math.h>
#include <queue>
#include <set>
#include <list>
#include <utility>
#include <functional>
#define MAX 4005
#define INF 987654321
#define MOD 1000000
#pragma warning(disable:4996)
using namespace std;
typedef long long ll;
typedef pair<int, int> pi;

ll ans = 0;
int n, d1, d2,map[MAX][4];
vector<int> a, b;

int main()
{
	scanf("%d", &n);
	for (int i = 0; i < n; i++)
		for (int k = 0; k < 4; k++)
			scanf("%d", &map[i][k]);
	for (int i = 0; i < n; i++)
		for (int k = 0; k < n; k++)
		{
			a.push_back(map[i][0] + map[k][1]);
			b.push_back(map[i][2] + map[k][3]);
		}

	sort(a.begin(), a.end());
	sort(b.begin(), b.end());
	
	for (int i = 0; i < a.size(); i++)
	{
		/*틀린 경우*/
		pair<vector<int>::iterator, vector<int>::iterator> iter1 = equal_range(a.begin(), a.end(), a[i]);
		pair<vector<int>::iterator, vector<int>::iterator> iter2 = equal_range(b.begin(), b.end(), -1*a[i]);
		d1 = distance(iter1.first, iter1.second);
		d2 = distance(iter2.first, iter2.second);
		ans += d1*d2;
		i += (d1 - 1);
		/*맞은 경우*/
		pair<vector<int>::iterator, vector<int>::iterator> iter2 = equal_range(b.begin(), b.end(), -1 * a[i]); 
		d2 = distance(iter2.first, iter2.second);
		ans += d2;
	}
	printf("%lld", ans);

	return 0;
}

djm03178 6년 전 0

단 한 번의 d1*d2 연산에서 오버플로가 발생할 수 있습니다. ans += (long long)d1 * d2; 로 하면 맞습니다.

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

간단한 실수같은데... 맞은경우와 틀린경우의 다른점이 뭘까요?

s7d8f9 6년 전 0

djm03178 6년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말