2012百度之星初赛第一场题A度度熊就是要第一个出场

第一题：度度熊就是要第一个出场
贴图不方便, 完整题目见这里

我的思路:

1,从下向上,找第一名的路径,把横线段做标记1,没有横线段的地方添加横线段出来,标记2

2,从上向下,找第K名(即度度熊的位置)的路径,遇到有标记的即表示本身就是第一(标记1),或加线段后可以是第一(标记2)

代码如下, 测试用例可以通过, 又做了几个简单的测试

// 2012 百度之星 初赛 第一场 题A:度度熊就是要第一个出场
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

typedef struct line
{
int elem1;
int elem2;
int elem3;
int path;
}line;

//升序排序,是"<",降序排列是">"号
bool cmp(line a,line b){
return a.elem3<b.elem3;
}

int main()
{
int loop;
cin>>loop;
while (loop--)
{
int i,j,tag=0,N,M,K,max3=0,noline=0;//员工数,线段树,度度熊的位置
cin>>N>>M>>K;
int tmpK = K;

vector<line> array;	//M条线段
line lin;
for(i=0;i<M;i++)
{
cin>>lin.elem1>>lin.elem2>>lin.elem3;
lin.path = 0;
array.push_back(lin);
}

//按线段高低降序排列按线段高低降序排列
sort(array.begin(),array.end(),cmp);

//从下向上,找第一名的路线,线段path标记为1
int firstpath = 1;
int preelem3;
preelem3 = 0;
for (i=0;i<M;i++)
{
max3 = max3>array[i].elem3?max3:array[i].elem3;
if (array[i].elem1==firstpath||array[i].elem2==firstpath)
{
noline++;
//与上一个高度差
if(array[i].elem3>preelem3+1){
for (j=array[i].elem3-1;j>preelem3;j--)
{
lin.elem1 = firstpath;
lin.elem2 = firstpath+1;
lin.elem3 = j;
lin.path  = 2;
array.push_back(lin);	//往右添加横线

lin.elem1 = firstpath-1;
lin.elem2 = firstpath;
array.push_back(lin);	//往左添加横线
}
}
//主轴移动到旁边一个轴上
firstpath = array[i].elem1==firstpath?array[i].elem2:array[i].elem1;
array[i].path = 1;
preelem3 = array[i].elem3;
}
}
if (noline==0)
{
for (j=0;j<max3+1;j++)
{
lin.elem1 = 1;
lin.elem2 = 2;
lin.elem3 = j;
lin.path  = 2;
array.push_back(lin);	//往右添加横线
}
}
//在主轴最大值上向左右加横线
for (i=preelem3+1;i<=max3+1;i++)
{
lin.elem1 = firstpath;
lin.elem2 = firstpath+1;
lin.elem3 = i;
lin.path  = 2;
array.push_back(lin);	//往右添加横线

lin.elem1 = firstpath-1;
lin.elem2 = firstpath;
array.push_back(lin);	//往左添加横线
}

//稳定排序,使固有线段在后加线段之前
stable_sort(array.begin(),array.end(),cmp);

//从上向下,找度度熊的路线,看能否遇到有标记的线段
preelem3 = 1000000;
for (i=array.size()-1;i>=0;i--)
{
if ( (array[i].elem2!=preelem3)&&(array[i].elem1==tmpK||array[i].elem2==tmpK) )
{
if (array[i].path>0)//线段标有1,表示度度熊本身就是第一; 是2表示可以通过加一个线段变为第一
{
tag++;
}
tmpK=array[i].elem1==tmpK?array[i].elem2:array[i].elem1;
preelem3 = array[i].elem3;
}
}
if (tag>0) //tag>0表示可以通过加线段改为第一,tag的值表示可添加线段的位置的个数
{
cout<<"Yes"<<endl;
}else{
cout<<"No"<<endl;
}
}
system("pause");
return 0;
}