[POI 2011]Sticks(亂搞)

題目鏈接

http://main.edu.pl/en/archive/oi/18/pat

題目大意

給你k種顏色，每種顏色有一些木棍，給出每種木棍的長度和顏色，問是否能選出三根顏色互不相同的木棍，構成一個三角形

思路

我們可以把所有的木棍丟到一起，按照長度升序排序，然後枚舉前i 根木棍，維護其中的長度最大的三根顏色互不相同的木棍，然後判斷這三根木棍是否能構成三角形。

實際上就是在枚舉長度最長的那根木棍，顯然另外兩個比它短的木棍的長度之和應該大於它，因此這兩根木棍的長度應該儘量長，所以上述做法是正確的

代碼

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <algorithm>

#define MAXN 1100000

using namespace std;

pair<int,int>pr[MAXN];
int tot=0,K;

int main()
{
    scanf("%d",&K);
    for(int i=1;i<=K;i++)
    {
        int cnt;
        scanf("%d",&cnt);
        for(int j=1;j<=cnt;j++)
        {
            tot++;
            scanf("%d",&pr[tot].first);
            pr[tot].second=i;
        }
    }
    sort(pr+1,pr+tot+1);
    pair<int,int>first=make_pair(-1,-1),second=make_pair(-1,-1),third=make_pair(-1,-1);
    for(int i=1;i<=tot;i++)
    {
        if(pr[i].second==first.second)
            first=pr[i];
        else if(pr[i].second==second.second)
        {
            second=first;
            first=pr[i];
        }
        else
        {
            third=second;
            second=first;
            first=pr[i];
        }
        if(first.first==-1||second.first==-1||third.first==-1) continue;
        if(second.first+third.first>first.first)
        {
            printf("%d %d %d %d %d %d\n",first.second,first.first,second.second,second.first,third.second,third.first);
            return 0;
        }
    }
    printf("NIE\n");
    return 0;
}