글 읽기 - 3649 로봇 프로젝트 질문입니다.

orange4glace 8년 전 0

기초적인 이분탐색 문제인것 같은데 .

어디가 틀린걸까요 ㅠㅠ

#include <cstdio>
#include <queue>
#include <algorithm>
#include <stack>
#include <iostream>
using namespace std;
 
#define MIN(a,b) (((a)<(b))?(a):(b))
#define MAX(a,b) (((a)>(b))?(a):(b))
 
int N, M, K, Q, X, Y, DIR;
int arr[1000005];
 
int solve(int start, int end, int t) {
    if (start > end) return -1;
    int mid = (start + end) / 2;
    if (t > arr[mid]) {
        return solve(mid + 1, end, t);
    }
    else if (t < arr[mid]) {
        return solve(start, mid - 1, t);
    }
    return mid;
}
 
int main() {
 
    Q = 1;
    for (int q = 0; q < Q; q++) {
        scanf("%d", &M);
        if (M > 2) {
            printf("danger");
            return 0;
        }
        M *= 10000000;
        scanf("%d", &N);
        for (int i = 0; i < N; i++) {
            scanf("%d", &arr[i]);
        }
        sort(arr, arr + N);
        int bb = -1;
        int aa = 0;
        for (int i = 0; i < N; i++) {
            aa = i;
            bb = solve(i + 1, N - 1, M - arr[i]);
            if (bb != -1) break;
        }
        if (bb == -1) printf("danger");
        else printf("yes %d %d", arr[aa], arr[bb]);
    }
 
    cin >> N;
 
    return 0;
}