1小木乃伊到我家(题目链接)
用来加深迪杰斯特拉算法印象的一个板子题,具体代码如下。
#include<cmath>
#include <iostream>
#include<stdio.h>
#include<string>
#include<string.h>
#include<vector>
#include<set>
#include<map>
#include<cstring>
#include<math.h>
#include<stack>
#include<algorithm>
#include<queue>
#include<bitset>
#include<sstream>
#define ll long long int
const int mod=1e9+7;
using namespace std;
struct node
{
ll to,next,wi;
}t[200010];
ll a[100010],dis[200010],tot=0,b[200010],ans=1000000000,s,vis[20];
priority_queue<pair<int,int> >p;
void add(int x,int y,int z)
{
t[++tot].to=y;
t[tot].wi=z;
t[tot].next=a[x];
a[x]=tot;
}
int main()
{
ios::sync_with_stdio(false);
ll n,m,i,j,x,y,z;
cin>>n>>m;
for(i=0;i<=n;i++)
{
a[i]=-1;
}
for(i=0;i<m;i++)
{
cin>>x>>y>>z;
add(x,y,z);
add(y,x,z);
}
memset(b,0,sizeof(b));
for(i=0;i<=n;i++)
dis[i]=1e11;
dis[1]=0;
p.push(make_pair(0,1));
while(!p.empty())
{
x=p.top().second;
p.pop();
if(b[x])
continue;
b[x]=1;
for(j=a[x];j!=-1;j=t[j].next)
{
if(dis[t[j].to]>dis[x]+t[j].wi)
{
dis[t[j].to]=dis[x]+t[j].wi;
p.push(make_pair(-dis[t[j].to],t[j].to));
}
}
}
if(dis[n]<1e11)
cout<<dis[n]<<endl;
else
cout<<"qwbbaka";
return 0;
}
2换个角度思考(题目链接)
这个题是个线段树的题,每一个区间内存的不仅有起点和终点,还有对应这个区间的所有数字的有序排列,然后用upper__bound来确定区间符合条件的数字数目即可。具体代码如下。
#include<cmath>
#include <iostream>
#include<stdio.h>
#include<string>
#include<string.h>
#include<vector>
#include<set>
#include<map>
#include<cstring>
#include<math.h>
#include<stack>
#include<algorithm>
#include<queue>
#include<bitset>
#define ll long long int
const int mod=10007;
using namespace std;
int a[100010];
struct no
{
int l,r;
vector<int>v;
}t[400010];
vector<int>::iterator it;
void bulid(int l,int r,int ri)
{
t[ri].l=l;
t[ri].r=r;
t[ri].v.clear();
int i;
for(i=l;i<=r;i++)
t[ri].v.push_back(a[i]);
sort(t[ri].v.begin(),t[ri].v.end());
if(l==r)
return;
int mid=(l+r)>>1;
bulid(l,mid,ri<<1);
bulid(mid+1,r,ri<<1|1);
return;
}
int cha(int l,int r,int ri,int k)
{
if(l<=t[ri].l&&r>=t[ri].r)
{
it=upper_bound(t[ri].v.begin(),t[ri].v.end(),k);
return it-t[ri].v.begin();
}
int mid=t[ri<<1].r,ans=0;
if(l<=mid)
{
ans+=cha(l,r,ri<<1,k);
}
if(r>mid)
{
ans+=cha(l,r,ri<<1|1,k);
}
return ans;
}
int main()
{
ios::sync_with_stdio(false);
int t,n,m,i,j=1,k;
cin>>n>>m;
for(i=1;i<=n;i++)
cin>>a[i];
bulid(1,n,1);
while(m--)
{
cin>>i>>j>>k;
cout<<cha(i,j,1,k)<<endl;
}
return 0;
}