1. 程式人生 > >[程式設計題] 木棒拼圖(今日頭條招聘)

[程式設計題] 木棒拼圖(今日頭條招聘)

有一個由很多木棒構成的集合,每個木棒有對應的長度,請問能否用集合中的這些木棒以某個順序首尾相連構成一個面積大於 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 <stdio.h>
using namespace std;
long long a[50010];
int main()
{
    int n;
    while(cin>>n)
    {
        long long op,l,count=0;
        long long sum=0,max=0;
        int p=0;
        for(int i=0; i<n; i++)
        {
            scanf("%I64d%I64d",&op,&l);
            if(op==1)
            {
                a[i]=l;
                sum+=l;
                if(l>max)max=l;
                count+=1;
            }
            if(op==2)
            {
                sum-=l;
                for(int j=p; j<i; j++)
                {
                    if(a[j]==l)
                    {
                        a[j]=0;
                        swap(a[j],a[p]);
                        p++;
                        break;
                    }
                }
                if(l==max)
                {
                    max=0;
                    for(int j=p; j<i; j++)
                    {
                        if(a[j]>max)max=a[j];
                    }
                }

                count-=1;

            }
           //cout<<max<<" sum: "<<sum<<" ount: "<<count<<endl;
            if(max<sum-max&&count>2)printf("Yes\n");
            else printf("No\n");



        }

    }
    return 0;
}

還有一種用map
#include <iostream>
#include <stdio.h>
#include <map>
using namespace std;
long long a[50010];
int main()
{
    int n;
    while(cin>>n)
    {
        long long op,l;
        long long sum=0,count=0;
        map<long long,int >m;
        for(int i=0; i<n; i++)
        {
            cin>>op>>l;
            if(op==1)
            {
                m[-l]++;
                sum+=l;
                count++;
            }
            if(op==2)
            {
                m[-l]--;
                if(m[-l]==0)
                    m.erase(-l);
                count--;
                sum-=l;
            }
            map<long long ,int >::iterator ir;
            ir=m.begin();
            long long k=ir->first;
            k=-k;
            if(k<sum-k&&count>2)
                cout<<"Yes"<<endl;
            else cout<<"No"<<endl;
        }


    }
    return 0;
}


動態連結串列

#include <iostream>
#include <stdio.h>
using namespace std;
struct Node{
long long x;
int pre;
int next;
}node[50010];

int main()
{
    int n;

    while(cin>>n)
    {
        long long op,l,count=0;
        long long sum=0,max=0;
        int p=1;
        node[0].next=1;
        node[0].x=0;
        for(int i=0; i<n; i++)
        {

            scanf("%I64d %I64d",&op,&l);
            if(op==1)
            {
                node[p].pre=p-1;
                node[p].next=p+1;
                node[p++].x=l;
                sum+=l;
                if(l>max)max=l;
                count+=1;
            }
            if(op==2)
            {
                sum-=l;
                for(int j=0; j<p;)
                {
                    if(node[j].x==l)
                    {
                        node[j].x=0;
                        node[node[j].pre].next=node[j].next;
                        node[node[j].next].pre=node[j].pre;
                        break;
                    }
                    j=node[j].next;
                   //cout<<"J: "<<j<<endl;
                }
                if(l==max)
                {
                    max=0;
                    for(int j=0; j<p;)
                    {
                        if(node[j].x>max)max=node[j].x;
                        j=node[j].next;
                    }
                }

                count-=1;

            }
           //cout<<max<<" sum: "<<sum<<" ount: "<<count<<endl;
            if(max<sum-max&&count>2)printf("Yes\n");
            else printf("No\n");



        }

    }
    return 0;
}