洛谷 P8318 题解

题目传送门

---

题意：

给出一个长度为 $n$ 的操作后的序列，然后给出 $m$ 次操作过程，每次给出操作类型 $op$、操作数 $x$ 和 $y$，求出操作前的原始序列。

在操作中，如果 $x=y$，那么新的 $x$ 就分别等于原始 $x$ 的
两倍或平方。即：如果 $op=1$，那么新的第 $x$ 个元素就等于它的两倍；如果 $op=2$，那么新的第 $x$ 个元素就等于它的平方。

---

思路：

模拟题，但是坑点很多。

• 由于给出的是操作后的序列，所以我们在还原时需要倒序还原，例如：$n=2$，两次操作分别是 $op1,x1,y1$ 和 $op2,x2,y2$。那么我们就需要先还原 $op2,x2,y2$ 再还原 $op1,x1,y1$。

• 同上，需要倒序还原，如果操作中是乘法，还原时就需要用除法，加减法同理。

• 题目中还提到 $x=y$ 的情况。当 $op=1$ 时，原操作是将第 $x$ 个元素 $×2$，那么我们在还原时就需要将它 $÷2$；同理，当 $op=2$ 时，原操作是将第 $x$ 个元素变成它的平方，那么我们在还原时就要将它进行 sqrt()。

---

code：

#include <bits/stdc++.h>
using namespace std;

inline long long read(){
    long long s=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9'){
        if(ch=='-') f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9'){
        s=s*10+ch-'0';
        ch=getchar();
    }
    return s*f;
}

int n,m;
long long a[1005];
int op[1005],x[1005],y[1005];

int main(){
    n=read();m=read();
    for(int i=1;i<=n;i++){
        a[i]=read();
    }
    for(int i=1;i<=m;i++){
        op[i]=read();x[i]=read();y[i]=read();
	}
    for(int i=m;i>=1;i--){ 
    	if(op[i]==1){
		    if(x[i]==y[i]){
		    	a[x[i]]/=2;
			}
			else{
	    		a[x[i]]-=a[y[i]];			
			} 


		}
		else{
		    if(x[i]==y[i]){
		    	a[x[i]]=sqrt(a[x[i]]);
			}
			else{
				a[x[i]]/=a[y[i]];
			}

		}
	}
    for(int i=1;i<=n;i++){
        printf("%lld ",a[i]);
    }
    return 0;
}