腾讯2016秋招笔试编程题

有一个由很多木棒构成的集合，每个木棒有对应的长度，请问能否用集合中的这些木棒以某个顺序首尾相连构成一个面积大于 0 的简单多边形且所有木棒都要用上，简单多边形即不会自交的多边形。 初始集合是空的，有两种操作，要么给集合添加一个长度为 L 的木棒，要么删去集合中已经有的某个木棒。每次操作结束后你都需要告知是否能用集合中的这些木棒构成一个简单多边形。

输入描述:
每组测试用例仅包含一组数据，每组数据第一行为一个正整数 n 表示操作的数量(1 ≤ n ≤ 50000) ， 接下来有n行，每行第一个整数为操作类型 i (i ∈ {1,2})，第二个整数为一个长度 L(1 ≤ L ≤ 1,000,000,000)。如果 i=1 代表在集合内插入一个长度为 L 的木棒，如果 i=2 代表删去在集合内的一根长度为 L 的木棒。输入数据保证删除时集合中必定存在长度为 L 的木棒，且任意操作后集合都是非空的。

输出描述:
对于每一次操作结束有一次输出，如果集合内的木棒可以构成简单多边形，输出 “Yes” ，否则输出 “No”。

输入例子:
5
1 1
1 1
1 1
2 1
1 2

输出例子:
No
No
Yes
No
No

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
using namespace std;

bool judge_polygon(vector<int> lengths)
{
    if(lengths.size() < 3)
    {
        return false;
    }
    else
    {
         sort(lengths.begin(), lengths.end());
         int total_sum = 0;
         int part_sum;
         for (int n : lengths)
         {
             total_sum += n;
         }
         int max_length = lengths[lengths.size()-2];
         part_sum = total_sum - max_length;
         if (part_sum > max_length)
         {
             return true;
         }
         else
         {
             return false;
         }
    }
}
int main()
{
    vector<int> lengths;
    vector<int>::iterator it;
    int flag;
    int count;
    int L;
    cin >> count;
    for(int i=0; i< count; ++i)
    {
        cin >> flag >> L;
        if( flag == 1)
        {
            lengths.push_back(L);
            if (judge_polygon(lengths))
            {
                cout << "Yes" << endl;
            }
            else
            {
                cout << "No" << endl;
            }
        }
        else
        {
            for(it = lengths.begin(); it!= lengths.end(); )
            {
                if(*it == L)
                {
                    it = lengths.erase(it);
                }
                else
                {
                    ++it;
                }
            }
            if (judge_polygon(lengths))
            {
                cout << "Yes" << endl;
            }
            else
            {
                cout << "No" << endl;
            }
        }
    }
    return 0;
}