Countries

Countries

题目链接：http://hihocoder.com/problemset/problem/1391

预处理+双指针

首先将A->B，B->A的导弹全部转化为B->A的导弹（因为不需要计算B承受的伤害，所以对于A->B的导弹，只需记录被B的防护罩返回来的导弹即可）.

然后对于每个导弹计算将其反弹回B，A所需要的最小的防护罩区间[l,r]（这个操作在O（1）的时间完成，画张图就很好推了）

于是问题就转化为了：

数轴上有一些固定区间，每个固定区间都有一个权值，现在有一个长度为Ta的防护罩区间，问用这个防护罩区间完整包含的固定区间的权值和最大为多少？

对于这个问题，可以将固定区间的左右两端点标号，存到一个数组（struct｛int position，index，value；｝），然后用一个vis数组标记第index个固定区间在防护罩区间内的情况，遍历一遍防护罩区间的L即可.

代码如下：

 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 #define N 20005
 #define R (L+Ta)
 using namespace std;
 typedef long long LL;
 LL Ta,Tb,X,n,m,k,sum,pt;
 bool vis[];
 struct point{
     LL times,index,v;
 }p[];
 struct node{
     LL fire,fly,dam;
     LL l,r;
 }launch[N];
 bool cmp(point a,point b){
     return a.times<b.times;
 }
 void init(){
     sum=;
     memset(vis,,sizeof(vis));
     memset(p,,sizeof(p));
     memset(launch,,sizeof(launch));
     scanf("%I64d",&X);
     scanf("%I64d%I64d",&n,&m);
     k=pt=;
     for(LL i=;i<n;++i){
         LL a,b,c;
         scanf("%I64d%I64d%I64d",&a,&b,&c);
         if(X<=a+b&&a+b<=X+Tb){
             launch[k].fire=a+b;
             launch[k].fly=b;
             launch[k].dam=c;
             sum+=c;
             k++;
         }
     }
     for(LL i=;i<m;++i){
         scanf("%I64d%I64d%I64d",&launch[k].fire,&launch[k].fly,&launch[k].dam);
         sum+=launch[k].dam;
         k++;
     }
     for(LL i=;i<k;++i){
         LL fire=launch[i].fire;
         LL fly=launch[i].fly;
         launch[i].l=fire+fly;
         LL t=*fly;
         if(fire+t<X||fire+t>X+Tb){
             launch[i].r=launch[i].l;
         }else{
             LL len=X+Tb-fire;
             LL times=len/t;
             launch[i].r=launch[i].l+times*t;
         }
     }
     for(LL i=;i<k;++i){
         if(launch[i].r-launch[i].l<=Ta){
             p[pt].times=launch[i].l;
             p[pt].index=pt/;
             p[pt++].v=launch[i].dam;
             p[pt].times=launch[i].r;
             p[pt].index=pt/;
             p[pt++].v=launch[i].dam;
         }
     }
     sort(p,p+pt,cmp);
 }
 int main(void){
     while(~scanf("%I64d%I64d",&Ta,&Tb)){
         init();
         LL Max=;
         LL temp=;
         LL l=,r=,L=p[].times;
         /*************防护罩L=p[0].times时的情况*************/
         for(;r<pt&&p[r].times<=R;++r){
             LL idx=p[r].index;
             if(!vis[idx])vis[idx]=;//[L,R]内有且只有一个idx，即l在[L,R]内
             else if(vis[idx]){//如果[L,R]内已经有一个idx，说明当前整个区间[l,r]都在[L,R]内
                 vis[idx]=;
                 temp+=p[r].v;
             }
         }

         /*****************将防护罩的L不断右移****************/
         if(Max<temp)Max=temp;
         while(r<pt){
             LL idx=p[l].index;//防护罩L右移，删去l
             if(!vis[idx]){//若vis[idx]为零，说明[L,R]中原来有两个idx，即现在[l,r]并不都在[L,R]内
                 vis[idx]=;
                 temp-=p[l].v;
             }else if(vis[idx])vis[idx]=;//原来[l,r]就不都在[L,R]内
             l++;
             while(l<pt&&p[l].times==p[l-].times){//如果更改后的l与l-1相同，则需要继续处理
                 idx=p[l].index;
                 if(!vis[idx]){
                     vis[idx]=;
                     temp-=p[l].v;
                 }else if(vis[idx])vis[idx]=;
                 l++;
             }
             if(l==pt)break;
             L=p[l].times;
             for(;r<pt&&p[r].times<=R;++r){
                 idx=p[r].index;
                 if(!vis[idx])vis[idx]=;
                 else if(vis[idx]){
                     vis[idx]=;
                     temp+=p[r].v;
                 }
             }
             if(Max<temp)Max=temp;
         }
         printf("%I64d\n",sum-Max);
     }
 }

然而出了点问题，昨天晚上debug到三点多，一直是WA，会不会写搓了？第二天起来看了好久没看出来，会不会是算法错了？问了游少，他也表示母鸡。

回到寝室，心血来潮把I64d改成lld交了一发：PE.？？？.jpg

全部改成了cin，cout：AC.浪费了我一天

代码如下：

 #include<iostream>
 #include<algorithm>
 #include<cstring>
 #define N 20005
 #define R (L+Ta)
 using namespace std;
 typedef long long LL;
 LL Ta,Tb,X,n,m,k,sum,pt;
 bool vis[];
 struct point{
     LL times,index,v;
 }p[];
 struct node{
     LL fire,fly,dam;
     LL l,r;
 }launch[N];
 bool cmp(point a,point b){
     return a.times<b.times;
 }
 void init(){
     sum=;
     memset(vis,,sizeof(vis));
     memset(p,,sizeof(p));
     memset(launch,,sizeof(launch));
     cin>>X;
     cin>>n>>m;
     k=pt=;
     for(LL i=;i<n;++i){
         LL a,b,c;
         cin>>a>>b>>c;
         if(X<=a+b&&a+b<=X+Tb){
             launch[k].fire=a+b;
             launch[k].fly=b;
             launch[k].dam=c;
             sum+=c;
             k++;
         }
     }
     for(LL i=;i<m;++i){
         cin>>launch[k].fire>>launch[k].fly>>launch[k].dam;
         sum+=launch[k].dam;
         k++;
     }
     for(LL i=;i<k;++i){
         LL fire=launch[i].fire;
         LL fly=launch[i].fly;
         launch[i].l=fire+fly;
         LL t=*fly;
         if(fire+t<X||fire+t>X+Tb){
             launch[i].r=launch[i].l;
         }else{
             LL len=X+Tb-fire;
             LL times=len/t;
             launch[i].r=launch[i].l+times*t;
         }
     }
     for(LL i=;i<k;++i){
         if(launch[i].r-launch[i].l<=Ta){
             p[pt].times=launch[i].l;
             p[pt].index=pt/;
             p[pt++].v=launch[i].dam;
             p[pt].times=launch[i].r;
             p[pt].index=pt/;
             p[pt++].v=launch[i].dam;
         }
     }
     sort(p,p+pt,cmp);
 }
 int main(void){
     while(cin>>Ta>>Tb){
         init();
         LL Max=-;
         LL temp=;
         LL l=,r=,L=p[].times;
         for(;r<pt&&p[r].times<=R;++r){
             LL idx=p[r].index;
             if(!vis[idx])vis[idx]=;
             else if(vis[idx]){
                 vis[idx]=;
                 temp+=p[r].v;
             }
         }
         if(Max<temp)Max=temp;
         while(r<pt){
             LL idx=p[l].index;
             if(!vis[idx]){
                 vis[idx]=;
                 temp-=p[l].v;
             }else if(vis[idx])vis[idx]=;
             l++;
             while(l<pt&&p[l].times==p[l-].times){
                 idx=p[l].index;
                 if(!vis[idx]){
                     vis[idx]=;
                     temp-=p[l].v;
                 }else if(vis[idx])vis[idx]=;
                 l++;
             }
             if(l==pt)break;
             L=p[l].times;
             for(;r<pt&&p[r].times<=R;++r){
                 idx=p[r].index;
                 if(!vis[idx])vis[idx]=;
                 else if(vis[idx]){
                     vis[idx]=;
                     temp+=p[r].v;
                 }
             }
             if(Max<temp)Max=temp;
         }
         cout<<sum-Max<<endl;
     }
 }