bzoj1835 [ZJOI2010]基站选址

  1 #include<algorithm>
  2 #include<iostream>
  3 #include<cstring>
  4 #include<cstdio>
  5 #include<string>
  6 #include<vector>
  7 #include<bitset>
  8 #include<cmath>
  9 #include<queue>
 10 #include<ctime>
 11 #include<map>
 12 #include<set>
 13 #define N 20039
 14 #define ll long long
 15 #define inf 0x3f3f3f3f
 16 using namespace std;
 17 struct seg
 18 {
 19     int l,r,w,lazy;
 20 }tree[N<<3];
 21 int x[N],L[N],R[N],co[N],rw[N],sum[N],dp[N],ans,stp;
 22 bool use[N];
 23 vector<int> inl[N],inr[N];
 24 void pull_up(int k)
 25 {
 26     tree[k].w=min(tree[k<<1].w,tree[k<<1|1].w);
 27 }
 28 void push_down(int k)
 29 {
 30     if(tree[k].lazy)
 31     {
 32         tree[k<<1].w+=tree[k].lazy,tree[k<<1|1].w+=tree[k].lazy;
 33         tree[k<<1].lazy+=tree[k].lazy,tree[k<<1|1].lazy+=tree[k].lazy;
 34         tree[k].lazy=0;
 35     }
 36 }
 37 void build(int k,int l,int r)
 38 {
 39     tree[k].l=l,tree[k].r=r,tree[k].lazy=0;
 40     if(l==r)
 41     {
 42         tree[k].w=dp[l];
 43         return; 
 44     }
 45     int mid=l+r>>1;
 46     build(k<<1,l,mid);
 47     build(k<<1|1,mid+1,r);
 48     pull_up(k);
 49 }
 50 void update(int k,int l,int r,int v)
 51 {
 52     if(l>r)
 53     {
 54         return;
 55     }
 56     push_down(k);
 57     if(tree[k].l>=l && tree[k].r<=r)
 58     {
 59         tree[k].w+=v;
 60         tree[k].lazy+=v;
 61         return;
 62     }
 63     int mid=tree[k].l+tree[k].r>>1;
 64     if(mid>=r)
 65     {
 66         update(k<<1,l,r,v);
 67     }
 68     else if(mid<l)
 69     {
 70         update(k<<1|1,l,r,v);
 71     }
 72     else
 73     {
 74         update(k<<1,l,r,v),update(k<<1|1,l,r,v);
 75     }
 76     pull_up(k);
 77 }
 78 int query(int k,int l,int r)
 79 {
 80     if(l>r)
 81     {
 82         return 0;
 83     }
 84     push_down(k);
 85     if(tree[k].l>=l && tree[k].r<=r)
 86     {
 87         return tree[k].w;
 88     }
 89     int mid=tree[k].l+tree[k].r>>1;
 90     if(mid>=r)
 91     {
 92         return query(k<<1,l,r);
 93     }
 94     else if(mid<l)
 95     {
 96         return query(k<<1|1,l,r);
 97     }
 98     else
 99     {
100         return min(query(k<<1,l,r),query(k<<1|1,l,r));
101     }
102 }
103 int main()
104 {
105     int n,k,a,b,c;
106     scanf("%d%d",&n,&k);
107     for(a=2;a<=n;a++)
108     {
109         scanf("%d",&x[a]);
110     }
111     for(a=1;a<=n;a++)
112     {
113         scanf("%d",&co[a]);
114     }
115     for(a=1;a<=n;a++)
116     {
117         scanf("%d",&b);
118         L[a]=lower_bound(x+1,x+n+1,x[a]-b)-x;
119         R[a]=lower_bound(x+1,x+n+1,x[a]+b)-x;
120         if(x[R[a]]>x[a]+b)
121         {
122             R[a]--;
123         }
124         inl[L[a]].push_back(a);
125         inr[R[a]].push_back(a);
126     }
127     for(a=1;a<=n;a++)
128     {
129         scanf("%d",&rw[a]);
130         ans+=rw[a];
131     }
132     for(a=1,stp=ans;a<=n;a++)
133     {
134         for(b=0;b<inl[a].size();b++)
135         {
136             if(!use[inl[a][b]])
137             {
138                 stp-=rw[inl[a][b]];
139             }
140             use[inl[a][b]]=1;
141         }
142         sum[a]=stp;
143     }
144     for(a=1;a<=k;a++)
145     {
146         if(a==1)
147         {
148             for(b=1,stp=0;b<=n;b++)
149             {
150                 dp[b]=stp+co[b];
151                 for(c=0;c<inr[b].size();c++)
152                 {
153                     stp+=rw[inr[b][c]];
154                 }
155                 ans=min(ans,dp[b]+sum[b]);
156             }
157             continue;
158         }
159         build(1,1,n);
160         for(b=1;b<=n;b++)
161         {
162             dp[b]=query(1,1,b-1)+co[b];
163             for(c=0;c<inr[b].size();c++)
164             {
165                 update(1,1,L[inr[b][c]]-1,rw[inr[b][c]]);
166             }
167             ans=min(ans,dp[b]+sum[b]);
168         }
169     }
170     printf("%d",ans);
171     return 0;
172 }