Problem - E - Codeforces
思路:最短路变种,每个点都有两种状态。分别是从不骑马到达的状态,和骑马到达的状态。要注意的是,如果当前点是有马的,而前一个点是不骑马过来的,那么更新也直接更新为当前骑马的状态,可以直接舍去当前点不骑马的状态。因为当前点不骑马的状态已经没有意义了,既然当前点可以骑马,那么后面肯定全部骑马。如果一个点可以由骑马的到达,也可以由不骑马的到达,那么两个状态都要保留,不一定骑马的就是更优的。因此vis要有两种状态,pq要存三个值,这样的话每个点最多遍历两次,不会超时。记录每个点由不骑马的到达和骑马的到达需要的最短时间后。枚举相遇点,取两种状态的最小值,再取最大值就是当前点最小的相遇时间。
int n,m,h;
vector<pair<int,int>> vct[200005];
bool mark[200005];
priority_queue<pair<int,pair<int,int>>> pq;
int dis1[200005][2],dis2[200005][2]; 0为没有骑马,1为骑马
map<pair<int,int>,int> vis; <from,状态>
void dijkstra(int s,int typ){
vis.clear();
if(typ==1) for(int i=1;i<=n;i++) dis1[i][0]=dis1[i][1]=LONG_LONG_MAX/2;
if(typ==2) for(int i=1;i<=n;i++) dis2[i][0]=dis2[i][1]=LONG_LONG_MAX/2;
typ==1?dis1[s][0]=0:dis2[s][0]=0;
if(mark[s]&&typ==1) dis1[s][1]=0;
if(mark[s]&&typ==2) dis2[s][1]=0;
pq.push({0,{s,mark[s]}}); <权重,<from,状态>>
while(pq.size()){
int from=pq.top().second.first,sta=pq.top().second.second;
pq.pop();
if(vis[{from,sta}]) continue;
vis[{from,sta}]=1;
for(auto v:vct[from]){
int to=v.first,w=v.second;
if(typ==1){
if(sta){
if(dis1[to][1]>dis1[from][1]+w/2){
dis1[to][1]=dis1[from][1]+w/2;
pq.push({-dis1[to][1],{to,1}});
}
}
else if(mark[to]){
if(dis1[to][1]>dis1[from][0]+w){
dis1[to][1]=dis1[from][0]+w;
pq.push({-dis1[to][1],{to,1}});
}
}
else{
if(dis1[to][0]>dis1[from][0]+w){
dis1[to][0]=dis1[from][0]+w;
pq.push({-dis1[to][0],{to,0}});
}
}
}
if(typ==2){
if(sta){
if(dis2[to][1]>dis2[from][1]+w/2){
dis2[to][1]=dis2[from][1]+w/2;
pq.push({-dis2[to][1],{to,1}});
}
}
else if(mark[to]){
if(dis2[to][1]>dis2[from][0]+w){
dis2[to][1]=dis2[from][0]+w;
pq.push({-dis2[to][1],{to,1}});
}
}
else{
if(dis2[to][0]>dis2[from][0]+w){
dis2[to][0]=dis2[from][0]+w;
pq.push({-dis2[to][0],{to,0}});
}
}
}
}
}
}
void solve(){ E
cin>>n>>m>>h;
for(int i=1;i<=n;i++) vct[i].clear(),mark[i]=false;
for(int i=1;i<=h;i++){
int x; cin>>x;
mark[x]=true;
}
for(int i=1;i<=m;i++){
int u,v,w; cin>>u>>v>>w;
vct[u].emplace_back(v,w);
vct[v].emplace_back(u,w);
}
dijkstra(1,1);
dijkstra(n,2);
if(dis1[n][0]==LONG_LONG_MAX/2&&dis1[n][1]==LONG_LONG_MAX/2){
cout<<"-1"<<endl;
return;
}
else{
int ans=LONG_LONG_MAX;
for(int i=1;i<=n;i++) ans=min(ans,max(min(dis1[i][0],dis1[i][1]),min(dis2[i][0],dis2[i][1])));
cout<<ans<<endl;
}
}
Problem - H - Codeforces
思路:莫队的模板题,直接从洛谷上面复制之前写的代码,改一下就行了.
int n,m,B;
int arr[250004],p[250004];
typedef struct Q{
int l,r,idx;
}Q;
Q q[250004];
//bool cmp(Q a,Q b){ 普通分块排序
// if(a.l/B!=b.l/B) return a.l<b.l;
// return a.r<b.r;
//}
bool cmp(Q a,Q b){ 玄学奇偶性排序
if (p[a.l] ^ p[b.l]) return a.l/B < b.l/B ;
else{
if (p[a.l] & 1) return a.r < b.r;
else return a.r > b.r;
}
}
int l=0,r=0,cnt[1000006],ans[250004];
P2709 小B的询问
https://www.luogu.com.cn/problem/P2709
void solve(){ H 莫队轻松过
cin>>n>>m;
B=sqrt(n);
for(int i=1;i<=n;i++) cin>>arr[i],cnt[arr[i]]=0;
for(int i=1;i<=m;i++) cin>>q[i].l>>q[i].r,q[i].idx=i,p[q[i].l]=q[i].l/B,p[q[i].r]=q[i].r/B;
sort(q+1,q+m+1,cmp);
l=q[1].l+1,r=q[1].l;
int cnt1=0;
for(int i=1;i<=m;i++){
while(l>q[i].l) {
--l;
cnt[arr[l]]^=1;
if(cnt[arr[l]]==1) cnt1++;
else cnt1--;
}
while(r<q[i].r) {
++r;
cnt[arr[r]]^=1;
if(cnt[arr[r]]==1) cnt1++;
else cnt1--;
}
while(l<q[i].l) {
cnt[arr[l]]^=1;
if(cnt[arr[l]]==1) cnt1++;
else cnt1--;
l++;
}
while(r>q[i].r) {
cnt[arr[r]]^=1;
if(cnt[arr[r]]==1) cnt1++;
else cnt1--;
r--;
}
ans[q[i].idx]=(cnt1==0);
}
for(int i=1;i<=m;i++){
if(ans[i]==1) cout<<"YES\n";
else cout<<"NO\n";
}
}
Problem - F - Codeforces
思路:入门的树形dp. 每个点有选和不选两种状态.
int n,c;
vector<int> vct[200005];
int gold[200005];
int dp[200005][2];
void dfs(int from,int fa){
dp[from][0]=0;
dp[from][1]=gold[from];
for(auto to:vct[from]){
if(to==fa) continue;
dfs(to,from);
如果这个当前这个不选择,加上孩子节点中较大者
dp[from][0]+=max(dp[to][0],dp[to][1]);
如果当前这个选择,如果上一个不选择,那么不用考虑c
如果上一个这个都选择,那么两者都会减c
dp[from][1]+=max(dp[to][0],dp[to][1]-2*c);
}
}
题目保证输入为一颗树.
简单树形dp--想清楚为什么可以这样转移.
void solve(){ F 简单的树形dp,每个点有选和不选两种状态
cin>>n>>c;
for(int i=1;i<=n;i++) cin>>gold[i],dp[i][0]=dp[i][1]=0,vct[i].clear(); init
for(int i=1;i<=n-1;i++){
int u,v; cin>>u>>v;
vct[v].emplace_back(u);
vct[u].emplace_back(v);
}
dfs(1,-1);
cout<<max(dp[1][0],dp[1][1])<<endl;
}