Gym - 101097I Sticks（思维）

https://vjudge.net/contest/160830#problem/I

题意：给你一堆不同颜色的不同长度的棍，问你是否这些棍是否能够组成三角形且所有三角形的每个边的颜色都不相同。

思路：在添加三角形的边时，如果颜色相同就取最大的边，如果找的了三条边，就去比较第一以及第三条边和是否能够组成三角形，如果能直接输出，否则将第一条边等于第二条边，接着进行下一次判断。如果找不到输出NIL

#include <iostream>
#include <bits/stdc++.h>
using namespace std;
struct node
{
    int data;
    int num;
} a[1000005];
int cmp(const node &p,const node &q)
{
    return p.data<q.data;
}
int main()
{
    freopen("sticks.in","r",stdin);
    freopen("sticks.out","w",stdout);
    int n,m,x;
    scanf("%d",&n);
    {
        int cnt=0;
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&m);
            for(int j=0; j<m; j++)
            {
                cnt++;
                scanf("%d",&a[cnt].data);
                a[cnt].num=i;
            }
        }
        sort(a+1,a+cnt+1,cmp);
        for(int i=1,j=1,k=1; i<cnt-1;)
        {
            while(i<cnt&&a[i].num==a[i+1].num) i++;
            j=i+1;
            while(j<cnt&&a[j].num==a[j+1].num) j++;
            k=j+1;
            if(k<=cnt&&a[k].num!=a[i].num)
            {
                if(a[i].data>a[k].data-a[j].data)
                {
                    printf("%d %d %d %d %d %d\n",a[i].num,a[i].data,a[j].num,a[j].data,a[k].num,a[k].data);
                    return 0;
                }
            }
            i=j;
        }
        printf("NIE\n");
    }
    return 0;
}