Problem A 钢铁侠的诞生
现在有$n$个数字$a_i leq 10^9 $，然后取出$m$个数字，保证合法。
从小到大输出剩余的$n-m$个数字。
对于100%的数据$mleq n leq 3 imes 10^5$
Sol : 直接map映射然后用iterator来遍历整个map输出答案即可。
复杂度大概是$O(n log_2 n)$
# pragma GCC optimize(2) # include<bits/stdc++.h> using namespace std; int n,m; map<int,int>mp; int main() { scanf("%d%d",&n,&m); for (int i=1;i<=n;i++) { int t; scanf("%d",&t); mp[t]++; } for (int i=1;i<=m;i++) { int t; scanf("%d",&t); mp[t]--; } map<int,int>::iterator it; for (it=mp.begin();it!=mp.end();++it) { pair<int,int>p=*it; for (int j=1;j<=p.second;j++) printf("%d ",p.first); } return 0; }
Problem B 钢铁侠的逃离
设函数$f(x) = $x的二进制中"1"的个数。 求 $sumlimits_{i=1}^n f(B+i imes A)$ 的值。
对于100%的数据$Aleq 10^4 ,B leq 10^{16} nleq 10^{12} $
Sol : 对于每一个二进制位分别考虑，我们想法是对于二进制数第$k$位 $ B+A $和 $ B+(2^k+1)A $是相同的。
证明的话直接可以把两个数同时减去一个A，然后就变成了$B$和$B+(2^k)A$第$k$位的关系。
左式的$2^k A$相当于把A向左移k位，这个时候第k位一定是B的k位的数+0 = B的k位的数。 那么就不会发生变化。
所以我们只需要考虑一个循环节$B+A ... B+2^k A$中1的个数即可。
对于每个循环节，显然（也可以通过打表发现）每一位连续是1连续是0的可能比较大。所以我们考虑对于连续的1和连续的0一起处理。
设$f(a,b,k,r) = sumlimits_{i=0} ^ {r} b+a imes i 中第k位1的个数 $
考虑如何算$f(a,b,k,r)$ . 首先我们需要在每一个01块内用O(1)的时间完成跳跃。
于是考虑什么时候是01的分界点。由于进位的问题所以每一个01块分界点一定是以$2^k$一个循环的，并且是$2^k$的倍数。
考虑一个位置 b 我们什么时候才能跳出这个块呢？ 比该点更加靠右的$2^k$倍数的点可以轻易算出 $2^k imes (left lfloor frac{b}{2^k} ight floor imes +1 ) $ 就是下一个01分界点的坐标。
注意到01分界点事实上已经完成了进位即完成了01转换。
所以如果要跳到严格的下一个01分界点的右侧，我们需要跳跃$c= frac{2^k imes (left lfloor frac{b}{2^k} ight floor +1 ) -1 }{a}+1$步
所以，对于$f(a,b,k,r)$的求法已经非常明确了：
$f(a,b,k,r) = left{egin{matrix} b and 2^k = 0 left{egin{matrix} 0 & r-c<0\ f(a,b+c imes a,k,r-c) & r geq c end{matrix} ight.\ b and 2^k = 1 left{egin{matrix} r+1 & r-c<0\ c+f(a,b+c imes a,k,r-c)& r geq c end{matrix} ight.\ end{matrix} ight.$
由于是从0开始的，所以n必须等于n-1 ，当前是从b+a开始的，所以是b=b+a
答案就是$ans = sumlimits_{i=0}^{bit\_num(a+nb)} f(a,b,i,2^{i+1}-1) imes left lfloor frac{n}{2^{i+1}} ight floor+ f(a,b+left lfloor frac{n}{2^{i+1}} ight floor imes 2^{i+1} imes a,i,n mod 2^{i+1})$
# include <bits/stdc++.h> # define int long long using namespace std; int fun(int a,int b,int k,int r){ int d=b/(1ll<<k),c=((1ll<<k)*(d+1)-1-b)/a+1; if(d&1) return c>r?r+1:c+fun(a,b+c*a,k,r-c); else return c>r?0:fun(a,b+c*a,k,r-c); } signed main() { int T; scanf("%d",&T); while (T--) { int a,b,n; scanf("%lld%lld%lld",&a,&b,&n); int ans=0; b+=a; n--; for (int i=0;i<=54;i++) { int c=n/(1ll<<i+1); ans+=fun(a,b,i,(1ll<<i+1)-1)*c+fun(a,b+(1ll<<i+1)*c*a,i,n-c*(1ll<<i+1)); } printf("%lld ",ans); } return 0; }
Problem C 钢铁侠的复仇
设一个$N imes M$的矩阵，设$A_{i,j}$表示点$(i,j)$被攻克的难度而$B_{i,j}$ 表示点$(i,j)$被攻克的时间。
从$(i_1,j_1)$点转移到$(i_2,j_2)$点需要花费 $|i_1 - i_2|+|j_1-j_2|$ 的代价。
一条合法的攻克路径满足：经过路径上的所有点不重复而且不能经过$A_{i,j} = B_{i,j} = 0$的点。
输出沿着最长时间路径访问的时间。
对于100%的数据$ n,m leq 10^3 $
Sol ：本题是一个DP题目。
设$f_{i,j}$表示经过到点$(i,j)$结尾的路径的最大时间。
显然需要按照$A_{i,j}$值递增的顺序依次更新每个点。
转移方程就是$f_{i,j} = maxlimits_{k=1,w=1} ^ {kleq n , wleq m} [(A_{k,w} < A_{i,j}) f_{k,w}+|i-k|+|j-w|] + b_{i,j} $
复杂度 $O(n^2 m^2)$ get 30pts
# pragma GCC optimize(2) # include <bits/stdc++.h> # define int long long using namespace std; const int N=1e3+10; struct rec{ int x,y,d; }; int n,m; struct cmp { bool operator () (rec a,rec b) { return a.d>b.d; } }; int a[N][N],b[N][N],f[N][N]; priority_queue<rec,vector<rec>,cmp>q; signed main() { scanf("%d%d",&n,&m); for (int i=1;i<=n;i++) for (int j=1;j<=m;j++) scanf("%d",&a[i][j]),q.push((rec){i,j,a[i][j]}); for (int i=1;i<=n;i++) for (int j=1;j<=m;j++) scanf("%d",&b[i][j]); int ans=0; while (!q.empty()) { rec u=q.top();q.pop(); if (a[u.x][u.y]==0&&b[u.x][u.y]==0) continue; f[u.x][u.y]=b[u.x][u.y]; for (int i=1;i<=n;i++) for (int j=1;j<=m;j++) if (a[i][j]!=0&&a[i][j]<a[u.x][u.y]&&!(i==u.x&&j==u.y)) f[u.x][u.y]=max(f[u.x][u.y],f[i][j]+b[u.x][u.y]+abs(u.x-i)+abs(u.y-j)); ans=max(ans,f[u.x][u.y]); } printf("%d ",ans); return 0; }
考虑优化这个转移。
显然绝对值符号可以被拆，例如$(x,y)$在$(i,j)$左上角的时候($i geq x , j geq y$)时 有$f_{i,j} +| i - x | +| j - y | = f_{i,j}+i+j - x - y$
还有其他的三种情况不再赘述。
按照@hjc20032003的方法，可以先按$A_{i,j}$的大小把元素分成若干块(每一块里面的所有元素$A_{i,j}$值都相同)。
显然第k块中的$f_{i,j}$是k-1那一块的$f_{i,j}$转移而来。
我们先考虑$(x,y)$在$(i,j)$左上角的情况。
可以把相邻两块和并（由于当前行一定是从前一个可能的$A_{i,j}$转移而来的），并按照x坐标排序，然后从先到后依次扫，如果当前的元素$(x,y)$之前在前一个块(k-1)中那么插入到线段树第y号位子中。
如果当前元素在当前块$k-1$中那么利用当前线段树的元素查询在他左上角元素的信息来更新当前位置的f值$f_{x,y}$。
由于插入线段树的元素的$x$的坐标都是小于当前的元素的$x$坐标的，并且我们可以通过查询当前元素$y$之前区间信息而控制转移来源的点是在当前点的左上方的。
当然，由于可能从4个不同的方向转移，类似的操作要做4次(x轴翻转,y轴翻转,o点翻转,不翻转) 。 具体可以参考[Violet]天使玩偶/SJY摆棋子 的处理方法。
复杂度大概是$O(4 imes nm log_2 nm)$
#include<bits/stdc++.h> #define GX(x,y) x=max(x,y) #define ls(x) x<<1 #define rs(x) x<<1|1 #define A first #define B second #define mk make_pair #define pb push_back #define int long long #define REP(i,s,t) for(int i=s;i<=t;i++) using namespace std; const int maxn=1005,inf=0x3fffffffffffffff; typedef pair<int,int> pii; vector<pii> vec[maxn*maxn]; int n,m,dis[maxn*maxn],a[maxn][maxn],b[maxn][maxn],f[maxn][maxn]; struct rec{int x,y; bool w;}q[maxn*maxn]; bool tag[maxn<<2]; int maxx[maxn<<2]; void push_down(int p){ if(tag[p]) tag[ls(p)]=tag[rs(p)]=true,maxx[ls(p)]=maxx[rs(p)]=-inf,tag[p]=false; } void _modify(int p,int l,int r,int tar,int val){ if(l==r){GX(maxx[p],val); return;} int m=l+r>>1; push_down(p); if(tar<=m) _modify(ls(p),l,m,tar,val); else _modify(rs(p),m+1,r,tar,val); maxx[p]=max(maxx[ls(p)],maxx[rs(p)]); } void modify(int tar,int val){_modify(1,1,m,tar,val);} int _query(int p,int l,int r,int nl,int nr){ if(nl<=l&&r<=nr) return maxx[p]; int m=l+r>>1,ret=-inf; push_down(p); if(nl<=m) GX(ret,_query(ls(p),l,m,nl,nr)); if(m<nr) GX(ret,_query(rs(p),m+1,r,nl,nr)); return ret; } int query(int nl,int nr){return _query(1,1,m,nl,nr);} bool cmp1(rec a,rec b){ if(a.x!=b.x) return a.x<b.x; return a.y<b.y; } signed main(){ // freopen("c.in","r",stdin); // freopen("c.out","w",stdout); scanf("%lld%lld",&n,&m); int dis_cnt=0; REP(i,1,n) REP(j,1,m) scanf("%lld",&a[i][j]); REP(i,1,n) REP(j,1,m) scanf("%lld",&b[i][j]); REP(i,1,n) REP(j,1,m) if(a[i][j]) dis[++dis_cnt]=a[i][j]; sort(dis+1,dis+1+dis_cnt); int cnt=unique(dis+1,dis+1+dis_cnt)-dis-1; REP(i,1,n) REP(j,1,m) if(a[i][j]) a[i][j]=lower_bound(dis+1,dis+cnt+1,a[i][j])-dis,vec[a[i][j]].pb(mk(i,j)); for(int i=0;i<vec[1].size();i++) f[vec[1][i].A][vec[1][i].B]=b[vec[1][i].A][vec[1][i].B]; REP(c,2,cnt){ int pt=0; for(int i=0;i<vec[c-1].size();i++) q[++pt]=(rec){vec[c-1][i].A,vec[c-1][i].B,0}; for(int i=0;i<vec[c].size();i++) q[++pt]=(rec){vec[c][i].A,vec[c][i].B,1}; sort(q+1,q+1+pt,cmp1); tag[1]=true; maxx[1]=-inf; REP(i,1,pt) if(!q[i].w) modify(q[i].y,f[q[i].x][q[i].y]-q[i].x-q[i].y); else GX(f[q[i].x][q[i].y],query(1,q[i].y)+b[q[i].x][q[i].y]+q[i].x+q[i].y); tag[1]=true; maxx[1]=-inf; for(int i=pt;i;i--) if(!q[i].w) modify(q[i].y,f[q[i].x][q[i].y]+q[i].x-q[i].y); else GX(f[q[i].x][q[i].y],query(1,q[i].y)+b[q[i].x][q[i].y]-q[i].x+q[i].y); tag[1]=true; maxx[1]=-inf; REP(i,1,pt) if(!q[i].w) modify(q[i].y,f[q[i].x][q[i].y]-q[i].x+q[i].y); else GX(f[q[i].x][q[i].y],query(q[i].y,m)+b[q[i].x][q[i].y]+q[i].x-q[i].y); tag[1]=true; maxx[1]=-inf; for(int i=pt;i;i--) if(!q[i].w) modify(q[i].y,f[q[i].x][q[i].y]+q[i].x+q[i].y); else GX(f[q[i].x][q[i].y],query(q[i].y,m)+b[q[i].x][q[i].y]-q[i].x-q[i].y); } int ans=0; REP(i,1,n) REP(j,1,m) GX(ans,f[i][j]); cout<<ans<<endl; return 0; }
但是，这样处理的常数非常大，由于绝对值符号的性质，按照上述四种转移方法只可能有一种转移方法是正确的。
由于绝对值的性质有$|i-x| geq i-x$ 所以有
$ f_{i,j} + |i-x| + |j - y| = max{ f_{i,j} + (i - x) + (j - y), f_{i,j} + (i - x) + (y - j), f_{i,j} + (x - i) + (j - y), f_{i,j} + (x - i) + (y - j)} $ 成立。
所以转移的时候可以直接无视$i,x,j,y$的大小关系，一并转移即可。
转移方程是 $ f_{x,y}= max{ -x - y + max(f_{i,j} + i + j), -x + y + max(f_{i,j} + i - j), x - y + max(f_{i,j} - i + j), x + y + max(f_{i,j} - i - j) ) $
其中$max(f_{i,j} + i + j) ... $是全局变量在每次转移完成后维护即可。
还要注意初始值不能加上坐标 ， 即非0的最小的$A_{i,j}$所对应的$f_{i,j}$一开始从自己转移不能加上横纵坐标!!!
# include <bits/stdc++.h> # define int long long # define inf (0x3f3f3f3f3f3f3f3f) using namespace std; const int N=1e3+10; int f[N][N],b[N][N]; vector<pair<int,int> >v[N*N]; pair<int,int>t[N*N]; int mx1,mx2,mx3,mx4,ans,mint; int n,m; int Max(int a,int b,int c,int d) { if (b>a) a=b; if (c>a) a=c; if (d>a) a=d; return a; } void work(int r) { if (v[r].size()==0) return; int cnt=0;for (int i=0;i<v[r].size();i++) t[++cnt]=v[r][i]; int mxa=0,mxb=0,mxc=0,mxd=0; for (int i=1;i<=cnt;i++) { int x=t[i].first,y=t[i].second; if (r==0) continue; if (r!=mint) f[x][y]=Max(-x-y+mx1,-x+y+mx2,x-y+mx3,x+y+mx4)+b[x][y]; else f[x][y]=b[x][y]; ans=max(ans,f[x][y]); mxa=max(mxa,f[x][y]+x+y); mxb=max(mxb,f[x][y]+x-y); mxc=max(mxc,f[x][y]-x+y); mxd=max(mxd,f[x][y]-x-y); } mx1=max(mx1,mxa); mx2=max(mx2,mxb); mx3=max(mx3,mxc); mx4=max(mx4,mxd); } signed main() { scanf("%lld%lld",&n,&m); mint=inf; for (int i=1;i<=n;i++) for (int j=1;j<=m;j++) { int t; scanf("%lld",&t); if (t!=0) mint=min(mint,t); v[t].push_back(make_pair(i,j)); } for (int i=1;i<=n;i++) for (int j=1;j<=m;j++) scanf("%lld",&b[i][j]); for (int a=0;a<=1000000;a++) work(a); printf("%lld ",ans); return 0; }