湘潭oj_1252 Defense Tower

贪心,先破坏攻击力大的塔再破坏攻击力小的,这样受到攻击力小的塔的伤害值,受伤害值最小。

//按最大攻击力排序再依次破坏掉,同时计算受到的伤害值。

#include
#include
#include
using namespace std;
struct f{
 int num;    //塔的编号
 int value;  //塔的攻击值
}c[100001];
bool cmp(f x,f y){
  return x.value>y.value;
}
int a[100001],b[100001],p[100001],book[100001];//a,b数组记录连接路的情况,book数组记录塔是否被破坏掉,p记录每个塔的攻击值。
int main(){
int n,i,j;
while(~scanf("%d",&n)){
    memset(book,0,sizeof(book));
  for(i=1;i<=n;i++){
    scanf("%d",&c[i].value);
    c[i].num=i;
    p[i]=c[i].value;
  }
  sort(c+1,c+n+1,cmp);
   for(i=1;i<=n-1;i++)
   {
       scanf("%d%d",&a[i],&b[i]);
   }
    int sum=0;
  for(i=1;i<=n;i++)   //n个塔遍历
  {
      for(j=1;j<=n-1;j++) //n-1个边遍历统计拆掉第c[i].num个塔受到的伤害
        if(!book[j]){     //判断这条路是否被拆掉了
          if(a[j]==c[i].num) {sum+=p[b[j]];book[j]=1;}
          else if(b[j]==c[i].num) {sum+=p[a[j]];book[j]=1;}
      }
  }
  printf("%d\n",sum);
}
return 0;
}//超时
#include  
int a[100001];
int main(){
    int i,n;
    while(~scanf("%d",&n)){
      for(i=1;i<=n;i++)
       scanf("%d",&a[i]);
       int x,y,sum=0;
     for(i=1;i<=n-1;i++)   //其实直接遍历逐一拆路
     {
        scanf("%d%d",&x,&y);
        if(a[x]>=a[y]) sum+=a[y]; //每次拆路受到攻击力小的塔的伤害(因为攻击力大的塔先被破坏)
        else sum+=a[x];
     }                    
     printf("%d\n",sum);
    }
return 0;
}//ac


你可能感兴趣的:(湘潭oj_1252 Defense Tower)