题意:
给出至多 50 种颜色的共 1e6 根木棍,问是否能有 3 支颜色不同的木棍可以拼成三角形。
思路:
首先注意可能爆 int 的细节。
队里三个人都在做这道题,结果三个人都思维江化僵化,拖了很久。
结果三个人写了三种代码,都是 AC 的,事后证明三个人的思路原理是一样的。
所以以下贴三种代码,各是各的思路。
首先都是按长度排序
A 和 B 的思路近似于构造一个存有三种不同颜色的木棍的框。
每次加入一根新的木棍时,若新棍的颜色与框内重复,则保留同色最大的。
若不重复,则删除框中长度最小的,加入新棍。
每次加入判断是否符合题意即可。
C 的思路和 A B 原理相同。
枚举每个木棍,找到与其长度最接近(不妨规定为大于等于)并且颜色不同的木棍。
这时候就得到了两根木棍,可以推算出第三根木棍的合法长度区间。
从 50 - 2 种颜色中二分寻找是否有符合条件的木棍即可。
代码:
A 同学的:O(n*k)
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;
int k,n,cur,num[55],s[55];
struct node
{
int color,l